chore: ecom schema

cube-js · Nov 27, 2023 · 47bd58a · 47bd58a
1 parent de1bf48
commit 47bd58a
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 39 deletions.
diff --git a/packages/cubejs-schema-compiler/test/unit/pre-aggregations.test.ts b/packages/cubejs-schema-compiler/test/unit/pre-aggregations.test.ts
@@ -1,5 +1,5 @@
 import { prepareCompiler, prepareYamlCompiler } from './PrepareCompiler';
-import { createSchemaYaml } from './utils';
+import { createECommerceSchema, createSchemaYaml } from './utils';
 import { PostgresQuery } from '../../src';
 
 describe('pre-aggregations', () => {
@@ -95,44 +95,7 @@ describe('pre-aggregations', () => {
   // @link https://github.com/cube-js/cube/issues/6623
   it('view and pre-aggregation granularity', async () => {
     const { compiler, cubeEvaluator, joinGraph } = prepareYamlCompiler(
-      createSchemaYaml({
-        cubes: [{
-          name: 'orders',
-          sql_table: 'orders',
-          measures: [{
-            name: 'count',
-            type: 'count',
-          }],
-          dimensions: [{
-            name: 'created_at',
-            sql: 'created_at',
-            type: 'time',
-          }],
-          preAggregations: [{
-            name: 'orders_by_day_with_day',
-            measures: ['count'],
-            timeDimension: 'created_at',
-            granularity: 'day',
-            partition_granularity: 'day',
-            build_range_start: {
-              sql: 'SELECT NOW() - INTERVAL \'1000 day\'',
-            },
-            build_range_end: {
-              sql: 'SELECT NOW()'
-            },
-          }]
-        }],
-        views: [{
-          name: 'orders_view',
-          cubes: [{
-            join_path: 'orders',
-            includes: [
-              'created_at',
-              'count'
-            ]
-          }]
-        }]
-      })
+      createSchemaYaml(createECommerceSchema())
     );
 
     await compiler.compile();

diff --git a/packages/cubejs-schema-compiler/test/unit/utils.ts b/packages/cubejs-schema-compiler/test/unit/utils.ts
@@ -1,3 +1,5 @@
+import YAML from 'js-yaml';
+
 interface CreateCubeSchemaOptions {
   name: string,
   publicly?: boolean,
@@ -69,6 +71,15 @@ export function createCubeSchema({ name, refreshKey = '', preAggregations = '',
   `;
 }
 
+export type CreateSchemaOptions = {
+  cubes?: unknown[],
+  views?: unknown[]
+};
+
+export function createSchemaYaml(schema: CreateSchemaOptions): string {
+  return YAML.dump(schema);
+}
+
 export function createCubeSchemaYaml({ name, sqlTable }: CreateCubeSchemaOptions): string {
   return ` 
     cubes:
@@ -98,6 +109,47 @@ export function createCubeSchemaYaml({ name, sqlTable }: CreateCubeSchemaOptions
   `;
 }
 
+export function createECommerceSchema() {
+  return {
+    cubes: [{
+      name: 'orders',
+      sql_table: 'orders',
+      measures: [{
+        name: 'count',
+        type: 'count',
+      }],
+      dimensions: [{
+        name: 'created_at',
+        sql: 'created_at',
+        type: 'time',
+      }],
+      preAggregations: [{
+        name: 'orders_by_day_with_day',
+        measures: ['count'],
+        timeDimension: 'created_at',
+        granularity: 'day',
+        partition_granularity: 'day',
+        build_range_start: {
+          sql: 'SELECT NOW() - INTERVAL \'1000 day\'',
+        },
+        build_range_end: {
+          sql: 'SELECT NOW()'
+        },
+      }]
+    }],
+    views: [{
+      name: 'orders_view',
+      cubes: [{
+        join_path: 'orders',
+        includes: [
+          'created_at',
+          'count'
+        ]
+      }]
+    }]
+  };
+}
+
 /**
  * Returns joined test cubes schema. Schema looks like: A -< B -< C >- D >- E.
  * The original data set can be found under the link.