Fix test hang on OneBranch

By actually setting timeout to 10 minutes and
looping until we have the extension object defined.

Author: andyleejordan

.vscode-test.mjs

@@ -4,17 +4,22 @@ import path from "path";
 
 export default defineConfig({
     files: "test/**/*.test.ts",
-    // The default user data directory had too many characters for the IPC connection on macOS in CI.
+    // It may break CI but we'll know sooner rather than later
+    version: "insiders",
     launchArgs: [
+        // The default chat extensions hang under OneBranch network constraints
+        "--disable-extensions",
+        // Undocumented but valid option to use a temporary profile for testing
         "--profile-temp",
+        // The default user data directory had too many characters for the IPC connection on macOS in CI.
         "--user-data-dir",
         path.join(os.tmpdir(), "vscode-user-data"),
     ],
     workspaceFolder: "test/TestEnvironment.code-workspace",
     mocha: {
         ui: "bdd", // describe, it, etc.
         require: ["esbuild-register"], // transpile TypeScript on-the-fly
-        slow: 2000, // 2 seconds for slow test
-        timeout: 60 * 1000, // 10 minutes to allow for debugging
+        slow: 2 * 1000, // 2 seconds for slow test
+        timeout: 10 * 60 * 1000, // 10 minutes to allow for debugging
     },
 });

test/utils.ts

@@ -6,6 +6,7 @@ import * as path from "path";
 import * as vscode from "vscode";
 import type { IPowerShellExtensionClient } from "../src/features/ExternalApi";
 import type { ILogger } from "../src/logging";
+import { sleep } from "../src/utils";
 
 // This lets us test the rest of our path assumptions against the baseline of
 // this test file existing at `<root>/test/utils.js`.
@@ -64,11 +65,17 @@ export class TestLogger implements ILogger {
 export const testLogger = new TestLogger();
 
 export async function ensureExtensionIsActivated(): Promise<IPowerShellExtensionClient> {
-    const extension = vscode.extensions.getExtension(extensionId);
-    if (!extension!.isActive) {
-        await extension!.activate();
+    let extension = vscode.extensions.getExtension(extensionId);
+    while (!extension) {
+        // Wait for VS Code to be ready
+        testLogger.writeDebug(`Extension ${extensionId} not yet found...`);
+        await sleep(200);
+        extension = vscode.extensions.getExtension(extensionId);
     }
-    return extension!.exports as IPowerShellExtensionClient;
+    if (!extension.isActive) {
+        await extension.activate();
+    }
+    return extension.exports as IPowerShellExtensionClient;
 }
 
 export async function ensureEditorServicesIsConnected(): Promise<IPowerShellExtensionClient> {