update mocks

github · Nov 27, 2024 · 9c87113 · 9c87113
1 parent ddf4df4
commit 9c87113
Show file tree

Hide file tree

Showing 3 changed files with 28 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,

diff --git a/...sions/ql-vscode/test/vscode-tests/minimal-workspace/local-queries/local-databases.test.ts b/...sions/ql-vscode/test/vscode-tests/minimal-workspace/local-queries/local-databases.test.ts
@@ -36,6 +36,7 @@ import {
 } from "../../../factories/databases/databases";
 import { findSourceArchive } from "../../../../src/databases/local-databases/database-resolver";
 import { LanguageContextStore } from "../../../../src/language-context-store";
+import type { LanguageClient } from "vscode-languageclient/node";
 
 describe("local databases", () => {
   let databaseManager: DatabaseManager;
@@ -109,6 +110,7 @@ describe("local databases", () => {
         resolveDatabase: resolveDatabaseSpy,
         packAdd: packAddSpy,
       }),
+      mockedObject<LanguageClient>({}),
       new LanguageContextStore(mockApp),
       mockedObject<Logger>({
         log: logSpy,

diff --git a/extensions/ql-vscode/test/vscode-tests/minimal-workspace/qlpack-generator.test.ts b/extensions/ql-vscode/test/vscode-tests/minimal-workspace/qlpack-generator.test.ts
@@ -11,6 +11,7 @@ import { mockedObject } from "../utils/mocking.helpers";
 import { ensureDir, readFile } from "fs-extra";
 import { load } from "js-yaml";
 import type { QlPackFile } from "../../../src/packaging/qlpack-file";
+import type { LanguageClient } from "vscode-languageclient/node";
 
 describe("QlPackGenerator", () => {
   let packFolderPath: string;
@@ -23,6 +24,7 @@ describe("QlPackGenerator", () => {
     typeof CodeQLCliServer.prototype.resolveQlpacks
   >;
   let mockCli: CodeQLCliServer;
+  let mockLangClient: LanguageClient;
   let dir: DirResult;
 
   beforeEach(async () => {
@@ -45,9 +47,14 @@ describe("QlPackGenerator", () => {
       resolveQlpacks: resolveQlpacksSpy,
     });
 
+    mockLangClient = mockedObject<LanguageClient>({
+      sendNotification: jest.fn().mockResolvedValue([]),
+    });
+
     generator = new QlPackGenerator(
       language as QueryLanguage,
       mockCli,
+      mockLangClient,
       packFolderPath,
       packFolderPath,
     );
@@ -131,6 +138,7 @@ describe("QlPackGenerator", () => {
       generator = new QlPackGenerator(
         language as QueryLanguage,
         mockCli,
+        mockLangClient,
         packFolderPath,
         packFolderPath,
         true,
@@ -165,6 +173,7 @@ describe("QlPackGenerator", () => {
         generator = new QlPackGenerator(
           language as QueryLanguage,
           mockCli,
+          mockLangClient,
           packFolderPath,
           packFolderPath,
           true,
@@ -200,6 +209,7 @@ describe("QlPackGenerator", () => {
         generator = new QlPackGenerator(
           language as QueryLanguage,
           mockCli,
+          mockLangClient,
           packFolderPath,
           packFolderPath,
           true,