Make ap return the printed object.

shageman · Apr 29, 2011 · 05726bf · 05726bf
1 parent 0f14846
commit 05726bf
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/ap/core_ext/kernel.rb b/lib/ap/core_ext/kernel.rb
@@ -13,6 +13,7 @@ def ai(options = {})
 
   def ap(object, options = {})
     puts object.ai(options)
+    object
   end
   alias :awesome_print :ap
 

diff --git a/spec/awesome_print_spec.rb b/spec/awesome_print_spec.rb
@@ -7,6 +7,13 @@
     stub_dotfile!
   end
 
+  describe "ap method" do
+    it "return value" do
+      object = ""
+      (ap object).should eql(object)
+    end
+  end
+
   describe "Array" do
     before(:each) do
       @arr = [ 1, :two, "three", [ nil, [ true, false] ] ]