implement column adding with foreign keys on sqlite

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()