tests

Author: KSDaemon

packages/cubejs-backend-native/test/globals.py

@@ -0,0 +1,14 @@
+from cube import TemplateContext
+import os
+from utils import answer_to_main_question
+
+
+template = TemplateContext()
+
+value_or_none = os.getenv('MY_ENV_VAR')
+template.add_variable('value_or_none', value_or_none)
+
+value_or_default = os.getenv('MY_OTHER_ENV_VAR', 'my_default_value')
+template.add_variable('value_or_default', value_or_default)
+
+template.add_variable('answer_to_main_question', answer_to_main_question())

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

@@ -28,6 +28,18 @@ async function loadConfigurationFile(fileName: string) {
   return config;
 }
 
+const nativeInstance = new native.NativeInstance();
+
+suite('Python Models', () => {
+  test('models import', async () => {
+    const fullFileName = path.join(process.cwd(), 'test', 'globals.py');
+    const content = await fs.readFile(fullFileName, 'utf8');
+
+    // Just checking it won't fail
+    await nativeInstance.loadPythonContext(fullFileName, content);
+  });
+});
+
 suite('Python Config', () => {
   let config: PyConfiguration;
 

packages/cubejs-backend-native/test/utils.py

@@ -3,3 +3,6 @@
 # Simple test function
 def test_function(query: dict) -> dict:
   return query
+
+def answer_to_main_question() -> str:
+  return "42"