dbcli/sqlglot
1
0
Fork 0
Code Issues Activity

Merging upstream version 20.1.0.

Browse source 
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
Daniel Baumann 2025-02-13 21:17:09 +01:00
parent d4fe7bdb16
commit 90988d8258
Signed by: daniel
GPG key ID: FBB4F0E80A80222F
127 changed files with 73384 additions and 73067 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/test_lineage.py
View file

@ -229,13 +229,36 @@ class TestLineage(unittest.TestCase):
"output",
"SELECT (SELECT max(t3.my_column) my_column FROM foo t3) AS output FROM table3",
)
self.assertEqual(node.name, "SUBQUERY")
self.assertEqual(node.name, "output")
node = node.downstream[0]
self.assertEqual(node.name, "my_column")
node = node.downstream[0]
self.assertEqual(node.name, "t3.my_column")
self.assertEqual(node.source.sql(), "foo AS t3")
node = lineage(
"y",
"SELECT SUM((SELECT max(a) a from x) + (SELECT min(b) b from x) + c) AS y FROM x",
)
self.assertEqual(node.name, "y")
self.assertEqual(len(node.downstream), 3)
self.assertEqual(node.downstream[0].name, "a")
self.assertEqual(node.downstream[1].name, "b")
self.assertEqual(node.downstream[2].name, "x.c")
node = lineage(
"x",
"WITH cte AS (SELECT a, b FROM z) SELECT sum(SELECT a FROM cte) AS x, (SELECT b FROM cte) as y FROM cte",
)
self.assertEqual(node.name, "x")
self.assertEqual(len(node.downstream), 1)
node = node.downstream[0]
self.assertEqual(node.name, "a")
node = node.downstream[0]
self.assertEqual(node.name, "cte.a")
node = node.downstream[0]
self.assertEqual(node.name, "z.a")
def test_lineage_cte_union(self) -> None:
query = """
WITH dataset AS (