def test_get_api_engine(self): with mock.patch.object(db_api, 'get_api_engine', return_value='api_engine') as mock_get_engine: engine = migration.get_engine('api') self.assertEqual('api_engine', engine) mock_get_engine.assert_called_once_with()
def test_get_main_engine(self): with mock.patch.object(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)
def test_get_api_engine(self): with mock.patch.object(db_api, "get_api_engine", return_value="api_engine") as mock_get_engine: engine = migration.get_engine("api") self.assertEqual("api_engine", engine) mock_get_engine.assert_called_once_with()