def test_configure(self):
"""Connection.configure"""
test1 = 'mongodb://localhost/test1'
test2 = 'mongodb://localhost/test2'
prefix2 = 'units_'
config = {
'test1': test1,
'test2': {
'uri': test2,
'prefix': prefix2,
},
}
try:
connection.configure(config)
con1 = connection.get('test1')
self.assertEqual(con1.prefix, "", "con1 prefix is incorrect")
self.assertEqual(con1.database.name, "test1", "con1 database is incorrect")
con2 = connection.get('test2')
self.assertEqual(con2.prefix, prefix2, "con2 prefix is incorrect")
self.assertEqual(con2.database.name, "test2", "con2 database is incorrect")
self.assertRaises(
errors.ConfigError, connection.configure, {'broken': ''})
self.assertRaises(
errors.ConfigError, connection.configure, {'broken': {'uri': ''}})
finally:
for name, con in six.iteritems(connection.connections):
con.close()
def test_get(self):
"""connection.get"""
con = connection.Connection(uri)
try:
name = "test"
connection.connections[name] = con
self.assertEqual(connection.get(name), con, "returned connection is incorrect")
finally:
con.close()
connection.connections = {}
def test_get(self):
"""connection.get"""
con = connection.Connection(uri)
try:
name = 'test'
connection.connections[name] = con
self.assertEqual(
connection.get(name), con, "returned connection is incorrect")
finally:
con.close()
connection.connections = {}
def test_configure(self):
"""Connection.configure"""
test1 = "mongodb://localhost/test1"
test2 = "mongodb://localhost/test2"
prefix2 = "units_"
config = {"test1": test1, "test2": {"uri": test2, "prefix": prefix2}}
try:
connection.configure(config)
con1 = connection.get("test1")
self.assertTrue(con1.client.alive(), "con1 is not alive")
self.assertEqual(con1.prefix, "", "con1 prefix is incorrect")
self.assertEqual(con1.database.name, "test1", "con1 database is incorrect")
con2 = connection.get("test2")
self.assertTrue(con2.client.alive(), "con2 is not alive")
self.assertEqual(con2.prefix, prefix2, "con2 prefix is incorrect")
self.assertEqual(con2.database.name, "test2", "con2 database is incorrect")
self.assertRaises(errors.ConfigError, connection.configure, {"broken": ""})
self.assertRaises(errors.ConfigError, connection.configure, {"broken": {"uri": ""}})
finally:
for name, con in connection.connections.iteritems():
con.close()
