mongo 1.9.2 → 1.10.0.rc0

data/test/replica_set/complex_connect_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -36,8 +36,8 @@ class ComplexConnectTest < Test::Unit::TestCase
 
     version = @client.server_version
 
-    @client['test']['foo'].insert({:a => 1})
-    assert @client['test']['foo'].find_one
+    @client[TEST_DB]['complext-connect-test'].insert({:a => 1})
+    assert @client[TEST_DB]['complext-connect-test'].find_one
 
     config = primary['local']['system.replset'].find_one
     old_config = config.dup
@@ -55,19 +55,18 @@ class ComplexConnectTest < Test::Unit::TestCase
     end
     @rs.start
 
-
     assert_raise ConnectionFailure do
-      primary['admin'].command(step_down_command)
+      perform_step_down(primary)
     end
 
     # isMaster is currently broken in 2.1+ when called on removed nodes
     puts version
     if version < "2.1"
       rescue_connection_failure do
-        assert @client['test']['foo'].find_one
+        assert @client[TEST_DB]['complext-connect-test'].find_one
       end
 
-      assert @client['test']['foo'].find_one
+      assert @client[TEST_DB]['complext-connect-test'].find_one
     end
 
     primary = MongoClient.new(host, @rs.primary.port)

data/test/replica_set/connection_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,7 +14,7 @@
 
 require 'test_helper'
 
-class ConnectionTest < Test::Unit::TestCase
+class ReplicaSetConnectionTest < Test::Unit::TestCase
 
   def setup
     ensure_cluster(:rs)
@@ -59,33 +59,37 @@ class ConnectionTest < Test::Unit::TestCase
   end
 
   def test_connect_with_connection_string_in_env_var
-    ENV['MONGODB_URI'] = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name}"
-    @connection = ReplSetConnection.new
-    assert !@connection.nil?
-    assert_equal 3, @connection.seeds.length
-    assert_equal @rs.replicas[0].host, @connection.seeds[0][0]
-    assert_equal @rs.replicas[1].host, @connection.seeds[1][0]
-    assert_equal @rs.replicas[2].host, @connection.seeds[2][0]
-    assert_equal @rs.replicas[0].port, @connection.seeds[0][1]
-    assert_equal @rs.replicas[1].port, @connection.seeds[1][1]
-    assert_equal @rs.replicas[2].port, @connection.seeds[2][1]
-    assert_equal @rs.repl_set_name, @connection.replica_set_name
-    assert @connection.connected?
+    uri = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name}"
+    with_preserved_env_uri(uri) do
+      @connection = ReplSetConnection.new
+      assert !@connection.nil?
+      assert_equal 3, @connection.seeds.length
+      assert_equal @rs.replicas[0].host, @connection.seeds[0][0]
+      assert_equal @rs.replicas[1].host, @connection.seeds[1][0]
+      assert_equal @rs.replicas[2].host, @connection.seeds[2][0]
+      assert_equal @rs.replicas[0].port, @connection.seeds[0][1]
+      assert_equal @rs.replicas[1].port, @connection.seeds[1][1]
+      assert_equal @rs.replicas[2].port, @connection.seeds[2][1]
+      assert_equal @rs.repl_set_name, @connection.replica_set_name
+      assert @connection.connected?
+    end
   end
 
   def test_connect_with_connection_string_in_implicit_mongodb_uri
-    ENV['MONGODB_URI'] = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name}"
-    @connection = Connection.from_uri
-    assert !@connection.nil?
-    assert_equal 3, @connection.seeds.length
-    assert_equal @rs.replicas[0].host, @connection.seeds[0][0]
-    assert_equal @rs.replicas[1].host, @connection.seeds[1][0]
-    assert_equal @rs.replicas[2].host, @connection.seeds[2][0]
-    assert_equal @rs.replicas[0].port, @connection.seeds[0][1]
-    assert_equal @rs.replicas[1].port, @connection.seeds[1][1]
-    assert_equal @rs.replicas[2].port, @connection.seeds[2][1]
-    assert_equal @rs.repl_set_name, @connection.replica_set_name
-    assert @connection.connected?
+    uri = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name}"
+    with_preserved_env_uri(uri) do
+      @connection = Connection.from_uri
+      assert !@connection.nil?
+      assert_equal 3, @connection.seeds.length
+      assert_equal @rs.replicas[0].host, @connection.seeds[0][0]
+      assert_equal @rs.replicas[1].host, @connection.seeds[1][0]
+      assert_equal @rs.replicas[2].host, @connection.seeds[2][0]
+      assert_equal @rs.replicas[0].port, @connection.seeds[0][1]
+      assert_equal @rs.replicas[1].port, @connection.seeds[1][1]
+      assert_equal @rs.replicas[2].port, @connection.seeds[2][1]
+      assert_equal @rs.repl_set_name, @connection.replica_set_name
+      assert @connection.connected?
+    end
   end
 
   def test_connect_with_new_seed_format
