add tests

Author: KSDaemon

packages/cubejs-schema-compiler/test/integration/postgres/sql-generation.test.ts

@@ -3,7 +3,7 @@ import { UserError } from '../../../src/compiler/UserError';
 import { PostgresQuery } from '../../../src/adapter/PostgresQuery';
 import { BigqueryQuery } from '../../../src/adapter/BigqueryQuery';
 import { PrestodbQuery } from '../../../src/adapter/PrestodbQuery';
-import { prepareJsCompiler } from '../../unit/PrepareCompiler';
+import { prepareJsCompiler, prepareYamlCompiler } from '../../unit/PrepareCompiler';
 import { dbRunner } from './PostgresDBRunner';
 import { createJoinedCubesSchema } from '../../unit/utils';
 import { testWithPreAggregation } from './pre-aggregation-utils';
@@ -4051,4 +4051,145 @@ SELECT 1 AS revenue,  cast('2024-01-01' AS timestamp) as time UNION ALL
       }]
     );
   });
+
+  describe('Transitive join paths', () => {
+    // eslint-disable-next-line @typescript-eslint/no-shadow
+    const { compiler, joinGraph, cubeEvaluator } = prepareYamlCompiler(`
+cubes:
+  - name: merchant_dims
+    sql: |
+      (
+        SELECT 101 AS merchant_sk, 'M1' AS merchant_id
+        UNION ALL
+        SELECT 102 AS merchant_sk, 'M2' AS merchant_id
+      )
+    dimensions:
+      - name: merchant_sk
+        sql: merchant_sk
+        type: number
+        primary_key: true
+      - name: merchant_id
+        sql: merchant_id
+        type: string
+
+  - name: product_dims
+    sql: |
+      (
+        SELECT 201 AS product_sk, 'P1' AS product_id
+        UNION ALL
+        SELECT 202 AS product_sk, 'P2' AS product_id
+      )
+    dimensions:
+      - name: product_sk
+        sql: product_sk
+        type: number
+        primary_key: true
+      - name: product_id
+        sql: product_id
+        type: string
+
+  - name: merchant_and_product_dims
+    sql: |
+      (
+        SELECT 'M1' AS merchant_id, 'P1' AS product_id, 'Organic' AS acquisition_channel
+        UNION ALL
+        SELECT 'M1' AS merchant_id, 'P2' AS product_id, 'Paid' AS acquisition_channel
+        UNION ALL
+        SELECT 'M2' AS merchant_id, 'P1' AS product_id, 'Referral' AS acquisition_channel
+      )
+    dimensions:
+      - name: product_id
+        sql: product_id
+        type: string
+        primary_key: true
+      - name: merchant_id
+        sql: merchant_id
+        type: string
+        primary_key: true
+      - name: acquisition_channel
+        sql: acquisition_channel
+        type: string
+
+  - name: test_facts
+    sql: |
+      (
+        SELECT DATE '2023-01-01' AS reporting_date, 101 AS merchant_sk, 201 AS product_sk, 100 AS amount
+        UNION ALL
+        SELECT DATE '2023-01-01' AS reporting_date, 101 AS merchant_sk, 202 AS product_sk, 150 AS amount
+        UNION ALL
+        SELECT DATE '2023-01-02' AS reporting_date, 102 AS merchant_sk, 201 AS product_sk, 200 AS amount
+      )
+    joins:
+      - name: merchant_dims
+        relationship: many_to_one
+        sql: "{CUBE}.merchant_sk = {merchant_dims.merchant_sk}"
+      - name: product_dims
+        relationship: many_to_one
+        sql: "{CUBE}.product_sk = {product_dims.product_sk}"
+      - name: merchant_and_product_dims # This join depends on merchant_dims and product_dims
+        relationship: many_to_one
+        sql: "{merchant_dims.merchant_id} = {merchant_and_product_dims.merchant_id} AND {product_dims.product_id} = {merchant_and_product_dims.product_id}"
+    dimensions:
+      - name: reporting_date
+        sql: reporting_date
+        type: time
+        primary_key: true
+      - name: merchant_sk
+        sql: merchant_sk
+        type: number
+        primary_key: true
+      - name: product_sk
+        sql: product_sk
+        type: number
+        primary_key: true
+      - name: acquisition_channel # This dimension triggers the join to merchant_and_product_dims
+        sql: "{merchant_and_product_dims.acquisition_channel}"
+        type: string
+    measures:
+      - name: amount_sum
+        sql: amount
+        type: sum
+    `);
+
+    it('querying cube dimension that require transitive joins', async () => {
+      await compiler.compile();
+      const query = new PostgresQuery({ joinGraph, cubeEvaluator, compiler }, {
+        measures: [],
+        dimensions: [
+          'test_facts.reporting_date',
+          'test_facts.merchant_sk',
+          'test_facts.product_sk',
+          'test_facts.acquisition_channel'
+        ],
+        order: [{
+          id: 'test_facts.acquisition_channel'
+        }],
+        timezone: 'America/Los_Angeles'
+      });
+
+      const res = await dbRunner.testQuery(query.buildSqlAndParams());
+      console.log(JSON.stringify(res));
+
+      expect(res).toEqual([
+        {
+          test_facts__acquisition_channel: 'Organic',
+          test_facts__merchant_sk: 101,
+          test_facts__product_sk: 201,
+          test_facts__reporting_date: '2023-01-01T00:00:00.000Z',
+        },
+        {
+          test_facts__acquisition_channel: 'Paid',
+          test_facts__merchant_sk: 101,
+          test_facts__product_sk: 202,
+          test_facts__reporting_date: '2023-01-01T00:00:00.000Z',
+        },
+        {
+          test_facts__acquisition_channel: 'Referral',
+          test_facts__merchant_sk: 102,
+          test_facts__product_sk: 201,
+          test_facts__reporting_date: '2023-01-02T00:00:00.000Z',
+        },
+      ]);
+    });
+  });
 });