Update snapshots.

dbt-labs · Nov 13, 2024 · 5d323c0 · 5d323c0
1 parent 8675e15
commit 5d323c0
Show file tree

Hide file tree

Showing 4 changed files with 128 additions and 2 deletions.
diff --git a/...ots/test_cte_table_alias_simplifier.py/str/test_table_alias_no_simplification__result.txt b/...ots/test_cte_table_alias_simplifier.py/str/test_table_alias_no_simplification__result.txt
@@ -0,0 +1,41 @@
+test_name: test_table_alias_no_simplification
+test_filename: test_cte_table_alias_simplifier.py
+docstring:
+  Tests that table aliases in the SELECT statement of a CTE are not removed when required.
+---
+optimizer:
+  SqlTableAliasSimplifier
+
+sql_before_optimizing:
+  -- Top-level SELECT
+  WITH cte_source_0 AS (
+    -- CTE source 0
+    SELECT
+      from_source_alias.col_0 AS cte_source_0__col_0
+    FROM test_schema.test_table_0 from_source_alias
+    INNER JOIN
+      test_schema.test_table_1 right_source_alias
+    ON
+      from_source_alias.col_0 = right_source_alias.col_0
+  )
+
+  SELECT
+    cte_source_0_alias.cte_source_0__col_0 AS top_level__col_0
+  FROM cte_source_0 cte_source_0_alias
+
+sql_after_optimizing:
+  -- Top-level SELECT
+  WITH cte_source_0 AS (
+    -- CTE source 0
+    SELECT
+      from_source_alias.col_0 AS cte_source_0__col_0
+    FROM test_schema.test_table_0 from_source_alias
+    INNER JOIN
+      test_schema.test_table_1 right_source_alias
+    ON
+      from_source_alias.col_0 = right_source_alias.col_0
+  )
+
+  SELECT
+    cte_source_0__col_0 AS top_level__col_0
+  FROM cte_source_0 cte_source_0_alias
diff --git a/...pshots/test_cte_table_alias_simplifier.py/str/test_table_alias_simplification__result.txt b/...pshots/test_cte_table_alias_simplifier.py/str/test_table_alias_simplification__result.txt
@@ -0,0 +1,85 @@
+test_name: test_table_alias_simplification
+test_filename: test_cte_table_alias_simplifier.py
+docstring:
+  Tests that table aliases in the SELECT statement of a CTE are removed when not needed.
+---
+optimizer:
+  SqlTableAliasSimplifier
+
+sql_before_optimizing:
+  -- Top-level SELECT
+  WITH cte_source_0 AS (
+    -- CTE source 0
+    SELECT
+      test_table_alias.col_0 AS cte_source_0__col_0
+      , test_table_alias.col_1 AS cte_source_0__col_1
+    FROM (
+      -- CTE source 0 sub-query
+      SELECT
+        test_table_alias.col_0 AS cte_source_0_subquery__col_0
+        , test_table_alias.col_0 AS cte_source_0_subquery__col_1
+      FROM test_schema.test_table test_table_alias
+    ) cte_source_0_subquery
+  )
+
+  , cte_source_1 AS (
+    -- CTE source 1
+    SELECT
+      test_table_alias.col_0 AS cte_source_1__col_0
+      , test_table_alias.col_1 AS cte_source_1__col_1
+    FROM (
+      -- CTE source 1 sub-query
+      SELECT
+        cte_source_0_alias.cte_source_0__col_0 AS cte_source_1_subquery__col_0
+        , cte_source_0_alias.cte_source_0__col_0 AS cte_source_1_subquery__col_1
+      FROM cte_source_0 cte_source_0_alias
+    ) cte_source_1_subquery
+  )
+
+  SELECT
+    cte_source_0_alias.cte_source_0__col_0 AS top_level__col_0
+    , right_source_alias.right_source__col_1 AS top_level__col_1
+  FROM cte_source_0 cte_source_0_alias
+  INNER JOIN
+    cte_source_1 right_source_alias
+  ON
+    cte_source_0_alias.cte_source_0__col_1 = right_source_alias.right_source__col_1
+
+sql_after_optimizing:
+  -- Top-level SELECT
+  WITH cte_source_0 AS (
+    -- CTE source 0
+    SELECT
+      col_0 AS cte_source_0__col_0
+      , col_1 AS cte_source_0__col_1
+    FROM (
+      -- CTE source 0 sub-query
+      SELECT
+        col_0 AS cte_source_0_subquery__col_0
+        , col_0 AS cte_source_0_subquery__col_1
+      FROM test_schema.test_table test_table_alias
+    ) cte_source_0_subquery
+  )
+
+  , cte_source_1 AS (
+    -- CTE source 1
+    SELECT
+      col_0 AS cte_source_1__col_0
+      , col_1 AS cte_source_1__col_1
+    FROM (
+      -- CTE source 1 sub-query
+      SELECT
+        cte_source_0__col_0 AS cte_source_1_subquery__col_0
+        , cte_source_0__col_0 AS cte_source_1_subquery__col_1
+      FROM cte_source_0 cte_source_0_alias
+    ) cte_source_1_subquery
+  )
+
+  SELECT
+    cte_source_0_alias.cte_source_0__col_0 AS top_level__col_0
+    , right_source_alias.right_source__col_1 AS top_level__col_1
+  FROM cte_source_0 cte_source_0_alias
+  INNER JOIN
+    cte_source_1 right_source_alias
+  ON
+    cte_source_0_alias.cte_source_0__col_1 = right_source_alias.right_source__col_1
diff --git a/...implifier.py/SqlQueryPlan/test_table_alias_simplification__after_alias_simplification.sql b/...implifier.py/SqlQueryPlan/test_table_alias_simplification__after_alias_simplification.sql
@@ -1,7 +1,7 @@
 test_name: test_table_alias_simplification
 test_filename: test_table_alias_simplifier.py
 docstring:
-  Tests a case where no pruning should occur.
+  Tests that table aliases are removed when not needed.
 ---
 -- test0
 SELECT

diff --git a/...mplifier.py/SqlQueryPlan/test_table_alias_simplification__before_alias_simplification.sql b/...mplifier.py/SqlQueryPlan/test_table_alias_simplification__before_alias_simplification.sql
@@ -1,7 +1,7 @@
 test_name: test_table_alias_simplification
 test_filename: test_table_alias_simplifier.py
 docstring:
-  Tests a case where no pruning should occur.
+  Tests that table aliases are removed when not needed.
 ---
 -- test0
 SELECT