@@ -110,21 +114,25 @@ class ConnectionTest < Test::Unit::TestCase
   end
 
   def test_connect_with_full_connection_string_in_env_var
-    ENV['MONGODB_URI'] = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name};safe=true;w=2;fsync=true;slaveok=true"
-    @connection = ReplSetConnection.new
-    assert !@connection.nil?
-    assert @connection.connected?
-    assert_equal 2, @connection.write_concern[:w]
-    assert @connection.write_concern[:fsync]
-    assert @connection.read_pool
+    uri = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name};safe=true;w=2;fsync=true;slaveok=true"
+    with_preserved_env_uri(uri) do
+      @connection = ReplSetConnection.new
+      assert !@connection.nil?
+      assert @connection.connected?
+      assert_equal 2, @connection.write_concern[:w]
+      assert @connection.write_concern[:fsync]
+      assert @connection.read_pool
+    end
   end
 
   def test_connect_options_override_env_var
-    ENV['MONGODB_URI'] = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name};safe=true;w=2;fsync=true;slaveok=true"
-    @connection = ReplSetConnection.new({:safe => {:w => 1}})
-    assert !@connection.nil?
-    assert @connection.connected?
-    assert_equal 1, @connection.write_concern[:w]
+    uri = "mongodb://#{@rs.repl_set_seeds_uri}?replicaset=#{@rs.repl_set_name};safe=true;w=2;fsync=true;slaveok=true"
+    with_preserved_env_uri(uri) do
+      @connection = ReplSetConnection.new({:safe => {:w => 1}})
+      assert !@connection.nil?
+      assert @connection.connected?
+      assert_equal 1, @connection.write_concern[:w]
+    end
   end
 
 end

data/test/replica_set/count_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,7 +21,7 @@ class ReplicaSetCountTest < Test::Unit::TestCase
     @client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :read => :primary_preferred)
     assert @client.primary_pool
     @primary = MongoClient.new(@client.primary_pool.host, @client.primary_pool.port)
-    @db = @client.db(MONGO_TEST_DB)
+    @db = @client.db(TEST_DB)
     @db.drop_collection("test-sets")
     @coll = @db.collection("test-sets")
   end

data/test/replica_set/cursor_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -67,7 +67,7 @@ class ReplicaSetCursorTest < Test::Unit::TestCase
     # Setup ReplicaSet Connection
     @client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :read => read)
 
-    @db = @client.db(MONGO_TEST_DB)
+    @db = @client.db(TEST_DB)
     @db.drop_collection("cursor_tests")
     @coll = @db.collection("cursor_tests")
     insert_docs
@@ -107,24 +107,24 @@ class ReplicaSetCursorTest < Test::Unit::TestCase
     read_opts[:comment] = object_id
 
     # set profiling level to 2 on client and member to which the query will be routed
-    @client.db(MONGO_TEST_DB).profiling_level = :all
+    @client.db(TEST_DB).profiling_level = :all
     @client.secondaries.each do |node|
       node = Mongo::MongoClient.new(node[0], node[1], :slave_ok => true)
-      node.db(MONGO_TEST_DB).profiling_level = :all
+      node.db(TEST_DB).profiling_level = :all
     end
 
     @cursor = @coll.find({}, read_opts)
     @cursor.next
 
     # on client and other members set profiling level to 0
-    @client.db(MONGO_TEST_DB).profiling_level = :off
+    @client.db(TEST_DB).profiling_level = :off
     @client.secondaries.each do |node|
       node = Mongo::MongoClient.new(node[0], node[1], :slave_ok => true)
-      node.db(MONGO_TEST_DB).profiling_level = :off
+      node.db(TEST_DB).profiling_level = :off
     end
     # do a query on system.profile of the reader to see if it was used for the query
