summaryrefslogtreecommitdiff
path: root/test/sql/test_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/sql/test_utils.py')
-rw-r--r--test/sql/test_utils.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/test/sql/test_utils.py b/test/sql/test_utils.py
index 1ccd1e123..a63e55c4e 100644
--- a/test/sql/test_utils.py
+++ b/test/sql/test_utils.py
@@ -30,3 +30,21 @@ class MiscTest(fixtures.TestBase):
subset_select = select([common.c.id, common.c.data]).alias()
eq_(sql_util.find_tables(subset_select), [common])
+
+ def test_find_tables_aliases(self):
+ metadata = MetaData()
+ common = Table(
+ "common",
+ metadata,
+ Column("id", Integer, primary_key=True),
+ Column("data", Integer),
+ Column("extra", String(45)),
+ )
+
+ calias = common.alias()
+ subset_select = select([common.c.id, calias.c.data]).subquery()
+
+ eq_(
+ set(sql_util.find_tables(subset_select, include_aliases=True)),
+ {common, calias, subset_select},
+ )