diff options
| author | Jan Dittberner <jan.dittberner@googlemail.com> | 2011-05-26 21:43:27 +0200 |
|---|---|---|
| committer | Jan Dittberner <jan.dittberner@googlemail.com> | 2011-05-26 21:43:27 +0200 |
| commit | 1af477e661b7e68730598fab70116fa1e15cfb1b (patch) | |
| tree | ff279f2c8a64b6389516ff73340c700fdb8646d1 /migrate/tests | |
| parent | f80e91f05edd8d27b9e959cb1769d8f6bea9e42a (diff) | |
| download | sqalchemy-migrate-1af477e661b7e68730598fab70116fa1e15cfb1b.tar.gz | |
fix unit test for adding new columns with foreign keys
Diffstat (limited to 'migrate/tests')
| -rw-r--r-- | migrate/tests/changeset/test_changeset.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/migrate/tests/changeset/test_changeset.py b/migrate/tests/changeset/test_changeset.py index d1b34cd..89225a4 100644 --- a/migrate/tests/changeset/test_changeset.py +++ b/migrate/tests/changeset/test_changeset.py @@ -169,6 +169,8 @@ class TestAddDropColumn(fixture.DB): # create column with fk col = Column('data', Integer, ForeignKey(reftable.c.id)) + if SQLA_07: + self.table.append_column(col) col.create(self.table) # check if constraint is added @@ -180,7 +182,12 @@ class TestAddDropColumn(fixture.DB): # TODO: test on db level if constraints work - self.assertEqual(reftable.c.id.name, col.foreign_keys[0].column.name) + if SQLA_07: + self.assertEqual(reftable.c.id.name, + list(col.foreign_keys)[0].column.name) + else: + self.assertEqual(reftable.c.id.name, + col.foreign_keys[0].column.name) col.drop(self.table) if self.engine.has_table(reftable.name): |