def test_revisions(self): alembic_config = get_alembic_config() directory = ScriptDirectory.from_config(alembic_config) versions_path = directory.versions files = os.listdir(versions_path) revisions = set([]) down_revisions = set([]) for f in files: if not f.endswith(".py"): # Skipping README and all .pyc files if there are any. continue module_path = os.path.join(versions_path, f) module = imp.load_source('module', six.text_type(module_path)) revision_id = module.revision self.assertFalse(revision_id in revisions, "Found two revisions with id %s" % revision_id) revisions.add(revision_id) down_revision_id = module.revision self.assertFalse( down_revision_id in down_revisions, "Found two revisions with down_revision_id %s" % revision_id) down_revisions.add(down_revision_id)
def test_revisions(self): alembic_config = get_alembic_config() directory = ScriptDirectory.from_config(alembic_config) versions_path = directory.versions files = os.listdir(versions_path) revisions = set([]) down_revisions = set([]) for f in files: if not f.endswith(".py"): # Skipping README and all .pyc files if there are any. continue module_path = os.path.join(versions_path, f) module = imp.load_source("module", six.text_type(module_path)) revision_id = module.revision self.assertFalse(revision_id in revisions, "Found two revisions with id %s" % revision_id) revisions.add(revision_id) down_revision_id = module.revision self.assertFalse( down_revision_id in down_revisions, "Found two revisions with down_revision_id %s" % revision_id ) down_revisions.add(down_revision_id)
def setup_db(self): self.db_name = "storyboard_test_db_%s" % uuid.uuid4() self.db_name = self.db_name.replace("-", "_") # The engine w/o db name engine = sqlalchemy.create_engine(self.test_connection) engine.execute("CREATE DATABASE %s" % self.db_name) alembic_config = get_alembic_config() alembic_config.storyboard_config = CONF CONF.set_override("connection", self.test_connection + "/%s" % self.db_name, group="database") command.upgrade(alembic_config, "head") self.addCleanup(self._drop_db)
def setup_db(self): self.db_name = "storyboard_test_db_%s" % uuid.uuid4() self.db_name = self.db_name.replace("-", "_") # The engine w/o db name engine = sqlalchemy.create_engine( self.test_connection) engine.execute("CREATE DATABASE %s" % self.db_name) alembic_config = get_alembic_config() alembic_config.storyboard_config = CONF CONF.set_override( "connection", self.test_connection + "/%s" % self.db_name, group="database") command.upgrade(alembic_config, "head") self.addCleanup(self._drop_db)
def setup_db(self): self.db_name = "storyboard_test_db_%s" % uuid.uuid4() self.db_name = self.db_name.replace("-", "_") dburi = self.test_connection + "/%s" % self.db_name if dburi.startswith('mysql+pymysql://'): dburi += "?charset=utf8mb4" CONF.set_override("connection", dburi, group="database") self._full_db_name = self.test_connection + '/' + self.db_name LOG.info('using database %s', CONF.database.connection) if self.test_connection.startswith('sqlite://'): self.using_sqlite = True else: self.using_sqlite = False # The engine w/o db name engine = sqlalchemy.create_engine(self.test_connection) engine.execute("CREATE DATABASE %s" % self.db_name) alembic_config = get_alembic_config() alembic_config.storyboard_config = CONF command.upgrade(alembic_config, "head") self.addCleanup(self._drop_db)
def setup_db(cls): cls.db_name = "storyboard_test_db_%s" % uuid.uuid4() cls.db_name = cls.db_name.replace("-", "_") dburi = cls.test_connection + "/%s" % cls.db_name if dburi.startswith('mysql+pymysql://'): dburi += "?charset=utf8mb4" CONF.set_override("connection", dburi, group="database") cls._full_db_name = cls.test_connection + '/' + cls.db_name LOG.info('using database %s', CONF.database.connection) if cls.test_connection.startswith('sqlite://'): cls.using_sqlite = True else: cls.using_sqlite = False # The engine w/o db name engine = sqlalchemy.create_engine( cls.test_connection) engine.execute("CREATE DATABASE %s" % cls.db_name) alembic_config = get_alembic_config() alembic_config.storyboard_config = CONF command.upgrade(alembic_config, "head") cls._engine = sqlalchemy.create_engine(dburi)