def test_db_version_raise_not_controlled_error_no_tables(self, mock_vc): with mock.patch.object(sqlalchemy, 'MetaData') as mock_meta: self.mock_api_db_version.side_effect = ( migrate_exception.DatabaseNotControlledError('oups'), self.init_version) my_meta = mock.MagicMock() my_meta.tables = {} mock_meta.return_value = my_meta migration.db_version(self.engine, self.path, self.init_version) mock_vc.assert_called_once_with(self.engine, self.return_value1, self.init_version)
def test_db_version_raise_not_controlled_error_first(self): with mock.patch.object(migration, 'db_version_control') as mock_ver: self.mock_api_db_version.side_effect = [ migrate_exception.DatabaseNotControlledError('oups'), self.test_version] ret_val = migration.db_version(self.path, self.init_version) self.assertEqual(ret_val, self.test_version) mock_ver.assert_called_once_with(self.path, self.init_version)
def test_db_version_return(self): ret_val = migration.db_version(self.engine, self.path, self.init_version) self.assertEqual(ret_val, self.test_version)
def version(self): return migration.db_version(self.engine, self.repository, init_version=self.init_version)
def version(self): return migration.db_version( self.engine, self.repository, init_version=self.init_version)