fix tests

KxSystems · Apr 9, 2024 · 28ea800 · 28ea800
1 parent 54bc20d
commit 28ea800
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/test/suite/commands.test.ts b/test/suite/commands.test.ts
@@ -47,6 +47,7 @@ import { ServerDetails, ServerType } from "../../src/models/server";
 import { NewConnectionPannel } from "../../src/panels/newConnection";
 import { MAX_STR_LEN } from "../../src/validators/kdbValidator";
 import { LocalConnection } from "../../src/classes/localConnection";
+import { ConnectionManagementService } from "../../src/services/connectionManagerService";
 
 // eslint-disable-next-line @typescript-eslint/no-var-requires
 const dsCmd = require("../../src/commands/dataSourceCommand");
@@ -1577,12 +1578,12 @@ describe("serverCommand", () => {
       executeQueryStub,
       writeResultsViewStub,
       writeResultsConsoleStub: sinon.SinonStub;
+    const connMangService = new ConnectionManagementService();
     beforeEach(() => {
       ext.connection = new LocalConnection("localhost:5001", "server1");
-      ext.connection.connected = true;
       ext.connectionNode = kdbNode;
       isVisibleStub = sinon.stub(ext.resultsViewProvider, "isVisible");
-      executeQueryStub = sinon.stub(ext.connection, "executeQuery");
+      executeQueryStub = sinon.stub(connMangService, "executeQuery");
       writeResultsViewStub = sinon.stub(
         serverCommand,
         "writeQueryResultsToView",

diff --git a/test/suite/services.test.ts b/test/suite/services.test.ts
@@ -322,7 +322,7 @@ describe("kdbTreeProvider", () => {
 
     assert.strictEqual(
       kdbNode1.label,
-      "kdbnode1 [kdbserveralias] (connected)",
+      "kdbnode1 [kdbserveralias]",
       "KdbNode node creation failed",
     );
   });