summaryrefslogtreecommitdiff
path: root/tests/test_postgresql.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2017-02-23 12:57:40 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2017-02-23 13:37:56 -0500
commit7d5f6eabeccb9328c1c21dd0cdf9ff1f2f85836d (patch)
tree809edf7f261e6249f32750fd3f8da2d4ff4a0384 /tests/test_postgresql.py
parent3f070ea17fa9f5618a299637421d73be6e98d2e5 (diff)
downloadalembic-7d5f6eabeccb9328c1c21dd0cdf9ff1f2f85836d.tar.gz
Allow target_metadata to be a sequence
The :paramref:`.EnvironmentContext.configure.target_metadata` parameter may now be optionally specified as a sequence of :class:`.MetaData` objects instead of a single :class:`.MetaData` object. The autogenerate process will process the sequence of :class:`.MetaData` objects in order. Change-Id: I6485c05d68219ff7af1611b34550487d316e0242 Fixes: #38
Diffstat (limited to 'tests/test_postgresql.py')
-rw-r--r--tests/test_postgresql.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py
index e8608b7..2ceeaa9 100644
--- a/tests/test_postgresql.py
+++ b/tests/test_postgresql.py
@@ -517,6 +517,8 @@ class PostgresqlDetectSerialTest(TestBase):
def _expect_default(self, c_expected, col, seq=None):
Table('t', self.metadata, col)
+ self.autogen_context.metadata = self.metadata
+
if seq:
seq._set_metadata(self.metadata)
self.metadata.create_all(config.db)
@@ -526,7 +528,7 @@ class PostgresqlDetectSerialTest(TestBase):
uo = ops.UpgradeOps(ops=[])
_compare_tables(
set([(None, 't')]), set([]),
- insp, self.metadata, uo, self.autogen_context)
+ insp, uo, self.autogen_context)
diffs = uo.as_diffs()
tab = diffs[0][1]
@@ -538,9 +540,10 @@ class PostgresqlDetectSerialTest(TestBase):
uo = ops.UpgradeOps(ops=[])
m2 = MetaData()
Table('t', m2, Column('x', BigInteger()))
+ self.autogen_context.metadata = m2
_compare_tables(
set([(None, 't')]), set([(None, 't')]),
- insp, m2, uo, self.autogen_context)
+ insp, uo, self.autogen_context)
diffs = uo.as_diffs()
server_default = diffs[0][0][4]['existing_server_default']
eq_(_render_server_default_for_compare(