From c38af99ea695dc5f337794a1275270a2e7c540f0 Mon Sep 17 00:00:00 2001
From: Russell Clarey <russell.clarey@databricks.com>
Date: Thu, 6 Aug 2026 11:40:40 +0200
Subject: [PATCH] testing improvements

---
 packages/databricks-vscode/package.json        |  2 +-
 packages/databricks-vscode/src/test/runTest.ts |  1 +
 packages/databricks-vscode/src/test/suite.ts   |  1 +
 yarn.lock                                      | 10 +++++-----
 4 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/packages/databricks-vscode/package.json b/packages/databricks-vscode/package.json
index 8504d8aca..157950f5d 100644
--- a/packages/databricks-vscode/package.json
+++ b/packages/databricks-vscode/package.json
@@ -1610,7 +1610,7 @@
         "@typescript-eslint/eslint-plugin": "^8.62.0",
         "@typescript-eslint/parser": "^8.62.0",
         "@typescript-eslint/utils": "^8.62.0",
-        "@vscode/test-electron": "^3.0.0",
+        "@vscode/test-electron": "^3.1.0",
         "@wdio/cli": "^9.29.0",
         "@wdio/local-runner": "^9.29.0",
         "@wdio/mocha-framework": "^9.29.0",
diff --git a/packages/databricks-vscode/src/test/runTest.ts b/packages/databricks-vscode/src/test/runTest.ts
index 6102316e6..81e1b7337 100644
--- a/packages/databricks-vscode/src/test/runTest.ts
+++ b/packages/databricks-vscode/src/test/runTest.ts
@@ -50,6 +50,7 @@ async function main() {
             launchArgs: [tmpDir, "--user-data-dir", tmpDir],
             extensionTestsEnv: {
                 [EXTENSION_DEVELOPMENT]: "true",
+                MOCHA_GREP: process.env.MOCHA_GREP,
             },
         });
     } catch (err) {
diff --git a/packages/databricks-vscode/src/test/suite.ts b/packages/databricks-vscode/src/test/suite.ts
index 2d0a05894..49ad9463e 100644
--- a/packages/databricks-vscode/src/test/suite.ts
+++ b/packages/databricks-vscode/src/test/suite.ts
@@ -7,6 +7,7 @@ export async function run(): Promise<void> {
     const mocha = new Mocha({
         ui: "bdd",
         color: true,
+        grep: process.env.MOCHA_GREP,
     });
 
     // Add files to the test suite
diff --git a/yarn.lock b/yarn.lock
index 7664da390..0c4630c9e 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -2071,16 +2071,16 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@vscode/test-electron@npm:^3.0.0":
-  version: 3.0.0
-  resolution: "@vscode/test-electron@npm:3.0.0"
+"@vscode/test-electron@npm:^3.1.0":
+  version: 3.1.0
+  resolution: "@vscode/test-electron@npm:3.1.0"
   dependencies:
     http-proxy-agent: ^7.0.2
     https-proxy-agent: ^7.0.5
     jszip: ^3.10.1
     ora: ^8.1.0
     semver: ^7.6.2
-  checksum: 644182d4cb5f69afd03c68d9e61c8e654c6de39acb1822ada18e89fd717e000756b03ff036e892b15ce9de845103c063478903904791ad13273ad67701c5e612
+  checksum: 39ac3edc476204d04c2123cd5600803255ff5a1c1c07a34cfad0df2f2fa9ed87ab5bfd71d21a0974cc6aacf4a5335c4240bb0462f4291d47f01c7d0019e7ca24
   languageName: node
   linkType: hard
 
@@ -4008,7 +4008,7 @@ __metadata:
     "@typescript-eslint/utils": ^8.62.0
     "@vscode/debugadapter": ^1.64.0
     "@vscode/extension-telemetry": ^1.5.2
-    "@vscode/test-electron": ^3.0.0
+    "@vscode/test-electron": ^3.1.0
     "@vscode/webview-ui-toolkit": ^1.4.0
     "@wdio/cli": ^9.29.0
     "@wdio/local-runner": ^9.29.0