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)
