spec/mongoid/token/finders_spec.rb in mongoid_token-2.1.1 vs spec/mongoid/token/finders_spec.rb in mongoid_token-2.1.2

@@ -1,11 +1,13 @@
 require File.join(File.dirname(__FILE__), %w[.. .. spec_helper])
 
 describe Mongoid::Token::Finders do
-  before :each do
+  after do
+    Object.send(:remove_const, :Document) if Object.constants.include?(:Document)
+    Object.send(:remove_const, :AnotherDocument) if Object.constants.include?(:AnotherDocument)
   end
-
+  
   it "define a finder based on a field_name" do
     klass = Class.new
     field = :another_token
     Mongoid::Token::Finders.define_custom_token_finder_for(klass, field)
     klass.singleton_methods.should include(:"find_by_#{field}")
@@ -30,12 +32,28 @@
     document = Document.create!(:token => "1234")
     Mongoid::Token::Finders.define_custom_token_finder_for(Document)
     Document.find_by_token("1234").should == document
   end
 
+  it 'retrieves multiple documents using the dynamic finder' do
+    class Document; include Mongoid::Document; field :token; end
+    document = Document.create!(:token => "1234")
+    document2 = Document.create!(:token => "5678")
+    Mongoid::Token::Finders.define_custom_token_finder_for(Document)
+    Document.find_by_token(["1234", "5678"]).should == [document, document2]
+  end
+
   it "retrieve a document using the `find` method" do
     class AnotherDocument; include Mongoid::Document; field :token; end
     document = AnotherDocument.create! :token => "1234"
     Mongoid::Token::Finders.define_custom_token_finder_for(AnotherDocument)
     AnotherDocument.find("1234").should == document
+  end
+
+  it 'retrieves multiple documents using the `find` method' do
+    class AnotherDocument; include Mongoid::Document; field :token; end
+    document = AnotherDocument.create! :token => "1234"
+    document2 = AnotherDocument.create! :token => "5678"
+    Mongoid::Token::Finders.define_custom_token_finder_for(AnotherDocument)
+    AnotherDocument.find(["1234", "5678"]).should == [document, document2]
   end
 end