import { describe, expect, it, vi } from "vitest";
import { REFINE_REQUEST_TIMEOUT_MS, RpcClient } from "../src/modes/rpc/rpc-client.js";

type RpcClientPrivate = {
	send: (command: { type: string }, timeoutMs?: number) => Promise<unknown>;
	getData: <T>(response: unknown) => T;
};

describe("RpcClient refine", () => {
	it("sends the refine command with the extended timeout", async () => {
		const client = new RpcClient();
		const privateClient = client as unknown as RpcClientPrivate;
		const send = vi.fn(async () => ({
			type: "response",
			command: "refine",
			success: true,
			data: { id: "refine_1", appliedEdits: [], harnessStatePath: "/tmp/harness_state.json" },
		}));
		privateClient.send = send;
		privateClient.getData = <T>(response: unknown): T => {
			return (response as { data: T }).data;
		};

		const result = await client.refine({ instructions: "tighten validation" });

		expect(send).toHaveBeenCalledWith(
			{ type: "refine", instructions: "tighten validation" },
			REFINE_REQUEST_TIMEOUT_MS,
		);
		expect(REFINE_REQUEST_TIMEOUT_MS).toBeGreaterThan(30000);
		expect(result).toMatchObject({ id: "refine_1" });
	});
});