some fixes in tests

Author: KSDaemon

packages/cubejs-schema-compiler/test/integration/postgres/yaml-compiler.test.ts

@@ -1,5 +1,5 @@
 import { PostgresQuery } from '../../../src/adapter/PostgresQuery';
-import { prepareYamlCompiler } from '../../unit/PrepareCompiler';
+import { prepareCompiler, prepareYamlCompiler } from '../../unit/PrepareCompiler';
 import { dbRunner } from './PostgresDBRunner';
 
 describe('YAMLCompiler', () => {
@@ -676,7 +676,7 @@ views:
     );
   });
 
-  it('calling cube\'s sql()', async () => {
+  it('calling cube\'s sql() (yaml-yaml)', async () => {
     const { compiler, joinGraph, cubeEvaluator } = prepareYamlCompiler(
       `cubes:
   - name: simple_orders
@@ -706,7 +706,7 @@ views:
   - name: simple_orders_sql_ext
 
     sql: >
-      SELECT * FROM ({simple_orders.sql()}) as parent
+      SELECT * FROM {simple_orders.sql()} as parent
       WHERE status = 'new'
 
     measures:
@@ -757,4 +757,100 @@ views:
       ]
     );
   });
+
+  it('calling cube\'s sql() (yaml-js)', async () => {
+    const { compiler, joinGraph, cubeEvaluator } = prepareCompiler([
+      {
+        content: `
+cube('simple_orders', {
+  sql: \`
+      SELECT 1 AS id, 100 AS amount, 'new' status, '2025-04-15'::TIMESTAMP AS created_at
+      UNION ALL
+      SELECT 2 AS id, 200 AS amount, 'new' status, '2025-04-16'::TIMESTAMP AS created_at
+      UNION ALL
+      SELECT 3 AS id, 300 AS amount, 'processed' status, '2025-04-17'::TIMESTAMP AS created_at
+      UNION ALL
+      SELECT 4 AS id, 500 AS amount, 'processed' status, '2025-04-18'::TIMESTAMP AS created_at
+      UNION ALL
+      SELECT 5 AS id, 600 AS amount, 'shipped' status, '2025-04-19'::TIMESTAMP AS created_at
+  \`,
+
+  dimensions: {
+    status: {
+      sql: 'status',
+      type: 'string',
+    },
+  },
+
+  measures: {
+    count: {
+      type: 'count',
+    },
+    total_amount: {
+      type: 'sum',
+      sql: 'total_amount',
+    },
+  },
+});
+    `,
+        fileName: 'cube.js',
+      },
+      {
+        content: `cubes:
+  - name: simple_orders_sql_ext
+
+    sql: >
+      SELECT * FROM {simple_orders.sql()} as parent
+      WHERE status = 'new'
+
+    measures:
+      - name: count
+        type: count
+
+      - name: total_amount
+        sql: amount
+        type: sum
+
+    dimensions:
+      - name: id
+        sql: id
+        type: number
+        primary_key: true
+
+      - name: created_at
+        sql: created_at
+        type: time
+    `,
+        fileName: 'cube.yml',
+      },
+    ]);
+
+    await compiler.compile();
+
+    const query = new PostgresQuery({ joinGraph, cubeEvaluator, compiler }, {
+      measures: ['simple_orders_sql_ext.count'],
+      timeDimensions: [{
+        dimension: 'simple_orders_sql_ext.created_at',
+        granularity: 'day',
+        dateRange: ['2025-04-01', '2025-05-01']
+      }],
+      timezone: 'UTC',
+      preAggregationsSchema: ''
+    });
+
+    const res = await dbRunner.evaluateQueryWithPreAggregations(query);
+
+    expect(res).toEqual(
+      [
+        {
+          simple_orders_sql_ext__count: '1',
+          simple_orders_sql_ext__created_at_day: '2025-04-15T00:00:00.000Z',
+        },
+        {
+          simple_orders_sql_ext__count: '1',
+          simple_orders_sql_ext__created_at_day: '2025-04-16T00:00:00.000Z',
+        }
+      ]
+    );
+  });
 });

packages/cubejs-schema-compiler/test/unit/yaml-schema.test.ts

@@ -604,7 +604,7 @@ describe('Yaml Schema Testing', () => {
   - name: simple_orders_sql_ext
 
     sql: >
-      SELECT * FROM {simple_orders.sql()}
+      SELECT * FROM {simple_orders.sql()} as q
       WHERE status = 'processed'
 
     measures: