test/test_identity_map.rb in mongoo-0.3.1 vs test/test_identity_map.rb in mongoo-0.4.0

@@ -6,9 +6,25 @@
     [Person, TvShow, SearchIndex].each do |obj|
       obj.drop
       obj.create_indexes
     end
   end
+  
+  should "be performant" do
+    1.upto(1000) do |i|
+      p = Person.new("name" => "Ben#{i}")
+      p.insert!
+    end
+    
+    Mongoo::IdentityMap.on!
+    
+    all = Person.find.to_a
+    
+    p = Person.find(name: "Ben5").next
+    assert_equal p.object_id, all[all.index(p)].object_id
+    
+    Mongoo::IdentityMap.off!
+  end
 
   should "set and get attributes" do
     p = Person.new("name" => "Ben")
     p.insert!
 
\ No newline at end of file