def test_initial_with_extension_version_None(self): """When performing a default migration, also migrate extensions.""" migration_helpers.sync_database_to_version(extension=None, version=None) for table in INITIAL_EXTENSION_TABLE_STRUCTURE: self.assertTableColumns(table, INITIAL_EXTENSION_TABLE_STRUCTURE[table])
def main(): version = CONF.command.version extension = CONF.command.extension populate = CONF.command.populate if populate: extension = 'initial_data' migration_helpers.sync_database_to_version(extension, version)
def setup_database(): db = dirs.tmp('test.db') pristine = dirs.tmp('test.db.pristine') if os.path.exists(db): os.unlink(db) if not os.path.exists(pristine): migration.db_sync(sql.get_engine(), migration_helpers.find_migrate_repo()) migration_helpers.sync_database_to_version(extension='revoke') shutil.copyfile(db, pristine) else: shutil.copyfile(pristine, db)
def _setup_database(extensions=None): if CONF.database.connection != tests.IN_MEM_DB_CONN_STRING: db = tests.dirs.tmp('test.db') pristine = tests.dirs.tmp('test.db.pristine') if os.path.exists(db): os.unlink(db) if not os.path.exists(pristine): migration.db_sync(sql.get_engine(), migration_helpers.find_migrate_repo()) for extension in (extensions or []): migration_helpers.sync_database_to_version(extension=extension) shutil.copyfile(db, pristine) else: shutil.copyfile(pristine, db)
def main(): version = CONF.command.version extension = CONF.command.extension migration_helpers.sync_database_to_version(extension, version)
def test_initial_with_extension_version_max(self): """When migrating to max version, do not migrate extensions.""" migration_helpers.sync_database_to_version(extension=None, version=self.max_version) for table in INITIAL_EXTENSION_TABLE_STRUCTURE: self.assertTableDoesNotExist(table)
def main(): assert_not_extension(CONF.command.extension) version = CONF.command.version migration_helpers.sync_database_to_version(version)