import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { describe, expect, it, vi } from "vitest";
import { McpLogger } from "../../src/core/logger.js";
describe("Logger", () => {
describe("McpLogger", () => {
it("should send log messages to MCP server", () => {
const mockSendLoggingMessage = vi.fn();
const mockServer = {
server: {
sendLoggingMessage: mockSendLoggingMessage,
},
};
const logger = new McpLogger(mockServer as unknown as McpServer);
logger.log("test message");
expect(mockSendLoggingMessage).toHaveBeenCalledWith({
level: "info",
data: "test message",
});
});
it("should handle server not connected errors gracefully", () => {
const mockSendLoggingMessage = vi.fn().mockImplementation(() => {
throw new Error("Not connected");
});
const mockServer = {
server: {
sendLoggingMessage: mockSendLoggingMessage,
},
};
const logger = new McpLogger(mockServer as unknown as McpServer);
expect(() => logger.log("test message")).not.toThrow();
expect(mockSendLoggingMessage).toHaveBeenCalled();
});
});
});
