summaryrefslogtreecommitdiff
path: root/migrate/tests/changeset/test_changeset.py
diff options
context:
space:
mode:
Diffstat (limited to 'migrate/tests/changeset/test_changeset.py')
-rw-r--r--migrate/tests/changeset/test_changeset.py24
1 files changed, 10 insertions, 14 deletions
diff --git a/migrate/tests/changeset/test_changeset.py b/migrate/tests/changeset/test_changeset.py
index b1a4aba..8eff822 100644
--- a/migrate/tests/changeset/test_changeset.py
+++ b/migrate/tests/changeset/test_changeset.py
@@ -166,23 +166,19 @@ class TestAddDropColumn(fixture.DB):
# create column with fk
col = Column('data', Integer, ForeignKey(reftable.c.id))
- if self.url.startswith('sqlite'):
- self.assertRaises(exceptions.NotSupportedError,
- col.create, self.table)
- else:
- col.create(self.table)
+ col.create(self.table)
- # check if constraint is added
- for cons in self.table.constraints:
- if isinstance(cons, sqlalchemy.schema.ForeignKeyConstraint):
- break
- else:
- self.fail('No constraint found')
+ # check if constraint is added
+ for cons in self.table.constraints:
+ if isinstance(cons, sqlalchemy.schema.ForeignKeyConstraint):
+ break
+ else:
+ self.fail('No constraint found')
- # TODO: test on db level if constraints work
+ # TODO: test on db level if constraints work
- self.assertEqual(reftable.c.id.name, col.foreign_keys[0].column.name)
- col.drop(self.table)
+ self.assertEqual(reftable.c.id.name, col.foreign_keys[0].column.name)
+ col.drop(self.table)
if self.engine.has_table(reftable.name):
reftable.drop()