test for yaml pre-agg gen

Author: KSDaemon

packages/cubejs-schema-compiler/test/unit/__snapshots__/cube_schema_converter.test.ts.snap

@@ -1,5 +1,41 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
+exports[`CubeSchemaConverter adds a pre-aggregation to JS model using CubePreAggregationConverter: single_cube.yaml 1`] = `
+"# Useless comment for compilation, but is checked in
+# CubeSchemaConverter tests
+cubes:
+  - name: yml_orders
+    sql_table: yml_orders
+
+    measures:
+      - name: count
+        type: count
+      - name: sum
+        type: sum
+        sql: amount
+      - name: min
+        sql: amount
+        type: min
+      - name: max
+        sql: amount
+        type: max
+    dimensions:
+      - name: id
+        sql: id
+        type: number
+        primary_key: true
+      - name: createdAt
+        sql: created_at
+        type: time
+    pre_aggregations:
+      - name: main
+        measures:
+          - yml_orders.count
+        timeDimension: yml_orders.createdAt
+        granularity: day
+"
+`;
+
 exports[`CubeSchemaConverter converts all schema repository models (no changes, without additional converters): multi_ecom.yaml 1`] = `
 "cubes:
   - name: orders

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

@@ -1,32 +1,56 @@
-import { CubeSchemaConverter } from '../../src';
+import { CubePreAggregationConverter, CubeSchemaConverter } from '../../src';
 import {
   createCubeSchema,
   createCubeSchemaWithCustomGranularities,
   createCubeSchemaYaml, createECommerceSchema,
   createSchemaYaml
 } from './utils';
 
+const repo = {
+  localPath: () => __dirname,
+  dataSchemaFiles: () => Promise.resolve([
+    { fileName: 'single_cube_no_preaggs.js', content: createCubeSchema({ name: 'single_cube' }) },
+    { fileName: 'orders_and_users.js', content: createCubeSchemaWithCustomGranularities('js_orders') },
+    { fileName: 'single_cube.yaml', content: createCubeSchemaYaml({ name: 'yml_orders', sqlTable: 'yml_orders' }) },
+    { fileName: 'multi_ecom.yaml', content: createSchemaYaml(createECommerceSchema()) },
+    { fileName: 'empty1.yaml', content: '       ' },
+    { fileName: 'empty2.yaml', content: 'string     ' },
+    { fileName: 'empty3.yaml', content: 'cubes: string     ' },
+    { fileName: 'empty4.yaml', content: '# just comment  ' },
+  ])
+};
+
 describe('CubeSchemaConverter', () => {
   it('converts all schema repository models (no changes, without additional converters)', async () => {
-    const repo = {
-      localPath: () => __dirname,
-      dataSchemaFiles: () => Promise.resolve([
-        { fileName: 'single_cube_no_preaggs.js', content: createCubeSchema({ name: 'single_cube' }) },
-        { fileName: 'orders_and_users.js', content: createCubeSchemaWithCustomGranularities('js_orders') },
-        { fileName: 'single_cube.yaml', content: createCubeSchemaYaml({ name: 'yml_orders', sqlTable: 'yml_orders' }) },
-        { fileName: 'multi_ecom.yaml', content: createSchemaYaml(createECommerceSchema()) },
-        { fileName: 'empty1.yaml', content: '       ' },
-        { fileName: 'empty2.yaml', content: 'string     ' },
-        { fileName: 'empty3.yaml', content: 'cubes: string     ' },
-        { fileName: 'empty4.yaml', content: '# just comment  ' },
-      ])
-    };
-
     const schemaConverter = new CubeSchemaConverter(repo, []);
     await schemaConverter.generate();
     const regeneratedFiles = schemaConverter.getSourceFiles();
     regeneratedFiles.forEach((regeneratedFile) => {
       expect(regeneratedFile.source).toMatchSnapshot(regeneratedFile.fileName);
     });
   });
+
+  it('adds a pre-aggregation to JS model using CubePreAggregationConverter', async () => {
+    const cubeName = 'yml_orders';
+    const preAggregationName = 'yml_orders_main';
+    const code = `
+name: main
+measures:
+  - yml_orders.count
+timeDimension: yml_orders.createdAt
+granularity: day
+`;
+
+    const schemaConverter = new CubeSchemaConverter(repo, [new CubePreAggregationConverter({
+      cubeName,
+      preAggregationName,
+      code
+    })]);
+
+    await schemaConverter.generate(cubeName);
+    const regeneratedFiles = schemaConverter.getSourceFiles();
+    regeneratedFiles.forEach((regeneratedFile) => {
+      expect(regeneratedFile.source).toMatchSnapshot(regeneratedFile.fileName);
+    });
+  });
 });