Files
2026-07-13 11:59:58 +08:00

74 lines
2.0 KiB
TypeScript

import { expect, test, rs } from "@rstest/core";
import {
buildThreadsSearchQueryOptions,
DEFAULT_THREAD_SEARCH_PARAMS,
THREAD_SEARCH_REFETCH_INTERVAL_MS,
} from "@/core/threads/thread-search-query";
import type { AgentThread } from "@/core/threads/types";
function makeThread(
threadId: string,
metadata: Record<string, unknown> = {},
): AgentThread {
return {
thread_id: threadId,
created_at: "2025-01-01T00:00:00Z",
updated_at: "2025-01-01T00:00:00Z",
metadata,
status: "idle",
values: { title: threadId, messages: [] },
} as unknown as AgentThread;
}
test("thread search query refreshes so IM-created sessions appear in the sidebar", () => {
const search = rs.fn();
const options = buildThreadsSearchQueryOptions(
{ threads: { search } },
DEFAULT_THREAD_SEARCH_PARAMS,
);
expect(options.refetchInterval).toBe(THREAD_SEARCH_REFETCH_INTERVAL_MS);
expect(options.refetchIntervalInBackground).toBe(false);
expect(options.refetchOnWindowFocus).toBe(false);
});
test("thread search hides sidecar threads from primary lists by default", async () => {
const search = rs
.fn()
.mockResolvedValue([
makeThread("primary-1"),
makeThread("sidecar-1", { deerflow_sidecar: true }),
makeThread("primary-2"),
]);
const options = buildThreadsSearchQueryOptions(
{ threads: { search } },
DEFAULT_THREAD_SEARCH_PARAMS,
);
await expect(options.queryFn()).resolves.toEqual([
makeThread("primary-1"),
makeThread("primary-2"),
]);
});
test("thread search can explicitly include sidecar threads for parent lookup", async () => {
const sidecar = makeThread("sidecar-1", {
deerflow_sidecar: true,
parent_thread_id: "parent-1",
});
const search = rs.fn().mockResolvedValue([sidecar]);
const options = buildThreadsSearchQueryOptions(
{ threads: { search } },
{
...DEFAULT_THREAD_SEARCH_PARAMS,
metadata: {
deerflow_sidecar: true,
parent_thread_id: "parent-1",
},
},
);
await expect(options.queryFn()).resolves.toEqual([sidecar]);
});