def test_disconnect(self):
Connection.disconnect()
connection_name_list = Connection.get_connection_name_list()
connection_name = Connection.get_default_connection_name()
self.assertEqual(['con'], connection_name_list)
self.assertEqual(connection_name, 'con')
def test_connection(self):
# test connect(), get_connection(), get_database()...
Connection.connect('monguo_test1')
connection = Connection.get_connection()
database = Connection.get_database()
db_name = Connection.get_default_database_name()
connection_name = Connection.get_default_connection_name()
self.assertIsInstance(connection, MotorClient)
self.assertIsInstance(database, MotorDatabase)
self.assertEqual(db_name, 'monguo_test1')
self.assertEqual(connection_name, Connection.DEFAULT_CONNECTION_NAME)
Connection.connect('monguo_test2', 'con')
db_name = Connection.get_default_database_name()
connection_name = Connection.get_default_connection_name()
self.assertEqual(db_name, 'monguo_test2')
self.assertEqual(connection_name, 'con')
# test get_connection_name_list()
connection_name_list = Connection.get_connection_name_list()
self.assertIn(Connection.DEFAULT_CONNECTION_NAME, connection_name_list)
self.assertIn('con', connection_name_list)
self.assertEqual(sorted([Connection.DEFAULT_CONNECTION_NAME, 'con']),
sorted(connection_name_list))
# test switch_connection()
Connection.switch_connection(Connection.DEFAULT_CONNECTION_NAME)
connection_name = Connection.get_default_connection_name()
self.assertEqual(connection_name, Connection.DEFAULT_CONNECTION_NAME)
# test switch_database()
Connection.switch_database('monguo_test3')
db_name = Connection.get_default_database_name()
self.assertEqual(db_name, 'monguo_test3')