-    profiled_queries = @read.db(MONGO_TEST_DB).collection('system.profile').find({
-      'ns' => "#{MONGO_TEST_DB}.cursor_tests", "query.$comment" => object_id })
+    profiled_queries = @read.db(TEST_DB).collection('system.profile').find({
+      'ns' => "#{TEST_DB}.cursor_tests", "query.$comment" => object_id })
 
     assert_equal 1, profiled_queries.count
   end

data/test/replica_set/insert_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -19,7 +19,8 @@ class ReplicaSetInsertTest < Test::Unit::TestCase
   def setup
     ensure_cluster(:rs)
     @client = MongoReplicaSetClient.new @rs.repl_set_seeds
-    @db = @client.db(MONGO_TEST_DB)
+    @version = @client.server_version
+    @db = @client.db(TEST_DB)
     @db.drop_collection("test-sets")
     @coll = @db.collection("test-sets")
   end
@@ -61,4 +62,65 @@ class ReplicaSetInsertTest < Test::Unit::TestCase
     end
   end
 
+  context "Bulk API CollectionView" do
+    setup do
+      setup
+    end
+
+    should "handle error with deferred write concern error - spec Merging Results" do
+      with_write_commands_and_operations(@db.connection) do |wire_version|
+        @coll.remove
+        @coll.ensure_index(BSON::OrderedHash[:a, Mongo::ASCENDING], {:unique => true})
+        bulk = @coll.initialize_ordered_bulk_op
+        bulk.insert({:a => 1})
+        bulk.find({:a => 2}).upsert.update({'$set' => {:a => 2}})
+        bulk.insert({:a => 1})
+        ex = assert_raise BulkWriteError do
+          bulk.execute({:w => 5, :wtimeout => 1})
+        end
+        result = ex.result
+        assert_match_document(
+            {
+                "ok" => 1,
+                "n" => 1,
+                "writeConcernError" => [
+                    {
+                        "errmsg" => /waiting for replication timed out|timed out waiting for slaves|timeout/,
+                        "code" => 64,
+                        "errInfo" => {"wtimeout" => true},
+                        "index" => 0
+                    }
+                ],
+                "code" => 65,
+                "errmsg" => "batch item errors occurred",
+                "nInserted" => 1
+            }, result, "wire_version:#{wire_version}")
+      end
+    end
+
+    should "handle unordered errors with deferred write concern error - spec Merging Results" do # TODO - spec review
+      with_write_commands_and_operations(@db.connection) do |wire_version|
+        @coll.remove
+        @coll.ensure_index(BSON::OrderedHash[:a, Mongo::ASCENDING], {:unique => true})
+        bulk = @coll.initialize_unordered_bulk_op
+        bulk.insert({:a => 1})
+        bulk.find({:a => 2}).upsert.update({'$set' => {:a => 1}})
+        bulk.insert({:a => 3})
+        ex = assert_raise BulkWriteError do
+          bulk.execute({:w => 5, :wtimeout => 1})
+        end
+        result = ex.result # unordered varies, don't use assert_bulk_exception
+        assert_equal(1, result["ok"], "wire_version:#{wire_version}")
+        assert_equal(2, result["n"], "wire_version:#{wire_version}")
+        assert(result["nInserted"] >= 1, "wire_version:#{wire_version}")
+        assert_equal(65, result["code"], "wire_version:#{wire_version}")
+        assert_equal("batch item errors occurred", result["errmsg"], "wire_version:#{wire_version}")
+        assert(result["writeErrors"].size >= 1,  "wire_version:#{wire_version}")
+        assert(result["writeConcernError"].size >= 1, "wire_version:#{wire_version}") if wire_version >= 2
+        assert(@coll.size >= 1, "wire_version:#{wire_version}")
+      end
+    end
+
+  end
+
 end

data/test/replica_set/max_values_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -29,25 +29,31 @@ class MaxValuesTest < Test::Unit::TestCase
     }
   end
 
-  def test_initial_max_sizes
-    assert @client.max_message_size
+  def test_initial_max_and_min_values
     assert @client.max_bson_size
+    assert @client.max_message_size
+    assert @client.max_wire_version
+    assert @client.min_wire_version
   end
 
