summaryrefslogtreecommitdiff
path: root/test/engine/test_reflection.py
diff options
context:
space:
mode:
authorijl <uijllji@gmail.com>2013-10-13 17:13:28 -0400
committerijl <uijllji@gmail.com>2013-10-13 17:13:28 -0400
commit52cfa74e094c32910fa1076375f1ae0c9795d45d (patch)
treef3408fa65519d4f17b16421fc6c2a8038b4c4d47 /test/engine/test_reflection.py
parent9d952e0a11709fe35ada2635a79043ca0fc7ffbf (diff)
downloadsqlalchemy-52cfa74e094c32910fa1076375f1ae0c9795d45d.tar.gz
#2183: Metadata.reflect() foreign keys include options when the dialect exposes it
Diffstat (limited to 'test/engine/test_reflection.py')
-rw-r--r--test/engine/test_reflection.py41
1 files changed, 41 insertions, 0 deletions
diff --git a/test/engine/test_reflection.py b/test/engine/test_reflection.py
index 21c050915..3b23bf31a 100644
--- a/test/engine/test_reflection.py
+++ b/test/engine/test_reflection.py
@@ -602,6 +602,47 @@ class ReflectionTest(fixtures.TestBase, ComparesTables):
is a2.c.user_id
assert u2.join(a2).onclause.compare(u2.c.id == a2.c.user_id)
+ @testing.only_on(['postgresql', 'mysql'])
+ @testing.provide_metadata
+ def test_fk_options(self):
+ """test that foreign key reflection includes options (on
+ backends with {dialect}.get_foreign_keys() support)"""
+
+ # fk whose attributes we're testing on reflection
+ addresses_user_id_fkey = sa.ForeignKey(
+ 'users.id',
+ name = 'addresses_user_id_fkey',
+ match='FULL',
+ onupdate='RESTRICT',
+ ondelete='RESTRICT',
+ deferrable=True,
+ initially='DEFERRED'
+ )
+
+ # only test implemented attrs
+ if testing.against('postgresql'):
+ test_attrs = ('match', 'onupdate', 'ondelete', 'deferrable', 'initially')
+ elif testing.against('mysql'):
+ test_attrs = ('onupdate', 'ondelete')
+
+ meta = self.metadata
+ Table('users', meta,
+ Column('id', sa.Integer, primary_key=True),
+ Column('name', sa.String(30)),
+ test_needs_fk=True)
+ Table('addresses', meta,
+ Column('id', sa.Integer, primary_key=True),
+ Column('user_id', sa.Integer, addresses_user_id_fkey),
+ test_needs_fk=True)
+ meta.create_all()
+
+ meta2 = MetaData()
+ meta2.reflect(testing.db)
+ for fk in meta2.tables['addresses'].foreign_keys:
+ ref = locals()[fk.name]
+ for attr in test_attrs:
+ assert getattr(fk, attr) == getattr(ref, attr)
+
def test_pks_not_uniques(self):
"""test that primary key reflection not tripped up by unique
indexes"""