PYTHON-472 - Fix raw BSON tests when run with authentication.

test/test_raw_bson.py

@@ -1,13 +1,11 @@
 import datetime
 import uuid
 
-import pymongo
-
 from bson import BSON
 from bson.binary import JAVA_LEGACY
 from bson.codec_options import CodecOptions
 from bson.raw_bson import RawBSONDocument
-from test import client_context, unittest, pair
+from test import client_context, unittest
 
 
 class TestRawBSONDocument(unittest.TestCase):
@@ -22,9 +20,13 @@ class TestRawBSONDocument(unittest.TestCase):
     )
     document = RawBSONDocument(bson_string)
 
+    @classmethod
+    def setUpClass(cls):
+        cls.client = client_context.rs_or_standalone_client
+
     def tearDown(self):
         if client_context.connected:
-            client_context.client.pymongo_test.test_raw.drop()
+            self.client.pymongo_test.test_raw.drop()
 
     def test_decode(self):
         self.assertEqual('Sherlock', self.document['name'])
@@ -37,9 +39,11 @@ def test_raw(self):
 
     @client_context.require_connection
     def test_round_trip(self):
-        client = pymongo.MongoClient(pair, document_class=RawBSONDocument)
-        client.pymongo_test.test_raw.insert_one(self.document)
-        result = client.pymongo_test.test_raw.find_one(self.document['_id'])
+        db = self.client.get_database(
+            'pymongo_test',
+            codec_options=CodecOptions(document_class=RawBSONDocument))
+        db.test_raw.insert_one(self.document)
+        result = db.test_raw.find_one(self.document['_id'])
         self.assertIsInstance(result, RawBSONDocument)
         self.assertEqual(dict(self.document.items()), dict(result.items()))
 
@@ -65,7 +69,7 @@ def test_round_trip_codec_options(self):
             'date': datetime.datetime(2015, 6, 3, 18, 40, 50, 826000),
             '_id': uuid.UUID('026fab8f-975f-4965-9fbf-85ad874c60ff')
         }
-        db = pymongo.MongoClient(pair).pymongo_test
+        db = self.client.pymongo_test
         coll = db.get_collection(
             'test_raw',
             codec_options=CodecOptions(uuid_representation=JAVA_LEGACY))
@@ -80,7 +84,7 @@ def test_round_trip_codec_options(self):
     @client_context.require_connection
     def test_raw_bson_document_embedded(self):
         doc = {'embedded': self.document}
-        db = client_context.client.pymongo_test
+        db = self.client.pymongo_test
         db.test_raw.insert_one(doc)
         result = db.test_raw.find_one()
         self.assertEqual(BSON(self.document.raw).decode(), result['embedded'])