Merge pull request #32 from tkowark/to_attrs_fix

Test and fix for the #to_attrs method of Sawyer::Resource
lostisland · Dec 2, 2014 · 454ca30 · 454ca30
2 parents 826e09b + 4160a3d
commit 454ca30
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/sawyer/resource.rb b/lib/sawyer/resource.rb
@@ -130,6 +130,8 @@ def to_attrs
       hash.keys.each do |k|
         if hash[k].is_a?(Sawyer::Resource)
           hash[k] = hash[k].to_attrs
+        elsif hash[k].is_a?(Array) && hash[k].all?{|el| el.is_a?(Sawyer::Resource)}
+          hash[k] = hash[k].collect{|el| el.to_attrs}          
         end
       end
       hash

diff --git a/test/resource_test.rb b/test/resource_test.rb
@@ -127,6 +127,12 @@ def test_to_h
       hash = {:a => 1 }
       assert_equal hash, res.to_h
     end
+
+    def test_to_attrs_for_sawyer_resource_arrays
+      res = Resource.new @agent, :a => 1, :b => [Resource.new(@agent, :a => 2)]
+      hash = {:a => 1, :b => [{:a => 2}]}
+      assert_equal hash, res.to_attrs
+    end
 
     def test_handle_hash_notation_with_string_key
       res = Resource.new @agent, :a => 1