chore: add nested cte to sushi for testing

Author: benfdking

examples/sushi/models/customers.sql

@@ -17,7 +17,7 @@ CREATE VIEW raw.demographics AS (
   SELECT 1 AS customer_id, '00000' AS zip
 );
 
-WITH current_marketing AS (
+WITH current_marketing_outer AS (
   SELECT
     customer_id,
     status
@@ -29,7 +29,15 @@ SELECT DISTINCT
   m.status,
   d.zip
   FROM sushi.orders AS o
-LEFT JOIN current_marketing AS m
+LEFT JOIN (
+  WITH current_marketing AS (
+    SELECT
+      customer_id,
+      status
+    FROM current_marketing_outer
+  )
+  SELECT * FROM current_marketing
+) AS m
   ON o.customer_id = m.customer_id
 LEFT JOIN raw.demographics AS d
   ON o.customer_id = d.customer_id

tests/test_forking.py

@@ -19,7 +19,7 @@ def test_parallel_load(assert_exp_eq, mocker):
     assert_exp_eq(
         context.render("sushi.customers"),
         """
-WITH "current_marketing" AS (
+WITH "current_marketing_outer" AS (
   SELECT
     "marketing"."customer_id" AS "customer_id",
     "marketing"."status" AS "status"
@@ -32,7 +32,18 @@ def test_parallel_load(assert_exp_eq, mocker):
   "m"."status" AS "status",
   "d"."zip" AS "zip"
 FROM "memory"."sushi"."orders" AS "o"
-LEFT JOIN "current_marketing" AS "m"
+LEFT JOIN (
+  WITH "current_marketing" AS (
+    SELECT
+      "current_marketing_outer"."customer_id" AS "customer_id",
+      "current_marketing_outer"."status" AS "status"
+    FROM "current_marketing_outer" AS "current_marketing_outer"
+  )
+  SELECT
+    "current_marketing"."customer_id" AS "customer_id",
+    "current_marketing"."status" AS "status"
+  FROM "current_marketing" AS "current_marketing"
+) AS "m"
   ON "m"."customer_id" = "o"."customer_id"
 LEFT JOIN "memory"."raw"."demographics" AS "d"
   ON "d"."customer_id" = "o"."customer_id"

tests/web/test_lineage.py

@@ -47,7 +47,7 @@ def test_get_lineage(client: TestClient, web_sushi_context: Context) -> None:
             "customer_id": {
                 "expression": 'CAST("o"."customer_id" AS INT) AS "customer_id" /* this comment should not be registered */',
                 "models": {'"memory"."sushi"."orders"': ["customer_id"]},
-                "source": '''WITH "current_marketing" AS (
+                "source": '''WITH "current_marketing_outer" AS (
   SELECT
     "marketing"."customer_id" AS "customer_id",
     "marketing"."status" AS "status"
@@ -58,7 +58,18 @@ def test_get_lineage(client: TestClient, web_sushi_context: Context) -> None:
 SELECT DISTINCT
   CAST("o"."customer_id" AS INT) AS "customer_id" /* this comment should not be registered */
 FROM "memory"."sushi"."orders" AS "o"
-LEFT JOIN "current_marketing" AS "m"
+LEFT JOIN (
+  WITH "current_marketing" AS (
+    SELECT
+      "current_marketing_outer"."customer_id" AS "customer_id",
+      "current_marketing_outer"."status" AS "status"
+    FROM "current_marketing_outer" AS "current_marketing_outer"
+  )
+  SELECT
+    "current_marketing"."customer_id" AS "customer_id",
+    "current_marketing"."status" AS "status"
+  FROM "current_marketing" AS "current_marketing"
+) AS "m"
   ON "m"."customer_id" = "o"."customer_id"
 LEFT JOIN "memory"."raw"."demographics" AS "d"
   ON "d"."customer_id" = "o"."customer_id"''',