diff options
Diffstat (limited to 'tests/test_autogen_render.py')
| -rw-r--r-- | tests/test_autogen_render.py | 31 |
1 files changed, 30 insertions, 1 deletions
diff --git a/tests/test_autogen_render.py b/tests/test_autogen_render.py index 1e25e6b..e020cec 100644 --- a/tests/test_autogen_render.py +++ b/tests/test_autogen_render.py @@ -254,6 +254,36 @@ unique=False, """ "schema='CamelSchema', type_='unique')" ) + def test_add_fk_constraint(self): + m = MetaData() + Table('a', m, Column('id', Integer, primary_key=True)) + b = Table('b', m, Column('a_id', Integer, ForeignKey('a.id'))) + fk = ForeignKeyConstraint(['a_id'], ['a.id'], name='fk_a_id') + b.append_constraint(fk) + eq_ignore_whitespace( + autogenerate.render._add_fk_constraint(fk, self.autogen_context), + "op.create_foreign_key('fk_a_id', 'b', 'a', ['a_id'], ['id'])" + ) + + def test_add_fk_constraint_schema(self): + m = MetaData() + Table( + 'a', m, Column('id', Integer, primary_key=True), + schema="CamelSchemaTwo") + b = Table( + 'b', m, Column('a_id', Integer, ForeignKey('a.id')), + schema="CamelSchemaOne") + fk = ForeignKeyConstraint( + ["a_id"], + ["CamelSchemaTwo.a.id"], name='fk_a_id') + b.append_constraint(fk) + eq_ignore_whitespace( + autogenerate.render._add_fk_constraint(fk, self.autogen_context), + "op.create_foreign_key('fk_a_id', 'b', 'a', ['a_id'], ['id']," + " source_schema='CamelSchemaOne', " + "referent_schema='CamelSchemaTwo')" + ) + def test_drop_fk_constraint(self): m = MetaData() Table('a', m, Column('id', Integer, primary_key=True)) @@ -267,7 +297,6 @@ unique=False, """ def test_drop_fk_constraint_schema(self): m = MetaData() - m = MetaData() Table( 'a', m, Column('id', Integer, primary_key=True), schema="CamelSchemaTwo") |