add and debug tests

Author: KSDaemon

packages/cubejs-backend-native/js/index.ts

@@ -354,8 +354,8 @@ export interface PyConfiguration {
   checkAuth?: (req: unknown, authorization: string) => Promise<{ 'security_context'?: unknown }>
   queryRewrite?: (query: unknown, ctx: unknown) => Promise<unknown>
   contextToApiScopes?: () => Promise<string[]>
-  scheduledRefreshContexts?: () => Promise<string[]>
-  scheduledRefreshTimeZones?: () => Promise<string[]>
+  scheduledRefreshContexts?: (ctx: unknown) => Promise<string[]>
+  scheduledRefreshTimeZones?: (ctx: unknown) => Promise<string[]>
   contextToRoles?: (ctx: unknown) => Promise<string[]>
 }
 

packages/cubejs-backend-native/package.json

@@ -21,7 +21,8 @@
     "test:server": "CUBEJS_NATIVE_INTERNAL_DEBUG=true CUBESQL_LOG_LEVEL=trace CUBESQL_PG_PORT=5555 node dist/test/server.js",
     "test:server:stream": "CUBESQL_STREAM_MODE=true CUBESQL_LOG_LEVEL=error CUBESQL_PG_PORT=5555 node dist/test/server.js",
     "test:python": "CUBEJS_NATIVE_INTERNAL_DEBUG=true CUBESQL_LOG_LEVEL=trace CUBESQL_PG_PORT=5555 node dist/test/python.js",
-    "test:unit": "jest --forceExit",
+    "unit": "jest --forceExit",
+    "test:unit": "yarn run unit",
     "test:cargo": "cargo test",
     "lint": "eslint test/ js/ --ext .ts",
     "lint:fix": "eslint --fix test/ js/ --ext .ts"

packages/cubejs-backend-native/python/cube/src/__init__.py

@@ -44,7 +44,7 @@ class Configuration:
     allow_js_duplicate_props_in_schema: bool
     jwt: Dict
     scheduled_refresh_timer: Any
-    scheduled_refresh_timezones: Union[Callable[[RequestContext], list[str]], list[str]]
+    scheduled_refresh_time_zones: Union[Callable[[RequestContext], list[str]], list[str]]
     scheduled_refresh_concurrency: int
     scheduled_refresh_batch_size: int
     compiler_cache_size: int
@@ -117,7 +117,7 @@ def __init__(self):
         self.query_rewrite = None
         self.extend_context = None
         self.scheduled_refresh_contexts = None
-        self.scheduled_refresh_timezones = None
+        self.scheduled_refresh_time_zones = None
         self.context_to_api_scopes = None
         self.repository_factory = None
         self.schema_version = None

packages/cubejs-backend-native/src/python/cube_config.rs

@@ -33,7 +33,7 @@ impl CubeConfigPy {
             "scheduled_refresh_batch_size",
             "scheduled_refresh_concurrency",
             "scheduled_refresh_timer",
-            "scheduled_refresh_timezones",
+            "scheduled_refresh_time_zones",
             "schema_path",
             "sql_cache",
             "sql_password",

packages/cubejs-backend-native/test/old-config.py

@@ -31,6 +31,35 @@ async def context_to_api_scopes():
 
 settings.context_to_api_scopes = context_to_api_scopes
 
+async def scheduled_refresh_time_zones(ctx):
+  print("[python] scheduled_refresh_time_zones ctx=", ctx)
+  return ["Europe/Kyiv", "Antarctica/Troll", "Australia/Sydney"]
+
+settings.scheduled_refresh_time_zones = scheduled_refresh_time_zones
+
+
+async def scheduled_refresh_contexts(ctx):
+  print("[python] scheduled_refresh_contexts ctx=", ctx)
+  return [
+    {
+      "securityContext": {
+        "appid": 'test1', "u": {"prop1": "value1"}
+      }
+    },
+    {
+      "securityContext": {
+        "appid": 'test2', "u": {"prop1": "value2"}
+      }
+    },
+    {
+      "securityContext": {
+        "appid": 'test3', "u": {"prop1": "value3"}
+      }
+    },
+  ]
+
+settings.scheduled_refresh_contexts = scheduled_refresh_contexts
+
 def schema_version(ctx):
     print('[python] schema_version', ctx)
 

packages/cubejs-backend-native/test/python.test.ts

@@ -47,6 +47,8 @@ suite('Python Config', () => {
       repositoryFactory: expect.any(Function),
       schemaVersion: expect.any(Function),
       contextToRoles: expect.any(Function),
+      scheduledRefreshContexts: expect.any(Function),
+      scheduledRefreshTimeZones: expect.any(Function),
     });
 
     if (!config.checkAuth) {
@@ -88,15 +90,15 @@ suite('Python Config', () => {
       throw new Error('scheduledRefreshTimeZones was not defined in config.py');
     }
 
-    expect(await config.scheduledRefreshTimeZones()).toEqual(['Europe/Kyiv', 'Antarctica/Troll', 'Australia/Sydney']);
+    expect(await config.scheduledRefreshTimeZones({})).toEqual(['Europe/Kyiv', 'Antarctica/Troll', 'Australia/Sydney']);
   });
 
   test('scheduled_refresh_contexts', async () => {
     if (!config.scheduledRefreshContexts) {
       throw new Error('scheduledRefreshContexts was not defined in config.py');
     }
 
-    expect(await config.scheduledRefreshContexts()).toEqual([
+    expect(await config.scheduledRefreshContexts({})).toEqual([
       {
         securityContext: {
           appid: 'test1', u: { prop1: 'value1' }
@@ -192,6 +194,9 @@ darwinSuite('Old Python Config', () => {
       queryRewrite: expect.any(Function),
       repositoryFactory: expect.any(Function),
       schemaVersion: expect.any(Function),
+      contextToRoles: expect.any(Function),
+      scheduledRefreshContexts: expect.any(Function),
+      scheduledRefreshTimeZones: expect.any(Function),
     });
 
     if (!config.checkAuth) {