Add session tests

Author: JustinGrote

test/core/session.test.ts

@@ -0,0 +1,170 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+import { type TelemetryReporter } from "@vscode/extension-telemetry";
+import * as assert from "assert";
+import Sinon from "sinon";
+import * as vscode from "vscode";
+import type { DocumentSelector } from "vscode-languageclient";
+import { SessionManager } from "../../src/session";
+import { stubInterface, testLogger } from "../utils";
+
+describe("SessionManager middleware", () => {
+    afterEach(() => {
+        Sinon.restore();
+    });
+
+    it("delays didOpen until session initialization completes", async () => {
+        const registerCommandStub = Sinon.stub(
+            vscode.commands,
+            "registerCommand",
+        ).returns(disposableStub());
+        const onDidChangeConfigurationStub = Sinon.stub(
+            vscode.workspace,
+            "onDidChangeConfiguration",
+        ).returns(disposableStub());
+
+        Sinon.stub(vscode.languages, "createLanguageStatusItem").returns(
+            stubInterface<vscode.LanguageStatusItem>({
+                text: "",
+                detail: "",
+                busy: false,
+                severity: vscode.LanguageStatusSeverity.Information,
+                dispose: () => {
+                    return;
+                },
+            }),
+        );
+
+        const manager = new SessionManager(
+            stubInterface<vscode.ExtensionContext>({
+                globalStorageUri: vscode.Uri.file("C:/tmp"),
+                extensionMode: vscode.ExtensionMode.Test,
+                subscriptions: [],
+                logUri: vscode.Uri.file("C:/tmp"),
+            }),
+            testLogger,
+            ["powershell"] as DocumentSelector,
+            "Visual Studio Code",
+            "PowerShell",
+            "2026.5.0",
+            "ms-vscode",
+            stubInterface<TelemetryReporter>(),
+        );
+
+        const document = stubInterface<vscode.TextDocument>();
+        let nextCalled = false;
+        const next = Sinon.spy(
+            (_document: vscode.TextDocument): Promise<void> => {
+                nextCalled = true;
+                return Promise.resolve();
+            },
+        );
+
+        const didOpenPromise = manager.didOpen(document, next);
+
+        await Promise.resolve();
+        assert.equal(
+            nextCalled,
+            false,
+            "didOpen should wait for initialization",
+        );
+
+        (
+            manager as unknown as {
+                started: PromiseWithResolvers<undefined>;
+            }
+        ).started.resolve(undefined);
+
+        await didOpenPromise;
+        assert.equal(
+            nextCalled,
+            true,
+            "didOpen should run after initialization",
+        );
+        assert.ok(
+            registerCommandStub.calledThrice,
+            "SessionManager should register its commands in constructor",
+        );
+        assert.ok(
+            onDidChangeConfigurationStub.calledOnce,
+            "SessionManager should register configuration listener in constructor",
+        );
+    });
+
+    it("delays didChange until session initialization completes", async () => {
+        Sinon.stub(vscode.commands, "registerCommand").returns(
+            disposableStub(),
+        );
+        Sinon.stub(vscode.workspace, "onDidChangeConfiguration").returns(
+            disposableStub(),
+        );
+
+        Sinon.stub(vscode.languages, "createLanguageStatusItem").returns(
+            stubInterface<vscode.LanguageStatusItem>({
+                text: "",
+                detail: "",
+                busy: false,
+                severity: vscode.LanguageStatusSeverity.Information,
+                dispose: () => {
+                    return;
+                },
+            }),
+        );
+
+        const manager = new SessionManager(
+            stubInterface<vscode.ExtensionContext>({
+                globalStorageUri: vscode.Uri.file("C:/tmp"),
+                extensionMode: vscode.ExtensionMode.Test,
+                subscriptions: [],
+                logUri: vscode.Uri.file("C:/tmp"),
+            }),
+            testLogger,
+            ["powershell"] as DocumentSelector,
+            "Visual Studio Code",
+            "PowerShell",
+            "2026.5.0",
+            "ms-vscode",
+            stubInterface<TelemetryReporter>(),
+        );
+
+        const changeEvent = stubInterface<vscode.TextDocumentChangeEvent>();
+        let nextCalled = false;
+        const next = Sinon.spy(
+            (_event: vscode.TextDocumentChangeEvent): Promise<void> => {
+                nextCalled = true;
+                return Promise.resolve();
+            },
+        );
+
+        const didChangePromise = manager.didChange(changeEvent, next);
+
+        await Promise.resolve();
+        assert.equal(
+            nextCalled,
+            false,
+            "didChange should wait for initialization",
+        );
+
+        (
+            manager as unknown as {
+                started: PromiseWithResolvers<undefined>;
+            }
+        ).started.resolve(undefined);
+
+        await didChangePromise;
+        assert.equal(
+            nextCalled,
+            true,
+            "didChange should run after initialization",
+        );
+    });
+});
+
+function disposableStub(): vscode.Disposable {
+    return {
+        dispose: (): void => {
+            return;
+        },
+    };
+}