1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#!coding: utf-8
from alembic.script import ScriptDirectory
from alembic.environment import EnvironmentContext
from alembic.migration import MigrationContext
from alembic.testing.fixtures import TestBase
from alembic.testing.mock import Mock, call
from alembic.testing.env import _no_sql_testing_config, \
staging_env, clear_staging_env
from alembic.testing import eq_, is_
class EnvironmentTest(TestBase):
def setUp(self):
staging_env()
self.cfg = _no_sql_testing_config()
def tearDown(self):
clear_staging_env()
def _fixture(self, **kw):
script = ScriptDirectory.from_config(self.cfg)
env = EnvironmentContext(
self.cfg,
script,
**kw
)
return env
def test_x_arg(self):
env = self._fixture()
self.cfg.cmd_opts = Mock(x="y=5")
eq_(
env.get_x_argument(),
"y=5"
)
def test_x_arg_asdict(self):
env = self._fixture()
self.cfg.cmd_opts = Mock(x=["y=5"])
eq_(
env.get_x_argument(as_dictionary=True),
{"y": "5"}
)
def test_x_arg_no_opts(self):
env = self._fixture()
eq_(
env.get_x_argument(),
[]
)
def test_x_arg_no_opts_asdict(self):
env = self._fixture()
eq_(
env.get_x_argument(as_dictionary=True),
{}
)
def test_tag_arg(self):
env = self._fixture(tag="x")
eq_(
env.get_tag_argument(),
"x"
)
def test_migration_context_has_config(self):
env = self._fixture()
env.configure(url="sqlite://")
ctx = env._migration_context
is_(ctx.config, self.cfg)
ctx = MigrationContext(ctx.dialect, None, {})
is_(ctx.config, None)
|