def test_uri_input(self):
test = connection.ConnectionPool()
yield test.disconnect()
test = connection.ConnectionPool("mongodb://127.0.0.1/dbname")
yield test.disconnect()
test = connection.ConnectionPool(u"mongodb://127.0.0.1/dbname")
yield test.disconnect()
self.assertRaises(AssertionError, connection.ConnectionPool, object)
self.assertRaises(AssertionError, connection.ConnectionPool, 1)
def setUp(self):
self.conn = connection.ConnectionPool("mongodb://127.0.0.1/dbname")
self.db = self.conn['dbname']
self.gfs = gridfs.GridFS(self.db)
for n in range(0, 10):
data = "Hello" + str(n)
yield self.gfs.put(data.encode('utf-8'), filename="world")
def test_ByURI(self):
uri = "mongodb://{0}:{1}@{2}:{3}/{4}?authMechanism=MONGODB-CR".format(
self.login1, self.password1, mongo_host, mongo_port, self.db1)
conn = connection.ConnectionPool(uri)
try:
yield conn[self.db1][self.coll].insert_one({'x': 42})
finally:
yield conn.disconnect()
def test_ConnectionPool(self):
conn = connection.ConnectionPool(mongo_uri)
try:
yield conn[self.db1].authenticate(self.login1,
self.password1,
mechanism="MONGODB-CR")
yield conn[self.db1][self.coll].insert_one({'x': 42})
finally:
yield conn.disconnect()
def test_Fail(self):
conn = connection.ConnectionPool(mongo_uri)
try:
yield conn[self.db1].authenticate(self.login1,
self.password1 + 'x',
mechanism="MONGODB-CR")
query = conn[self.db1][self.coll].insert_one({'x': 42})
yield self.assertFailure(query, OperationFailure)
finally:
yield conn.disconnect()
def test_AuthConnectionPoolUriAuthSource(self):
def authenticate(database, username, password, mechanism):
self.assertEqual(database, self.db2)
with patch('txmongo.connection.ConnectionPool.authenticate',
side_effect=authenticate):
conn = connection.ConnectionPool(
"mongodb://{0}:{1}@{2}:{3}/{4}?authSource={5}".format(
self.login1, self.password1, mongo_host, mongo_port,
self.db1, self.db2))
yield conn.disconnect()
def test_AuthConnectionPoolUri(self):
pool_size = 5
conn = connection.ConnectionPool(
"mongodb://{0}:{1}@{2}:{3}/{4}".format(self.login1, self.password1,
mongo_host, mongo_port,
self.db1))
db = conn.get_default_database()
coll = db[self.coll]
n = pool_size + 1
try:
yield defer.gatherResults(
[coll.insert({'x': 42}) for _ in range(n)])
cnt = yield coll.count()
self.assertEqual(cnt, n)
finally:
yield conn.disconnect()
def __get_connection(self, pool_size=1):
return connection.ConnectionPool(mongo_uri, pool_size)
def setUp(self):
self.named_conn = connection.ConnectionPool(
"mongodb://127.0.0.1/dbname")
self.unnamed_conn = connection.ConnectionPool("127.0.0.1")
def setUp(self):
self.conn = connection.ConnectionPool()
self.db = self.conn.db
self.coll = self.db.coll
yield self.coll.insert({'x': 42})
yield self.assert_coll_count(1)
