def test_version_locations(self, separator, string_value, expected_result):
cfg = config.Config()
if separator is not None:
cfg.set_main_option(
"version_path_separator",
separator,
)
cfg.set_main_option("script_location", tempfile.gettempdir())
cfg.set_main_option("version_locations", string_value)
if isinstance(expected_result, ValueError):
with expect_raises_message(ValueError, expected_result.args[0]):
ScriptDirectory.from_config(cfg)
else:
s = ScriptDirectory.from_config(cfg)
eq_(s.version_locations, expected_result)
def test_error_on_passing_engine(self):
env = self._fixture()
engine = _sqlite_file_db()
a_rev = "arev"
env.script.generate_revision(a_rev, "revision a", refresh=True)
write_script(
env.script,
a_rev,
"""\
"Rev A"
revision = '%s'
down_revision = None
from alembic import op
def upgrade():
pass
def downgrade():
pass
"""
% a_rev,
)
migration_fn = mock.MagicMock()
def upgrade(rev, context):
migration_fn(rev, context)
return env.script._upgrade_revs(a_rev, rev)
with expect_raises_message(
util.CommandError,
r"'connection' argument to configure\(\) is "
r"expected to be a sqlalchemy.engine.Connection ",
):
env.configure(
connection=engine, fn=upgrade, transactional_ddl=False
)