mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-07-31 12:22:14 +03:00
47468636e4
Unify audit access under the logs experience and add richer active request visibility with sanitized client and provider payload previews. Expose provider warnings from upstream responses in compliance audit logs, surface learned rate-limit header data in health monitoring, and add MCP cache stats and cache flush tools with test coverage. Also add local provider catalog support for SD WebUI and ComfyUI, include video generation in endpoint docs and UI, add eval run storage migrations, and refresh docs and translations to match the expanded feature set.
63 lines
2.2 KiB
TypeScript
63 lines
2.2 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
|
|
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
import {
|
|
MCP_TOOLS,
|
|
MCP_TOOL_MAP,
|
|
cacheStatsInput,
|
|
cacheFlushInput,
|
|
cacheStatsTool,
|
|
cacheFlushTool,
|
|
} from "../schemas/tools.ts";
|
|
import { createMcpServer } from "../server.ts";
|
|
|
|
vi.mock("../audit.ts", () => ({
|
|
logToolCall: vi.fn().mockResolvedValue(undefined),
|
|
closeAuditDb: vi.fn(),
|
|
}));
|
|
|
|
describe("cache MCP tools", () => {
|
|
it("should be registered in MCP_TOOLS and MCP_TOOL_MAP", () => {
|
|
expect(MCP_TOOLS.find((tool) => tool.name === "omniroute_cache_stats")).toBeDefined();
|
|
expect(MCP_TOOLS.find((tool) => tool.name === "omniroute_cache_flush")).toBeDefined();
|
|
expect(MCP_TOOL_MAP.omniroute_cache_stats).toBeDefined();
|
|
expect(MCP_TOOL_MAP.omniroute_cache_flush).toBeDefined();
|
|
});
|
|
|
|
it("should validate cache tool input schemas", () => {
|
|
expect(cacheStatsInput.safeParse({}).success).toBe(true);
|
|
expect(cacheFlushInput.safeParse({ model: "openai/gpt-4.1" }).success).toBe(true);
|
|
expect(cacheFlushInput.safeParse({ signature: "sig_123" }).success).toBe(true);
|
|
});
|
|
|
|
it("should expose the correct cache scopes", () => {
|
|
expect(cacheStatsTool.scopes).toContain("read:cache");
|
|
expect(cacheFlushTool.scopes).toContain("write:cache");
|
|
});
|
|
});
|
|
|
|
describe("cache MCP tools registration", () => {
|
|
let client: Client;
|
|
|
|
beforeEach(async () => {
|
|
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
const server = createMcpServer();
|
|
await server.connect(serverTransport);
|
|
client = new Client({ name: "cache-tools-test", version: "1.0.0" });
|
|
await client.connect(clientTransport);
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await client.close();
|
|
});
|
|
|
|
it("should appear in tools/list after registration", async () => {
|
|
const { tools } = await client.listTools();
|
|
const names = tools.map((tool) => tool.name);
|
|
|
|
expect(names).toContain("omniroute_cache_stats");
|
|
expect(names).toContain("omniroute_cache_flush");
|
|
expect(tools.length).toBeGreaterThanOrEqual(29);
|
|
});
|
|
});
|