From 013e5132a1cfa74f4cd332165abb8e1b7112e095 Mon Sep 17 00:00:00 2001
From: Fabian Jakobs <fabian.jakobs@databricks.com>
Date: Tue, 17 Sep 2024 10:23:56 +0200
Subject: [PATCH] fix tests

---
 databricks-vscode.code-workspace                          | 4 +++-
 .../databricks-vscode/src/utils/envVarGenerators.test.ts  | 8 ++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/databricks-vscode.code-workspace b/databricks-vscode.code-workspace
index 3a2047c52..7bbbdc42b 100644
--- 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
index 4c55b00d3..e3538cd93 100644
--- 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",
             });
         });
     });