update mocks

github · Nov 27, 2024 · 656363b · 656363b
1 parent ddf4df4
commit 656363b
Showing 1 changed file with 16 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,11 @@ import {
   createQueryTreeFolderItem,
 } from "../../../../src/queries-panel/query-tree-view-item";
 import { dump } from "js-yaml";
+import type { LanguageClient } from "vscode-languageclient/node";
 
 describe("SkeletonQueryWizard", () => {
   let mockCli: CodeQLCliServer;
+  let mockLanguageClient: LanguageClient;
   let mockApp: App;
   let wizard: SkeletonQueryWizard;
   let mockDatabaseManager: DatabaseManager;
@@ -88,6 +90,11 @@ describe("SkeletonQueryWizard", () => {
         ]),
       resolveQlpacks: resolveQlpacksMock,
     });
+
+    mockLanguageClient = mockedObject<LanguageClient>({
+      sendNotification: jest.fn().mockResolvedValue([]),
+    });
+
     mockApp = createMockApp();
 
     mockDatabaseManager = mockedObject<DatabaseManager>({
@@ -142,6 +149,7 @@ describe("SkeletonQueryWizard", () => {
 
     wizard = new SkeletonQueryWizard(
       mockCli,
+      mockLanguageClient,
       jest.fn(),
       mockApp,
       mockDatabaseManager,
@@ -165,6 +173,7 @@ describe("SkeletonQueryWizard", () => {
     beforeEach(() => {
       wizard = new SkeletonQueryWizard(
         mockCli,
+        mockLanguageClient,
         jest.fn(),
         mockApp,
         mockDatabaseManager,
@@ -313,6 +322,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManagerWithItems,
@@ -362,6 +372,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManagerWithItems,
@@ -466,6 +477,7 @@ describe("SkeletonQueryWizard", () => {
 
       wizard = new SkeletonQueryWizard(
         mockCli,
+        mockLanguageClient,
         jest.fn(),
         mockApp,
         mockDatabaseManager,
@@ -687,6 +699,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -716,6 +729,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -749,6 +763,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,
@@ -792,6 +807,7 @@ describe("SkeletonQueryWizard", () => {
 
           wizard = new SkeletonQueryWizard(
             mockCli,
+            mockLanguageClient,
             jest.fn(),
             mockApp,
             mockDatabaseManager,