Add support for test shards (#817)

databricks · Jul 25, 2023 · 6a67a75 · 6a67a75
1 parent 86d991f
commit 6a67a75
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/databricks-vscode/src/utils/urlUtils.ts b/packages/databricks-vscode/src/utils/urlUtils.ts
@@ -23,7 +23,7 @@ export function normalizeHost(host: string): URL {
     }
     if (
         !url.hostname.match(
-            /(\.databricks\.azure\.us|\.databricks\.azure\.cn|\.azuredatabricks\.net|\.gcp\.databricks\.com|\.cloud\.databricks\.com)$/
+            /(\.databricks\.azure\.us|\.databricks\.azure\.cn|\.azuredatabricks\.net|\.gcp\.databricks\.com|\.cloud\.databricks\.com|\.dev\.databricks\.com)$/
         )
     ) {
         throw new Error("Not a Databricks host");
@@ -43,5 +43,7 @@ export function isGcpHost(url: URL): boolean {
 }
 
 export function isAwsHost(url: URL): boolean {
-    return !!url.hostname.match(/\.cloud\.databricks\.com$/);
+    return !!url.hostname.match(
+        /(\.cloud\.databricks\.com|\.dev\.databricks\.com)$/
+    );
 }