feat: expand API coverage with new tools, params, and 90% test threshold
This commit is contained in:
parent
8bb8e5507e
commit
fa913d1561
21 changed files with 1573 additions and 88 deletions
190
tests/tools/scan-request.test.ts
Normal file
190
tests/tools/scan-request.test.ts
Normal file
|
|
@ -0,0 +1,190 @@
|
|||
import { describe, it, expect, vi } from "vitest";
|
||||
import { createMockServer } from "../helpers.js";
|
||||
import { registerScanRequest } from "../../src/tools/scan-request.js";
|
||||
import { TorrentClawClient, ApiError } from "../../src/api-client.js";
|
||||
|
||||
function createMockClient(overrides: Partial<TorrentClawClient> = {}) {
|
||||
return {
|
||||
search: vi.fn(),
|
||||
autocomplete: vi.fn(),
|
||||
getPopular: vi.fn(),
|
||||
getRecent: vi.fn(),
|
||||
getWatchProviders: vi.fn(),
|
||||
getCredits: vi.fn(),
|
||||
getStats: vi.fn(),
|
||||
getTorrentDownloadUrl: vi.fn(),
|
||||
track: vi.fn(),
|
||||
submitScanRequest: vi.fn(),
|
||||
getScanStatus: vi.fn(),
|
||||
...overrides,
|
||||
} as unknown as TorrentClawClient;
|
||||
}
|
||||
|
||||
describe("submit_scan_request tool", () => {
|
||||
it("submits scan request successfully", async () => {
|
||||
const submitMock = vi.fn().mockResolvedValue({
|
||||
status: "pending",
|
||||
createdAt: "2026-01-01T00:00:00Z",
|
||||
});
|
||||
const client = createMockClient({ submitScanRequest: submitMock });
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("submit_scan_request");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
email: "test@example.com",
|
||||
});
|
||||
|
||||
expect(result.isError).toBeUndefined();
|
||||
expect(result.content[0].text).toContain("Scan request submitted");
|
||||
expect(result.content[0].text).toContain("pending");
|
||||
expect(submitMock).toHaveBeenCalledWith(
|
||||
"aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
"test@example.com",
|
||||
);
|
||||
});
|
||||
|
||||
it("returns isError on ApiError", async () => {
|
||||
const client = createMockClient({
|
||||
submitScanRequest: vi
|
||||
.fn()
|
||||
.mockRejectedValue(new ApiError(429, "Rate limited")),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("submit_scan_request");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
email: "test@example.com",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("TorrentClaw API error (429)");
|
||||
});
|
||||
|
||||
it("returns isError on generic error", async () => {
|
||||
const client = createMockClient({
|
||||
submitScanRequest: vi.fn().mockRejectedValue(new Error("Timeout")),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("submit_scan_request");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
email: "test@example.com",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("Request failed: Timeout");
|
||||
});
|
||||
|
||||
it("returns isError on non-Error throw", async () => {
|
||||
const client = createMockClient({
|
||||
submitScanRequest: vi.fn().mockRejectedValue("string error"),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("submit_scan_request");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
email: "test@example.com",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("Unknown error");
|
||||
});
|
||||
});
|
||||
|
||||
describe("get_scan_status tool", () => {
|
||||
it("returns scan status", async () => {
|
||||
const statusMock = vi.fn().mockResolvedValue({
|
||||
status: "completed",
|
||||
source: "scan_request",
|
||||
createdAt: "2026-01-01T00:00:00Z",
|
||||
completedAt: "2026-01-01T00:05:00Z",
|
||||
});
|
||||
const client = createMockClient({ getScanStatus: statusMock });
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("get_scan_status");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
});
|
||||
|
||||
expect(result.isError).toBeUndefined();
|
||||
expect(result.content[0].text).toContain("completed");
|
||||
expect(result.content[0].text).toContain("scan_request");
|
||||
});
|
||||
|
||||
it("returns minimal status when fields are missing", async () => {
|
||||
const statusMock = vi.fn().mockResolvedValue({
|
||||
status: "not_scanned",
|
||||
});
|
||||
const client = createMockClient({ getScanStatus: statusMock });
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("get_scan_status");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
});
|
||||
|
||||
expect(result.content[0].text).toContain("not_scanned");
|
||||
expect(result.content[0].text).not.toContain("Source:");
|
||||
});
|
||||
|
||||
it("returns isError on ApiError", async () => {
|
||||
const client = createMockClient({
|
||||
getScanStatus: vi
|
||||
.fn()
|
||||
.mockRejectedValue(new ApiError(404, "Not found")),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("get_scan_status");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("TorrentClaw API error (404)");
|
||||
});
|
||||
|
||||
it("returns isError on generic error", async () => {
|
||||
const client = createMockClient({
|
||||
getScanStatus: vi.fn().mockRejectedValue(new Error("DNS failure")),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("get_scan_status");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("DNS failure");
|
||||
});
|
||||
|
||||
it("returns isError on non-Error throw", async () => {
|
||||
const client = createMockClient({
|
||||
getScanStatus: vi.fn().mockRejectedValue(42),
|
||||
});
|
||||
const { server, getToolHandler } = createMockServer();
|
||||
registerScanRequest(server, client);
|
||||
|
||||
const handler = getToolHandler("get_scan_status");
|
||||
const result = await handler({
|
||||
info_hash: "aaf1e71c0a0e3b1c0f1a2b3c4d5e6f7a8b9c0d1e",
|
||||
});
|
||||
|
||||
expect(result.isError).toBe(true);
|
||||
expect(result.content[0].text).toContain("Unknown error");
|
||||
});
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue