update

Author: blva

src/index.ts

@@ -1,7 +1,7 @@
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { Server } from "./server.js";
 
-async function runServer() {
+export async function runServer() {
     const server = new Server();
 
     const transport = new StdioServerTransport();

tests/unit/index.test.ts

@@ -1,37 +1,29 @@
 import { jest, describe, it, expect, beforeEach } from "@jest/globals";
-import type { Transport } from "@modelcontextprotocol/sdk/shared/transport";
+import {runServer} from "../../src/index";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio";
 
-// Mock modules with proper typing
-jest.mock("@modelcontextprotocol/sdk/server/stdio", () => ({
-    StdioServerTransport: jest.fn().mockImplementation(() => ({}) as Transport),
-}));
 
-// Properly type the mock function to return Promise<void>
-const mockConnect = jest.fn<() => Promise<void>>().mockResolvedValue();
-jest.mock("../../src/server", () => ({
-    Server: jest.fn().mockImplementation(() => ({
-        state: undefined,
-        apiClient: undefined,
-        initialized: false,
-        connect: mockConnect,
-    })),
-}));
+// mock the StdioServerTransport
+jest.mock('@modelcontextprotocol/sdk/server/stdio.js');
+// mock Server class and its methods
+jest.mock('../../src/server.ts', () => {
+    return {
+        Server: jest.fn().mockImplementation(() => {
+            return {
+                connect: jest.fn().mockImplementation((transport) => {
+                    return new Promise((resolve) => {
+                        resolve(transport);
+                    });
+                }),
+            };
+        }),
+    };
+});
 
-describe("Server initialization", () => {
-    beforeEach(() => {
-        jest.clearAllMocks();
-        jest.resetModules();
-    });
 
+describe("Server initialization", () => {
     it("should create a server instance", async () => {
-        const { Server } = await import("../../src/server");
-        const { StdioServerTransport } = await import("@modelcontextprotocol/sdk/server/stdio");
-
-        // Import the module under test
-        await import("../../src/index");
-
-        expect(Server).toHaveBeenCalledTimes(1);
-        expect(StdioServerTransport).toHaveBeenCalledTimes(1);
-        expect(mockConnect).toHaveBeenCalledTimes(1);
+        const server = await runServer();
+        expect(StdioServerTransport).toHaveBeenCalled();
     });
 });