-  def test_updated_max_sizes_after_node_config_change
+  def test_updated_max_and_min_sizes_after_node_config_change
     @db.stubs(:command).returns(
       @ismaster.merge({'ismaster' => true}),
       @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 1024 * MESSAGE_SIZE_FACTOR}),
-      @ismaster.merge({'secondary' => true, 'maxBsonObjectSize' => 1024})
+      @ismaster.merge({'secondary' => true, 'maxBsonObjectSize' => 1024}),
+      @ismaster.merge({'secondary' => true, 'maxWireVersion' => 0}),
+      @ismaster.merge({'secondary' => true, 'minWireVersion' => 0})
     )
     @client.local_manager.stubs(:refresh_required?).returns(true)
     @client.refresh
 
     assert_equal 1024, @client.max_bson_size
     assert_equal 1024 * MESSAGE_SIZE_FACTOR, @client.max_message_size
+    assert_equal 0, @client.max_wire_version
+    assert_equal 0, @client.min_wire_version
   end
 
-  def test_neither_max_sizes_in_config
+  def test_no_values_in_config
     @db.stubs(:command).returns(
       @ismaster.merge({'ismaster' => true}),
       @ismaster.merge({'secondary' => true}),
@@ -58,6 +64,8 @@ class MaxValuesTest < Test::Unit::TestCase
 
     assert_equal DEFAULT_MAX_BSON_SIZE, @client.max_bson_size
     assert_equal DEFAULT_MAX_BSON_SIZE * MESSAGE_SIZE_FACTOR, @client.max_message_size
+    assert_equal 0, @client.max_wire_version
+    assert_equal 0, @client.min_wire_version
   end
 
   def test_only_bson_size_in_config
@@ -71,20 +79,61 @@ class MaxValuesTest < Test::Unit::TestCase
 
     assert_equal 1024, @client.max_bson_size
     assert_equal 1024 * MESSAGE_SIZE_FACTOR, @client.max_message_size
+    assert_equal 0, @client.max_wire_version
+    assert_equal 0, @client.min_wire_version
   end
 
-  def test_both_sizes_in_config
+  def test_values_in_config
+    #ismaster is called three times on the first node
     @db.stubs(:command).returns(
-      @ismaster.merge({'ismaster' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR, 'maxBsonObjectSize' => 1024}),
-      @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR, 'maxBsonObjectSize' => 1024}),
-      @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR, 'maxBsonObjectSize' => 1024})
+      @ismaster.merge({'ismaster' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR,
+                       'maxBsonObjectSize' => 1024, 'maxWireVersion' => 2, 'minWireVersion' => 1}),
+      @ismaster.merge({'ismaster' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR,
+                       'maxBsonObjectSize' => 1024, 'maxWireVersion' => 2, 'minWireVersion' => 1}),
+      @ismaster.merge({'ismaster' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR,
+                       'maxBsonObjectSize' => 1024, 'maxWireVersion' => 2, 'minWireVersion' => 1}),
+      @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR,
+                       'maxBsonObjectSize' => 1024, 'maxWireVersion' => 2, 'minWireVersion' => 0}),
+      @ismaster.merge({'secondary' => true, 'maxMessageSizeBytes' => 1024 * 2 * MESSAGE_SIZE_FACTOR,
+                       'maxBsonObjectSize' => 1024, 'maxWireVersion' => 1, 'minWireVersion' => 0})
     )
     @client.local_manager.stubs(:refresh_required?).returns(true)
     @client.refresh
 
     assert_equal 1024, @client.max_bson_size
     assert_equal 1024 * 2 * MESSAGE_SIZE_FACTOR, @client.max_message_size
+    assert_equal 1, @client.max_wire_version # minimum of all max_wire_version
+    assert_equal 1, @client.min_wire_version # maximum of all min_wire_version
   end
 
