fix: ParsedQuery subselect edge case (#13602)

This commit is contained in:
Erik Ritter
2021-03-12 14:54:02 -08:00
committed by GitHub
parent 4fc41e1cce
commit 06d6d7f8e2
2 changed files with 18 additions and 3 deletions

View File

@@ -158,6 +158,13 @@ class TestSupersetSqlParse(unittest.TestCase):
query = "SELECT f1, (SELECT count(1) FROM t2) FROM t1"
self.assertEqual({Table("t1"), Table("t2")}, self.extract_tables(query))
query = "SELECT f1, (SELECT count(1) FROM t2) as f2 FROM t1"
self.assertEqual({Table("t1"), Table("t2")}, self.extract_tables(query))
def test_parentheses(self):
query = "SELECT f1, (x + y) AS f2 FROM t1"
self.assertEqual({Table("t1")}, self.extract_tables(query))
def test_union(self):
query = "SELECT * FROM t1 UNION SELECT * FROM t2"
self.assertEqual({Table("t1"), Table("t2")}, self.extract_tables(query))