def test_get_api_engine(self):
with mock.patch.object(
api_db_api, 'get_engine', return_value='engine',
) as mock_get_engine:
engine = migration._get_engine('api')
self.assertEqual('engine', engine)
mock_get_engine.assert_called_once_with()
def test_get_main_engine(self):
with mock.patch.object(
main_db_api, 'get_engine', return_value='engine',
) as mock_get_engine:
engine = migration._get_engine()
self.assertEqual('engine', engine)
mock_get_engine.assert_called_once_with(context=None)