def test_get_db(mocker):
db_name = "test_db"
coll_name = "test_coll"
mongo_client = mongomock.MongoClient()
conn = MongoConnection(db_name, coll_name)
mocker.patch.object(mongo_client, 'get_database', return_value=True)
mocker.patch.object(conn, 'raise_exception')
conn.client = mongo_client
conn.get_db(db_name)
conn.client.get_database.assert_called_with(db_name)
assert conn.client.get_database.call_count == 2
conn.raise_exception.assert_not_called()
def test_get_db_false(mocker):
db_name = "test_db"
coll_name = "test_coll"
mongo_client = mongomock.MongoClient()
conn = MongoConnection(db_name, coll_name)
conn.client = mongo_client
message = 'Database test_db does not exist. Please check your configuration file parameters and try again.'
mocker.patch.object(mongo_client, 'get_database', return_value=False)
mocker.patch.object(conn, 'raise_exception')
conn.get_db(db_name)
conn.client.get_database.assert_called_once_with(db_name)
conn.raise_exception.assert_called_once_with(message)