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)