spec/support/helpers/document_finder_stubs.rb in mongomodel-0.5.5 vs spec/support/helpers/document_finder_stubs.rb in mongomodel-0.5.6

- old
+ new

@@ -1,15 +1,15 @@ require 'rspec/mocks' module DocumentFinderStubs include RSpec::Mocks::ExampleMethods - + def stub_find(result) find_result = double('find result', :to_a => result.map { |doc| doc.to_mongo }, :count => result.size).as_null_object collection.stub(:find).and_return(find_result) end - + def should_find(expected={}, result=[]) selector, options = MongoModel::MongoOptions.new(self, expected).to_a find_result = double('find result', :to_a => result.map { |doc| doc.to_mongo }).as_null_object collection.should_receive(:find).once.with(selector, options).and_return(find_result) yield if block_given? @@ -24,25 +24,25 @@ selector, options = MongoModel::MongoOptions.new(self, expected).to_a find_result = double('find result', :count => result).as_null_object collection.should_receive(:find).once.with(selector, options).and_return(find_result) yield if block_given? end - + def should_not_count collection.should_not_receive(:find) yield if block_given? end - + def stub_delete collection.stub(:remove) end - + def should_delete(conditions={}) selector, options = MongoModel::MongoOptions.new(self, :conditions => conditions).to_a collection.should_receive(:remove).once.with(selector, options) yield if block_given? end - + def should_update(conditions={}, updates={}) selector, options = MongoModel::MongoOptions.new(self, :conditions => conditions).to_a collection.should_receive(:update).once.with(selector, { "$set" => updates }, { :multi => true }) yield if block_given? end