Example #1
0
 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()
Example #2
0
 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_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()
 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()