From 1af477e661b7e68730598fab70116fa1e15cfb1b Mon Sep 17 00:00:00 2001 From: Jan Dittberner Date: Thu, 26 May 2011 21:43:27 +0200 Subject: fix unit test for adding new columns with foreign keys --- migrate/tests/changeset/test_changeset.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'migrate/tests/changeset/test_changeset.py') 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): -- cgit v1.2.1