Fix tests

Author: EhabY

test/mocks/vscode.runtime.ts

@@ -28,6 +28,11 @@ export const TreeItemCollapsibleState = E({
 export const StatusBarAlignment = E({ Left: 1, Right: 2 });
 export const ExtensionMode = E({ Production: 1, Development: 2, Test: 3 });
 export const UIKind = E({ Desktop: 1, Web: 2 });
+export const InputBoxValidationSeverity = E({
+	Info: 1,
+	Warning: 2,
+	Error: 3,
+});
 
 export class Uri {
 	constructor(
@@ -142,6 +147,7 @@ const vscode = {
 	StatusBarAlignment,
 	ExtensionMode,
 	UIKind,
+	InputBoxValidationSeverity,
 	Uri,
 	EventEmitter,
 	window,

test/unit/login/loginCoordinator.test.ts

@@ -153,18 +153,15 @@ describe("LoginCoordinator", () => {
 				createTestContext();
 			const user = createMockUser();
 
-			// First call fails (no token), second succeeds after user enters token
-			mockAdapter
-				.mockRejectedValueOnce({
-					response: { status: 401, data: { message: "Unauthorized" } },
-				})
-				.mockResolvedValueOnce({
-					data: user,
-					status: 200,
-					statusText: "OK",
-					headers: {},
-					config: {},
-				});
+			// No stored token, so goes directly to input box flow
+			// Mock succeeds when validateInput calls getAuthenticatedUser
+			mockAdapter.mockResolvedValueOnce({
+				data: user,
+				status: 200,
+				statusText: "OK",
+				headers: {},
+				config: {},
+			});
 
 			// User enters a new token in the input box
 			userInteraction.setInputBoxValue("new-token");
@@ -198,9 +195,12 @@ describe("LoginCoordinator", () => {
 
 	describe("same-window guard", () => {
 		it("prevents duplicate login calls for same hostname", async () => {
-			const { mockAdapter, coordinator } = createTestContext();
+			const { mockAdapter, userInteraction, coordinator } = createTestContext();
 			const user = createMockUser();
 
+			// User enters a token in the input box
+			userInteraction.setInputBoxValue("new-token");
+
 			let resolveAuth: (value: unknown) => void;
 			mockAdapter.mockReturnValue(
 				new Promise((resolve) => {
@@ -220,7 +220,7 @@ describe("LoginCoordinator", () => {
 				safeHostname: TEST_HOSTNAME,
 			});
 
-			// Resolve the auth
+			// Resolve the auth (this validates the token from input box)
 			resolveAuth!({
 				data: user,
 				status: 200,
@@ -234,8 +234,8 @@ describe("LoginCoordinator", () => {
 			expect(result1.success).toBe(true);
 			expect(result1).toEqual(result2);
 
-			// Adapter should only be called once (guard prevents duplicate requests)
-			expect(mockAdapter).toHaveBeenCalledTimes(1);
+			// Input box should only be shown once (guard prevents duplicate prompts)
+			expect(vscode.window.showInputBox).toHaveBeenCalledTimes(1);
 		});
 	});