Check for release extension also in tests (#1790)

microsoft · Jul 24, 2024 · 5763700 · 5763700
1 parent d62d88f
commit 5763700
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/vscode/test/suites/extensionUtils.ts b/vscode/test/suites/extensionUtils.ts
@@ -8,8 +8,15 @@ import { type ExtensionApi } from "../../src/extension";
 const extensionLogLevel = "warn";
 
 export async function activateExtension() {
-  // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-  const ext = vscode.extensions.getExtension("quantum.qsharp-lang-vscode-dev")!;
+  // Check for pre-release or stable builds of the extension, as could be in release pipeline
+  const ext =
+    vscode.extensions.getExtension("quantum.qsharp-lang-vscode-dev") ??
+    vscode.extensions.getExtension("quantum.qsharp-lang-vscode");
+
+  if (!ext) {
+    throw new Error("qsharp extension not found");
+  }
+
   if (ext.isActive) {
     return;
   }