mongo-ruby-driver/test/cursor_message_test.rb

44 lines
1.0 KiB
Ruby

require File.expand_path("../test_helper", __FILE__)
require 'logger'
class CursorMessageTest < Test::Unit::TestCase
include Mongo
@@connection = standard_connection
@@db = @@connection.db(MONGO_TEST_DB)
@@coll = @@db.collection('test')
@@version = @@connection.server_version
def setup
@@coll.remove
@@coll.insert('a' => 1) # collection not created until it's used
@@coll_full_name = "#{MONGO_TEST_DB}.test"
end
def test_valid_batch_sizes
assert_raise ArgumentError do
@@coll.find({}, :batch_size => 1, :limit => 5)
end
assert_raise ArgumentError do
@@coll.find({}, :batch_size => -1, :limit => 5)
end
assert @@coll.find({}, :batch_size => 0, :limit => 5)
end
def test_batch_size
@@coll.remove
200.times do |n|
@@coll.insert({:a => n})
end
list = @@coll.find({}, :batch_size => 2, :limit => 6).to_a
assert_equal 6, list.length
list = @@coll.find({}, :batch_size => 100, :limit => 101).to_a
assert_equal 101, list.length
end
end