45 lines
1.1 KiB
Ruby
45 lines
1.1 KiB
Ruby
require './test/test_helper'
|
|
require 'logger'
|
|
|
|
class CursorTest < Test::Unit::TestCase
|
|
|
|
include Mongo
|
|
|
|
@@connection = Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
|
|
ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT)
|
|
@@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
|