mongo-parser-rb 0.0.3 → 0.0.4

data/lib/mongo-parser-rb/field.rb

@@ -9,7 +9,12 @@ module MongoParserRB
     def value_in_document(document)
       document = stringify_keys(document)
       @field_parts.reduce(document) do |value, field|
-        value[field]
+        case value
+        when Array
+          value[field.to_i]
+        when Hash
+          value[field]
+        end
       end
     rescue NoMethodError
       nil
@@ -30,7 +35,15 @@ module MongoParserRB
 
     def stringify_keys(document)
       document.reduce({}) do |new_document, (k,v)|
-        new_document[k.to_s] = v.kind_of?(Hash) ? stringify_keys(v) : v
+        new_document[k.to_s] = case v
+        when Hash
+          stringify_keys(v)
+        when Array
+          v.map { |e| e.kind_of?(Hash) ? stringify_keys(e) : e }
+        else
+          v
+        end
+
         new_document
       end
     end

data/lib/mongo-parser-rb/version.rb

@@ -1,3 +1,3 @@
 module MongoParserRB
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/test/mongo-parser-rb/field_test.rb

@@ -23,4 +23,18 @@ class FieldTest < MiniTest::Unit::TestCase
     refute field.in_document?(:custom_data => {:tracked_users => 10})
   end
 
+  def test_returning_array_element
+    field = MongoParserRB::Field.new(:"something.array.0")
+    assert_equal 1, field.value_in_document(:something => {:array => [1,2]})
+  end
+
+  def test_returning_hash_in_array
+    field = MongoParserRB::Field.new(:"something.array.0.key")
+    document = {:something => {:array => [{:key => 'hello world'}, {:key => 'bye world'}]}}
+    assert_equal 'hello world', field.value_in_document(document)
+
+    field = MongoParserRB::Field.new(:"something.array.1.key")
+    assert_equal 'bye world', field.value_in_document(document)
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongo-parser-rb
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors: