update mocks

github · Nov 27, 2024 · b6f83e0 · b6f83e0
1 parent ddf4df4
commit b6f83e0
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/...s/ql-vscode/test/vscode-tests/cli-integration/local-queries/skeleton-query-wizard.test.ts b/...s/ql-vscode/test/vscode-tests/cli-integration/local-queries/skeleton-query-wizard.test.ts
@@ -36,9 +36,12 @@ import {
   createQueryTreeFolderItem,
 } from "../../../../src/queries-panel/query-tree-view-item";
 import { dump } from "js-yaml";
+import type { LanguageClient } from "vscode-languageclient/node";
+import { mock } from "node:test";
 
 describe("SkeletonQueryWizard", () => {
   let mockCli: CodeQLCliServer;
+  let mockLanguageClient: LanguageClient;
   let mockApp: App;
   let wizard: SkeletonQueryWizard;
   let mockDatabaseManager: DatabaseManager;
@@ -88,6 +91,11 @@ describe("SkeletonQueryWizard", () => {
         ]),
       resolveQlpacks: resolveQlpacksMock,
     });
+
+    mockLanguageClient = mockedObject<LanguageClient>({
+      sendNotification: jest.fn().mockResolvedValue([]),
+    });
+
     mockApp = createMockApp();
 
     mockDatabaseManager = mockedObject<DatabaseManager>({
@@ -142,6 +150,7 @@ describe("SkeletonQueryWizard", () => {
 
     wizard = new SkeletonQueryWizard(
       mockCli,
+      mockLanguageClient,
       jest.fn(),
       mockApp,
       mockDatabaseManager,
@@ -165,6 +174,7 @@ describe("SkeletonQueryWizard", () => {
     beforeEach(() => {
       wizard = new SkeletonQueryWizard(
         mockCli,
+        mockLanguageClient,
         jest.fn(),
         mockApp,
         mockDatabaseManager,
@@ -313,6 +323,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManagerWithItems,
@@ -362,6 +373,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManagerWithItems,
@@ -466,6 +478,7 @@ describe("SkeletonQueryWizard", () => {
 
       wizard = new SkeletonQueryWizard(
         mockCli,
+        mockLanguageClient,
         jest.fn(),
         mockApp,
         mockDatabaseManager,
@@ -687,6 +700,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -716,6 +730,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -749,6 +764,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -792,6 +808,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,