File size: 2,342 Bytes
3201ca6 | 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 | import { describe, expect, it, vi, beforeEach, afterEach } from "vitest";
import { renderHook } from "@testing-library/react";
import { useHasAttachedSource } from "#/hooks/use-has-attached-source";
import { setStoredConversationMetadata } from "#/api/conversation-metadata-store";
const useActiveConversationMock = vi.fn();
vi.mock("#/hooks/query/use-active-conversation", () => ({
useActiveConversation: () => useActiveConversationMock(),
}));
describe("useHasAttachedSource", () => {
beforeEach(() => {
window.localStorage.clear();
useActiveConversationMock.mockReset();
});
afterEach(() => {
window.localStorage.clear();
});
it("returns true when the conversation has a selected_repository", () => {
useActiveConversationMock.mockReturnValue({
data: {
id: "conv-1",
selected_repository: "octocat/hello-world",
},
isLoading: false,
});
const { result } = renderHook(() => useHasAttachedSource());
expect(result.current.hasAttachedSource).toBe(true);
expect(result.current.isLoading).toBe(false);
});
it("returns true when the conversation has a stored selected_workspace (no repo)", () => {
setStoredConversationMetadata("conv-2", {
selected_repository: null,
selected_branch: null,
git_provider: null,
selected_workspace: "/home/me/code/foo",
});
useActiveConversationMock.mockReturnValue({
data: { id: "conv-2", selected_repository: null },
isLoading: false,
});
const { result } = renderHook(() => useHasAttachedSource());
expect(result.current.hasAttachedSource).toBe(true);
});
it("returns false when neither a repo nor a workspace is attached", () => {
useActiveConversationMock.mockReturnValue({
data: { id: "conv-3", selected_repository: null },
isLoading: false,
});
const { result } = renderHook(() => useHasAttachedSource());
expect(result.current.hasAttachedSource).toBe(false);
});
it("propagates the active-conversation isLoading flag", () => {
useActiveConversationMock.mockReturnValue({
data: undefined,
isLoading: true,
});
const { result } = renderHook(() => useHasAttachedSource());
expect(result.current.hasAttachedSource).toBe(false);
expect(result.current.isLoading).toBe(true);
});
});
|