diff --git a/src/core/webview/__tests__/webviewMessageHandler.spec.ts b/src/core/webview/__tests__/webviewMessageHandler.spec.ts
index d8dc1801e1..f73a531e2d 100644
--- a/src/core/webview/__tests__/webviewMessageHandler.spec.ts
+++ b/src/core/webview/__tests__/webviewMessageHandler.spec.ts
@@ -14,6 +14,10 @@ vi.mock("vscode", () => ({
 	ConfigurationTarget: {
 		Global: 1,
 	},
+	window: {
+		showErrorMessage: vi.fn().mockResolvedValue(undefined),
+		showInformationMessage: vi.fn().mockResolvedValue(undefined),
+	},
 }))
 
 // Mock Package