pahagon-mongo-abd 0.14.1

data/test/test_cursor.rb

@@ -0,0 +1,331 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'mongo'
+require 'test/unit'
+
+# NOTE: assumes Mongo is running
+class CursorTest < Test::Unit::TestCase
+
+  include Mongo
+
+  @@db = Connection.new(ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost',
+                        ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT).db('ruby-mongo-test')
+  @@coll = @@db.collection('test')
+
+  def setup
+    @@coll.clear
+    @@coll.insert('a' => 1)     # collection not created until it's used
+    @@coll_full_name = 'ruby-mongo-test.test'
+  end
+
+  def teardown
+    @@coll.clear
+    @@db.error
+  end
+
+  def test_explain
+    cursor = @@coll.find('a' => 1)
+    explaination = cursor.explain
+    assert_not_nil explaination['cursor']
+    assert_kind_of Numeric, explaination['n']
+    assert_kind_of Numeric, explaination['millis']
+    assert_kind_of Numeric, explaination['nscanned']
+  end
+
+  def test_count
+    @@coll.clear
+
+    assert_equal 0, @@coll.find().count()
+
+    10.times do |i|
+      @@coll.save("x" => i)
+    end
+
+    assert_equal 10, @@coll.find().count()
+    assert_kind_of Integer, @@coll.find().count()
+    assert_equal 10, @@coll.find({}, :limit => 5).count()
+    assert_equal 10, @@coll.find({}, :skip => 5).count()
+
+    assert_equal 1, @@coll.find({"x" => 1}).count()
+    assert_equal 5, @@coll.find({"x" => {"$lt" => 5}}).count()
+
+    a = @@coll.find()
+    b = a.count()
+    a.each do |doc|
+      break
+    end
+    assert_equal b, a.count()
+
+    assert_equal 0, @@db['acollectionthatdoesn'].count()
+  end
+
+  def test_sort
+    @@coll.clear
+    5.times{|x| @@coll.insert({"a" => x}) }
+
+    assert_kind_of Cursor, @@coll.find().sort({:a => 1})
+
+    assert_equal 0, @@coll.find().sort({:a => 1}).next_object["a"]
+    assert_equal 4, @@coll.find().sort({:a => -1}).next_object["a"]
+    assert_equal 0, @@coll.find().sort(["a"]).next_object["a"]
+
+    assert_kind_of Cursor, @@coll.find().sort({:a => -1, :b => 1})
+
+    assert_equal 4, @@coll.find().sort({:a => 1}).sort({:a => -1}).next_object["a"]
+    assert_equal 0, @@coll.find().sort({:a => -1}).sort({:a => 1}).next_object["a"]
+
+    cursor = @@coll.find()
+    cursor.next_object()
+    assert_raise InvalidOperation do
+      cursor.sort(["a"])
+    end
+  end
+
+  def test_limit
+    @@coll.clear
+
+    10.times do |i|
+      @@coll.save("x" => i)
+    end
+    assert_equal 10, @@coll.find().count()
+
+    results = @@coll.find().limit(5).to_a
+    assert_equal 5, results.length
+  end
+
+  def test_limit_exceptions
+    assert_raise ArgumentError do
+      cursor = @@coll.find().limit('not-an-integer')
+    end
+
+    cursor      = @@coll.find()
+    firstResult = cursor.next_object()
+    assert_raise InvalidOperation, "Cannot modify the query once it has been run or closed." do
+      cursor.limit(1)
+    end
+
+    cursor = @@coll.find()
+    cursor.close
+    assert_raise InvalidOperation, "Cannot modify the query once it has been run or closed." do
+      cursor.limit(1)
+    end
+  end
+
+  def test_skip
+    @@coll.clear
+
+    10.times do |i|
+      @@coll.save("x" => i)
+    end
+    assert_equal 10, @@coll.find().count()
+
+    all_results    = @@coll.find().to_a
+    skip_results = @@coll.find().skip(2).to_a
+    assert_equal 10, all_results.length
+    assert_equal 8,  skip_results.length
+
+    assert_equal all_results.slice(2...10), skip_results
+  end
+
+  def test_skip_exceptions
+    assert_raise ArgumentError do
+      cursor = @@coll.find().skip('not-an-integer')
+    end
+
+    cursor      = @@coll.find()
+    firstResult = cursor.next_object()
+    assert_raise InvalidOperation, "Cannot modify the query once it has been run or closed." do
+      cursor.skip(1)
+    end
+
+    cursor = @@coll.find()
+    cursor.close
+    assert_raise InvalidOperation, "Cannot modify the query once it has been run or closed." do
+      cursor.skip(1)
+    end
+  end
+
+  def test_limit_skip_chaining
+    @@coll.clear
+    10.times do |i|
+      @@coll.save("x" => i)
+    end
+
+    all_results = @@coll.find().to_a
+    limited_skip_results = @@coll.find().limit(5).skip(3).to_a
+
+    assert_equal all_results.slice(3...8), limited_skip_results
+  end
+
+  def test_close_no_query_sent
+    begin
+      cursor = @@coll.find('a' => 1)
+      cursor.close
+      assert cursor.closed?
+    rescue => ex
+      fail ex.to_s
+    end
+  end
+
+  def test_refill_via_get_more
+    begin
+      assert_equal 1, @@coll.count
+      1000.times { |i|
+        assert_equal 1 + i, @@coll.count
+        @@coll.insert('a' => i)
+      }
+
+      assert_equal 1001, @@coll.count
+      count = 0
+      @@coll.find.each { |obj|
+        count += obj['a']
+      }
+      assert_equal 1001, @@coll.count
+
+      # do the same thing again for debugging
+      assert_equal 1001, @@coll.count
+      count2 = 0
+      @@coll.find.each { |obj|
+        count2 += obj['a']
+      }
+      assert_equal 1001, @@coll.count
+
+      assert_equal count, count2
+      assert_equal 499501, count
+    rescue Test::Unit::AssertionFailedError => ex
+      p @@db.collection_names
+      Process.exit 1
+    end
+  end
+
+  def test_refill_via_get_more_alt_coll
+    begin
+      coll = @@db.collection('test-alt-coll')
+      coll.clear
+      coll.insert('a' => 1)     # collection not created until it's used
+      assert_equal 1, coll.count
+
+      1000.times { |i|
+        assert_equal 1 + i, coll.count
+        coll.insert('a' => i)
+      }
+
+      assert_equal 1001, coll.count
+      count = 0
+      coll.find.each { |obj|
+        count += obj['a']
+      }
+      assert_equal 1001, coll.count
+
+      # do the same thing again for debugging
+      assert_equal 1001, coll.count
+      count2 = 0
+      coll.find.each { |obj|
+        count2 += obj['a']
+      }
+      assert_equal 1001, coll.count
+
+      assert_equal count, count2
+      assert_equal 499501, count
+    rescue Test::Unit::AssertionFailedError => ex
+      p @@db.collection_names
+      Process.exit 1
+    end
+  end
+
+  def test_close_after_query_sent
+    begin
+      cursor = @@coll.find('a' => 1)
+      cursor.next_object
+      cursor.close
+      assert cursor.closed?
+    rescue => ex
+      fail ex.to_s
+    end
+  end
+
+  def test_kill_cursors
+    @@coll.drop
+
+    client_cursors = @@db.db_command("cursorInfo" => 1)["clientCursors_size"]
+    by_location = @@db.db_command("cursorInfo" => 1)["byLocation_size"]
+
+    10000.times do |i|
+      @@coll.insert("i" => i)
+    end
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    10.times do |i|
+      @@coll.find_one()
+    end
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    10.times do |i|
+      a = @@coll.find()
+      a.next_object()
+      a.close()
+    end
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    a = @@coll.find()
+    a.next_object()
+
+    assert_not_equal(client_cursors,
+                     @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_not_equal(by_location,
+                     @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    a.close()
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    a = @@coll.find({}, :limit => 10).next_object()
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    @@coll.find() do |cursor|
+      cursor.next_object()
+    end
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+
+    @@coll.find() { |cursor|
+      cursor.next_object()
+    }
+
+    assert_equal(client_cursors,
+                 @@db.db_command("cursorInfo" => 1)["clientCursors_size"])
+    assert_equal(by_location,
+                 @@db.db_command("cursorInfo" => 1)["byLocation_size"])
+  end
+
+  def test_count_with_fields
+    @@coll.clear
+    @@coll.save("x" => 1)
+
+    @@coll.find({}, :fields => ["a"]).each do |doc|
+      fail "shouldn't have any results here"
+    end
+    assert_equal(0, @@coll.find({}, :fields => ["a"]).count())
+  end
+end

data/test/test_db.rb

@@ -0,0 +1,185 @@
+$LOAD_PATH[0,0] = File.join(File.dirname(__FILE__), '..', 'lib')
+require 'digest/md5'
+require 'mongo'
+require 'test/unit'
+
+class TestPKFactory
+  def create_pk(row)
+    row['_id'] ||= Mongo::ObjectID.new
+    row
+  end
+end
+
+# NOTE: assumes Mongo is running
+class DBTest < Test::Unit::TestCase
+
+  include Mongo
+
+  @@host = ENV['MONGO_RUBY_DRIVER_HOST'] || 'localhost'
+  @@port = ENV['MONGO_RUBY_DRIVER_PORT'] || Connection::DEFAULT_PORT
+  @@db = Connection.new(@@host, @@port).db('ruby-mongo-test')
+  @@users = @@db.collection('system.users')
+
+  def setup
+    @spongebob = 'spongebob'
+    @spongebob_password = 'squarepants'
+    @@users.clear
+    @@users.insert(:user => @spongebob, :pwd => @@db.send(:hash_password, @spongebob, @spongebob_password))
+  end
+
+  def teardown
+    @@users.clear if @@users
+    @@db.error
+  end
+
+  def test_close
+    @@db.close
+    assert !@@db.connected?
+    begin
+      @@db.collection('test').insert('a' => 1)
+      fail "expected 'NilClass' exception"
+    rescue => ex
+      assert_match /NilClass/, ex.to_s
+    ensure
+      @@db = Connection.new(@@host, @@port).db('ruby-mongo-test')
+      @@users = @@db.collection('system.users')
+    end
+  end
+
+  def test_full_coll_name
+    coll = @@db.collection('test')
+    assert_equal 'ruby-mongo-test.test', @@db.full_coll_name(coll.name)
+  end
+
+  def test_collection_names
+    @@db.collection("test").insert("foo" => 5)
+    @@db.collection("test.mike").insert("bar" => 0)
+
+    colls = @@db.collection_names()
+    assert colls.include?("test")
+    assert colls.include?("test.mike")
+    colls.each { |name|
+      assert !name.include?("$")
+    }
+  end
+
+  def test_collections
+    @@db.collection("test.durran").insert("foo" => 5)
+    @@db.collection("test.les").insert("bar" => 0)
+
+    colls = @@db.collections()
+    assert_not_nil colls.select { |coll| coll.name == "test.durran" }
+    assert_not_nil colls.select { |coll| coll.name == "test.les" }
+    assert_equal [], colls.select { |coll| coll.name == "does_not_exist" }
+
+    assert_kind_of Collection, colls[0]
+  end
+
+  def test_pair
+    @@db.close
+    @@users = nil
+    @@db = Connection.new({:left => "this-should-fail", :right => [@@host, @@port]}).db('ruby-mongo-test')
+    assert @@db.connected?
+  ensure
+    @@db = Connection.new(@@host, @@port).db('ruby-mongo-test') unless @@db.connected?
+    @@users = @@db.collection('system.users')
+  end
+
+  def test_pk_factory
+    db = Connection.new(@@host, @@port).db('ruby-mongo-test', :pk => TestPKFactory.new)
+    coll = db.collection('test')
+    coll.clear
+
+    insert_id = coll.insert('name' => 'Fred', 'age' => 42)
+    # new id gets added to returned object
+    row = coll.find_one({'name' => 'Fred'})
+    oid = row['_id']
+    assert_not_nil oid
+    assert_equal insert_id, oid
+
+    oid = ObjectID.new
+    data = {'_id' => oid, 'name' => 'Barney', 'age' => 41}
+    coll.insert(data)
+    row = coll.find_one({'name' => data['name']})
+    db_oid = row['_id']
+    assert_equal oid, db_oid
+    assert_equal data, row
+
+    coll.clear
+  end
+
+  def test_pk_factory_reset
+    db = Connection.new(@@host, @@port).db('ruby-mongo-test')
+    db.pk_factory = Object.new # first time
+    begin
+      db.pk_factory = Object.new
+      fail "error: expected exception"
+    rescue => ex
+      assert_match /can not change PK factory/, ex.to_s
+    ensure
+      db.close
+    end
+  end
+
+  def test_authenticate
+    assert !@@db.authenticate('nobody', 'nopassword')
+    assert !@@db.authenticate(@spongebob, 'squareliederhosen')
+    assert @@db.authenticate(@spongebob, @spongebob_password)
+  end
+
+  def test_logout
+    @@db.logout                  # only testing that we don't throw exception
+  end
+
+  def test_auto_connect
+    @@db.close
+    db = Connection.new(@@host, @@port, :auto_reconnect => true).db('ruby-mongo-test')
+    assert db.connected?
+    assert db.auto_reconnect?
+    db.close
+    assert !db.connected?
+    assert db.auto_reconnect?
+    db.collection('test').insert('a' => 1)
+    assert db.connected?
+  ensure
+    @@db = Connection.new(@@host, @@port).db('ruby-mongo-test')
+    @@users = @@db.collection('system.users')
+  end
+
+  def test_error
+    @@db.reset_error_history
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert_nil @@db.previous_error
+
+    @@db.send(:db_command, :forceerror => 1)
+    assert @@db.error?
+    assert_not_nil @@db.error
+    assert_not_nil @@db.previous_error
+
+    @@db.send(:db_command, :forceerror => 1)
+    assert @@db.error?
+    assert @@db.error
+    prev_error = @@db.previous_error
+    assert_equal 1, prev_error['nPrev']
+    assert_equal prev_error["err"], @@db.error
+
+    @@db.collection('test').find_one
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert @@db.previous_error
+    assert_equal 2, @@db.previous_error['nPrev']
+
+    @@db.reset_error_history
+    assert_nil @@db.error
+    assert !@@db.error?
+    assert_nil @@db.previous_error
+  end
+
+  def test_text_port_number
+    db = DB.new('ruby-mongo-test', [[@@host, @@port.to_s]])
+    # If there is no error, all is well
+    db.collection('users').clear
+  end
+
+end