Ensure that include? works on many doc associations.

mongomapper · Aug 4, 2010 · bcd0e52 · bcd0e52
1 parent f1e52bc
commit bcd0e52
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/lib/mongo_mapper/plugins/associations/collection.rb b/lib/mongo_mapper/plugins/associations/collection.rb
@@ -12,6 +12,11 @@ def to_ary
           end
         end
 
+        def include?(*args)
+          load_target
+          target.include?(*args)
+        end
+
         def reset
           super
           target = []

diff --git a/test/functional/associations/test_many_documents_proxy.rb b/test/functional/associations/test_many_documents_proxy.rb
@@ -307,6 +307,16 @@ def setup
       @project2.save
     end
 
+    context "include?" do
+      should "return true if in association" do
+        @project1.statuses.should include(@brand_new)
+      end
+
+      should "return false if not in association" do
+        @project1.statuses.should_not include(@in_progress)
+      end
+    end
+
     context "dynamic finders" do
       should "work with single key" do
         @project1.statuses.find_by_name('New').should == @brand_new