diff options
Diffstat (limited to 'test/dialect/postgresql/test_reflection.py')
| -rw-r--r-- | test/dialect/postgresql/test_reflection.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py index 1d6a41765..26de23902 100644 --- a/test/dialect/postgresql/test_reflection.py +++ b/test/dialect/postgresql/test_reflection.py @@ -1,5 +1,6 @@ # coding: utf-8 +from sqlalchemy.engine import reflection from sqlalchemy.testing.assertions import eq_, assert_raises, \ AssertsExecutionResults from sqlalchemy.testing import fixtures @@ -622,6 +623,26 @@ class ReflectionTest(fixtures.TestBase): for fk in fks: eq_(fk, fk_ref[fk['name']]) + @testing.provide_metadata + def test_inspect_enums_custom_schema(self): + conn = testing.db.connect() + enum_type = postgresql.ENUM('sad', 'ok', 'happy', name='mood', + metadata=self.metadata, schema='test_schema') + enum_type.create(conn) + inspector = reflection.Inspector.from_engine(conn.engine) + eq_(inspector.load_enums(conn, 'test_schema'), { + u'test_schema.mood': {'labels': [u'sad', u'ok', u'happy']}}) + + @testing.provide_metadata + def test_inspect_enums_schema(self): + conn = testing.db.connect() + enum_type = postgresql.ENUM('cat', 'dog', 'rat', name='pet', + metadata=self.metadata) + enum_type.create(conn) + inspector = reflection.Inspector.from_engine(conn.engine) + eq_(inspector.load_enums(conn), { + u'pet': {'labels': [u'cat', u'dog', u'rat']}}) + class CustomTypeReflectionTest(fixtures.TestBase): |