+  def test_wire_version_not_in_range
+    min_wire_version, max_wire_version = [Mongo::MongoClient::MIN_WIRE_VERSION-1, Mongo::MongoClient::MIN_WIRE_VERSION-1]
+    #ismaster is called three times on the first node
+    @db.stubs(:command).returns(
+      @ismaster.merge({'ismaster' => true, 'maxWireVersion' => max_wire_version, 'minWireVersion' => min_wire_version}),
+      @ismaster.merge({'ismaster' => true, 'maxWireVersion' => max_wire_version, 'minWireVersion' => min_wire_version}),
+      @ismaster.merge({'ismaster' => true, 'maxWireVersion' => max_wire_version, 'minWireVersion' => min_wire_version}),
+      @ismaster.merge({'secondary' => true, 'maxWireVersion' => max_wire_version, 'minWireVersion' => min_wire_version}),
+      @ismaster.merge({'secondary' => true, 'maxWireVersion' => max_wire_version, 'minWireVersion' => min_wire_version})
+    )
+    @client.local_manager.stubs(:refresh_required?).returns(true)
+    assert_raises Mongo::ConnectionFailure do
+      @client.refresh
+    end
+  end
+
+  def test_use_write_command
+    with_write_commands(@client) do
+      assert_true @client.use_write_command?({:w => 1})
+      assert_false @client.use_write_command?({:w => 0})
+    end
+    with_write_operations(@client) do
+      assert_false @client.use_write_command?({:w => 1})
+      assert_false @client.use_write_command?({:w => 0})
+    end
+    @client.local_manager.primary_pool.node.expects(:wire_version_feature?).at_least_once.returns(true)
+    assert_true @client.use_write_command?({:w => 1})
+    assert_false @client.use_write_command?({:w => 0})
+  end
 end
 

data/test/replica_set/pinning_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -18,7 +18,7 @@ class ReplicaSetPinningTest < Test::Unit::TestCase
   def setup
     ensure_cluster(:rs)
     @client = MongoReplicaSetClient.new(@rs.repl_set_seeds, :name => @rs.repl_set_name)
-    @db = @client.db(MONGO_TEST_DB)
+    @db = @client.db(TEST_DB)
     @coll = @db.collection("test-sets")
     @coll.insert({:a => 1})
   end

data/test/replica_set/query_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -19,7 +19,7 @@ class ReplicaSetQueryTest < Test::Unit::TestCase
   def setup
     ensure_cluster(:rs)
     @client = MongoReplicaSetClient.new @rs.repl_set_seeds
-    @db = @client.db(MONGO_TEST_DB)
+    @db = @client.db(TEST_DB)
     @db.drop_collection("test-sets")
     @coll = @db.collection("test-sets")
   end

data/test/replica_set/read_preference_test.rb

@@ -1,4 +1,4 @@
-# Copyright (C) 2013 10gen Inc.
+# Copyright (C) 2009-2013 MongoDB, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,7 +22,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
     # Insert data
     primary = @rs.primary
     conn = Connection.new(primary.host, primary.port)
-    db = conn.db(MONGO_TEST_DB)
+    db = conn.db(TEST_DB)
     coll = db.collection("test-sets")
     coll.save({:a => 20}, {:w => 2})
   end
@@ -103,7 +103,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
 
     # Test that reads are going to the right members
     assert_raise_error ConnectionFailure do
-      @primary[MONGO_TEST_DB]['test-sets'].find_one
+      @primary[TEST_DB]['test-sets'].find_one
     end
     assert_query_route(@primary_preferred, @secondary_direct)
     assert_query_route(@secondary, @secondary_direct)
@@ -142,7 +142,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
     assert_query_route(@primary, @primary_direct)
     assert_query_route(@primary_preferred, @primary_direct)
     assert_raise_error ConnectionFailure do
-      @secondary[MONGO_TEST_DB]['test-sets'].find_one
+      @secondary[TEST_DB]['test-sets'].find_one
     end
     assert_query_route(@secondary_preferred, @primary_direct)
 
@@ -166,7 +166,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
 
   def test_write_lots_of_data
     @conn = make_connection(:secondary_preferred)
-    @db = @conn[MONGO_TEST_DB]
+    @db = @conn[TEST_DB]
     @coll = @db.collection("test-sets", {:w => 2})
 
     6000.times do |n|
@@ -206,7 +206,7 @@ class ReadPreferenceTest < Test::Unit::TestCase
     #puts "#{test_connection.read_pool.port} #{expected_target.read_pool.port}"
     queries_before = query_count(expected_target)
     assert_nothing_raised do
-      test_connection['MONGO_TEST_DB']['test-sets'].find_one
+      test_connection[TEST_DB]['test-sets'].find_one
     end
     queries_after = query_count(expected_target)
     assert_equal 1, queries_after - queries_before