File size: 2,438 Bytes
a070805 | 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 | import { act, renderHook } from "@testing-library/react";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { useAgentState } from "./use-agent-state";
import { useActiveConversation } from "./query/use-active-conversation";
import { useOptionalConversationId } from "./use-conversation-id";
import { useConversationStateStore } from "#/stores/conversation-state-store";
import { AgentState } from "#/types/agent-state";
import { ExecutionStatus } from "#/types/agent-server/core/base/common";
const TEST_CONVERSATION_ID = "test-conversation-id";
vi.mock("./query/use-active-conversation", () => ({
useActiveConversation: vi.fn(),
}));
vi.mock("./use-conversation-id", () => ({
useOptionalConversationId: vi.fn(),
}));
describe("useAgentState", () => {
const mockUseActiveConversation = vi.mocked(useActiveConversation);
const mockUseOptionalConversationId = vi.mocked(useOptionalConversationId);
beforeEach(() => {
act(() => {
useConversationStateStore.getState().reset();
});
mockUseActiveConversation.mockReturnValue({
data: null,
} as ReturnType<typeof useActiveConversation>);
mockUseOptionalConversationId.mockReturnValue({
conversationId: TEST_CONVERSATION_ID,
});
});
it("prefers live websocket execution status over cached conversation status", () => {
mockUseActiveConversation.mockReturnValue({
data: {
execution_status: ExecutionStatus.FINISHED,
},
} as ReturnType<typeof useActiveConversation>);
act(() => {
useConversationStateStore
.getState()
.setExecutionStatus(TEST_CONVERSATION_ID, ExecutionStatus.RUNNING);
});
const { result } = renderHook(() => useAgentState());
expect(result.current.executionStatus).toBe(ExecutionStatus.RUNNING);
expect(result.current.curAgentState).toBe(AgentState.RUNNING);
});
it("falls back to cached conversation execution status when live state is empty", () => {
mockUseActiveConversation.mockReturnValue({
data: {
execution_status: ExecutionStatus.WAITING_FOR_CONFIRMATION,
},
} as ReturnType<typeof useActiveConversation>);
const { result } = renderHook(() => useAgentState());
expect(result.current.executionStatus).toBe(
ExecutionStatus.WAITING_FOR_CONFIRMATION,
);
expect(result.current.curAgentState).toBe(
AgentState.AWAITING_USER_CONFIRMATION,
);
});
});
|