File size: 2,417 Bytes
fd2c364 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import {
__resetActiveStoreForTests,
setActiveSelection,
setRegisteredBackends,
} from "#/api/backend-registry/active-store";
import type { Backend } from "#/api/backend-registry/types";
import AgentServerConversationService from "#/api/conversation-service/agent-server-conversation-service.api";
import {
getFetchCall,
getJsonBody,
mockJsonResponse,
} from "./fetch-test-utils";
vi.mock("@openhands/typescript-client/clients", async (importOriginal) => {
const actual =
await importOriginal<
typeof import("@openhands/typescript-client/clients")
>();
return {
...actual,
ConversationClient: vi.fn().mockImplementation(() => ({
updateConversation: vi
.fn()
.mockRejectedValue(
new Error("Cloud title updates must not use ConversationClient"),
),
})),
};
});
const cloudBackend: Backend = {
id: "prod",
name: "Production",
host: "https://app.all-hands.dev",
apiKey: "bearer-token",
kind: "cloud",
};
const originalFetch = global.fetch;
const fetchMock = vi.fn();
beforeEach(() => {
window.localStorage.clear();
__resetActiveStoreForTests();
fetchMock.mockReset();
fetchMock.mockResolvedValue(
mockJsonResponse({ id: "conv-abc", title: "Renamed conversation" }),
);
global.fetch = fetchMock as typeof fetch;
});
afterEach(() => {
window.localStorage.clear();
__resetActiveStoreForTests();
fetchMock.mockReset();
global.fetch = originalFetch;
});
describe("AgentServerConversationService.updateConversationTitle", () => {
it("PATCHes the app-conversation directly on a cloud backend", async () => {
setRegisteredBackends([cloudBackend]);
setActiveSelection({ backendId: cloudBackend.id });
const result = await AgentServerConversationService.updateConversationTitle(
"conv-abc",
"Renamed conversation",
);
expect(fetchMock).toHaveBeenCalledOnce();
const [url, init] = getFetchCall(fetchMock);
expect(url).toBe(`${cloudBackend.host}/api/v1/app-conversations/conv-abc`);
expect(init).toMatchObject({
method: "PATCH",
headers: { Authorization: "Bearer bearer-token" },
});
expect(getJsonBody(init)).toEqual({ title: "Renamed conversation" });
expect(result).toMatchObject({
id: "conv-abc",
title: "Renamed conversation",
});
});
});
|