add tests for CubePropContextTranspiler with userAttributes

Author: KSDaemon

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

@@ -5,6 +5,7 @@ import babelTraverse from '@babel/traverse';
 import { prepareJsCompiler } from './PrepareCompiler';
 import { ImportExportTranspiler } from '../../src/compiler/transpilers';
 import { ErrorReporter } from '../../src/compiler/ErrorReporter';
+import { PostgresQuery } from '../../src';
 
 describe('Transpilers', () => {
   it('CubeCheckDuplicatePropTranspiler', async () => {
@@ -50,6 +51,66 @@ describe('Transpilers', () => {
     await compiler.compile();
   });
 
+  it('CubePropContextTranspiler without userAttributes should work normally', async () => {
+    const { joinGraph, cubeEvaluator, compiler } = prepareJsCompiler(`
+        cube(\`Test\`, {
+          sql: 'SELECT * FROM users',
+          dimensions: {
+            userId: {
+              sql: \`CASE WHEN userId = '\${securityContext.userId}' THEN id ELSE 0 END\`,
+              type: 'string'
+            }
+          }
+        })
+    `);
+
+    await compiler.compile();
+
+    const query = new PostgresQuery({ joinGraph, cubeEvaluator, compiler }, {
+      dimensions: ['Test.userId'],
+      timeDimensions: [],
+      timezone: 'America/Los_Angeles',
+      contextSymbols: {
+        securityContext: { userId: 'user123' }
+      }
+    });
+
+    const [res] = query.buildSqlAndParams();
+    console.log(res);
+
+    expect(res).toContain('CASE WHEN userId = \'user123\' THEN id ELSE 0 END');
+  });
+
+  it('CubePropContextTranspiler with shorthand userAttributes should work normally', async () => {
+    const { joinGraph, cubeEvaluator, compiler } = prepareJsCompiler(`
+        cube(\`Test\`, {
+          sql: 'SELECT * FROM users',
+          dimensions: {
+            userId: {
+              sql: \`CASE WHEN userId = '\${userAttributes.userId}' THEN id ELSE 0 END\`,
+              type: 'string'
+            }
+          }
+        })
+    `);
+
+    await compiler.compile();
+
+    const query = new PostgresQuery({ joinGraph, cubeEvaluator, compiler }, {
+      dimensions: ['Test.userId'],
+      timeDimensions: [],
+      timezone: 'America/Los_Angeles',
+      contextSymbols: {
+        securityContext: { cubeCloud: { userAttributes: { userId: 'user123' } } }
+      }
+    });
+
+    const [res] = query.buildSqlAndParams();
+    console.log(res);
+
+    expect(res).toContain('CASE WHEN userId = \'user123\' THEN id ELSE 0 END');
+  });
+
   it('ImportExportTranspiler', async () => {
     const ieTranspiler = new ImportExportTranspiler();
     const errorsReport = new ErrorReporter();