chore: add tests for the actual connection status

kmruiz · kmruiz · commit 28d8b69ad28b · 2025-08-06T10:15:54.000+02:00
Also remove the code that forces the reconnection in the stdio
transport test as it's not needed anymore.
diff --git a/tests/integration/common/connectionManager.test.ts b/tests/integration/common/connectionManager.test.ts
@@ -78,6 +78,10 @@ describeWithMongoDB("Connection Manager", (integration) => {
             it("should notify that it was disconnected before connecting", () => {
                 expect(connectionManagerSpies["connection-closed"]).toHaveBeenCalled();
             });
+
+            it("should be marked explicitly as disconnected", () => {
+                expect(connectionManager().currentConnectionState.tag).toEqual("disconnected");
+            });
         });
 
         describe("when reconnects", () => {
@@ -95,6 +99,10 @@ describeWithMongoDB("Connection Manager", (integration) => {
             it("should notify that it was connected again", () => {
                 expect(connectionManagerSpies["connection-succeeded"]).toHaveBeenCalled();
             });
+
+            it("should be marked explicitly as connected", () => {
+                expect(connectionManager().currentConnectionState.tag).toEqual("connected");
+            });
         });
 
         describe("when fails to connect to a new cluster", () => {
@@ -116,6 +124,10 @@ describeWithMongoDB("Connection Manager", (integration) => {
             it("should notify that it failed connecting", () => {
                 expect(connectionManagerSpies["connection-errored"]).toHaveBeenCalled();
             });
+
+            it("should be marked explicitly as connected", () => {
+                expect(connectionManager().currentConnectionState.tag).toEqual("errored");
+            });
         });
     });
 
diff --git a/tests/integration/transports/stdio.test.ts b/tests/integration/transports/stdio.test.ts
@@ -4,13 +4,6 @@ import { Client } from "@modelcontextprotocol/sdk/client/index.js";
 import { describeWithMongoDB } from "../tools/mongodb/mongodbHelpers.js";
 
 describeWithMongoDB("StdioRunner", (integration) => {
-    beforeEach(() => {
-        integration.mcpServer().userConfig.connectionString = integration.connectionString();
-        integration.mcpServer().session.connectionManager.changeState("connection-succeeded", {
-            tag: "connected",
-        });
-    });
-
     describe("client connects successfully", () => {
         let client: Client;
         let transport: StdioClientTransport;
