fix tests

databricks · Sep 17, 2024 · 013e513 · 013e513
1 parent 75bd656
commit 013e513
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/databricks-vscode.code-workspace b/databricks-vscode.code-workspace
@@ -22,6 +22,8 @@
         },
         "[typescript]": {
             "editor.defaultFormatter": "esbenp.prettier-vscode"
-        }
+        },
+        "jupyter.interactiveWindow.cellMarker.codeRegex": "^# COMMAND ----------|^# Databricks notebook source|^(#\\s*%%|#\\s*\\<codecell\\>|#\\s*In\\[\\d*?\\]|#\\s*In\\[ \\])",
+        "jupyter.interactiveWindow.cellMarker.default": "# COMMAND ----------"
     }
 }
diff --git a/packages/databricks-vscode/src/utils/envVarGenerators.test.ts b/packages/databricks-vscode/src/utils/envVarGenerators.test.ts
@@ -110,7 +110,7 @@ describe(__filename, () => {
             assert.deepEqual(actual, {
                 SPARK_CONNECT_USER_AGENT: "test/0.0.1",
                 DATABRICKS_PROJECT_ROOT: mockWorkspacePath.fsPath,
-                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "true",
+                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "1",
             });
         });
 
@@ -127,7 +127,7 @@ describe(__filename, () => {
             assert.deepEqual(actual, {
                 SPARK_CONNECT_USER_AGENT: "existing test/0.0.1",
                 DATABRICKS_PROJECT_ROOT: mockWorkspacePath.fsPath,
-                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "true",
+                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "1",
             });
         });
 
@@ -150,7 +150,7 @@ describe(__filename, () => {
             assert.deepEqual(actual, {
                 SPARK_CONNECT_USER_AGENT: "test/0.0.1",
                 DATABRICKS_PROJECT_ROOT: mockWorkspacePath.fsPath,
-                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "true",
+                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "1",
                 SPARK_REMOTE: `sc://${
                     Uri.parse(mockHost).authority
                 }:443/;token=token;use_ssl=true;x-databricks-cluster-id=${mockClusterId}`,
@@ -176,7 +176,7 @@ describe(__filename, () => {
             assert.deepEqual(actual, {
                 SPARK_CONNECT_USER_AGENT: "test/0.0.1",
                 DATABRICKS_PROJECT_ROOT: mockWorkspacePath.fsPath,
-                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "true",
+                SPARK_CONNECT_PROGRESS_BAR_ENABLED: "1",
             });
         });
     });