4-10 树域 Rust 家族化

wolai-frontend/src/components/sidebar/use-preferred-sidebar-snapshot.test.tsx

@@ -39,6 +39,7 @@ function Harness(props: {
   initialData: SidebarInitialData;
   sidebarQueryData: SidebarInitialData;
   treeStreamData: SidebarInitialData | null;
+  treeStreamStatus?: "idle" | "connecting" | "live" | "fallback";
   onState: (state: ReturnType<typeof usePreferredSidebarSnapshot>) => void;
 }) {
   const state = usePreferredSidebarSnapshot(props);
@@ -71,7 +72,55 @@ describe("usePreferredSidebarSnapshot", () => {
     container.remove();
   });
 
-  it("tree stream 落后于 query refetch 时应优先使用更新后的 query 快照", async () => {
+  it("tree stream live 时即使 query 更新也应继续优先使用 stream 快照", async () => {
+    const initialData = buildSidebarData([buildDocument({ title: "标题 A" })]);
+    const staleTreeStream = buildSidebarData([buildDocument({ title: "标题 A" })]);
+    const refreshedQuery = buildSidebarData([
+      buildDocument({
+        title: "标题 B",
+        updated_at: "2026-04-21T00:00:01.000Z",
+      }),
+    ]);
+    await act(async () => {
+      root.render(
+        <Harness
+          initialData={initialData}
+          sidebarQueryData={initialData}
+          treeStreamData={staleTreeStream}
+          treeStreamStatus="live"
+          onState={onState}
+        />,
+      );
+    });
+
+    expect(onState.mock.lastCall?.[0]).toMatchObject({
+      source: "tree_stream",
+      data: expect.objectContaining({
+        kernelSidebarTree: [expect.objectContaining({ title: "标题 A" })],
+      }),
+    });
+
+    await act(async () => {
+      root.render(
+        <Harness
+          initialData={initialData}
+          sidebarQueryData={refreshedQuery}
+          treeStreamData={staleTreeStream}
+          treeStreamStatus="live"
+          onState={onState}
+        />,
+      );
+    });
+
+    expect(onState.mock.lastCall?.[0]).toMatchObject({
+      source: "tree_stream",
+      data: expect.objectContaining({
+        kernelSidebarTree: [expect.objectContaining({ title: "标题 A" })],
+      }),
+    });
+  });
+
+  it("tree stream 进入 fallback 后应回退到 query 快照", async () => {
     const initialData = buildSidebarData([buildDocument({ title: "标题 A" })]);
     const staleTreeStream = buildSidebarData([buildDocument({ title: "标题 A" })]);
     const refreshedQuery = buildSidebarData([
@@ -87,30 +136,13 @@ describe("usePreferredSidebarSnapshot", () => {
       }),
     ]);
 
-    await act(async () => {
-      root.render(
-        <Harness
-          initialData={initialData}
-          sidebarQueryData={initialData}
-          treeStreamData={staleTreeStream}
-          onState={onState}
-        />,
-      );
-    });
-
-    expect(onState.mock.lastCall?.[0]).toMatchObject({
-      source: "tree_stream",
-      data: expect.objectContaining({
-        kernelSidebarTree: [expect.objectContaining({ title: "标题 A" })],
-      }),
-    });
-
     await act(async () => {
       root.render(
         <Harness
           initialData={initialData}
           sidebarQueryData={refreshedQuery}
           treeStreamData={staleTreeStream}
+          treeStreamStatus="fallback"
           onState={onState}
         />,
       );
@@ -129,6 +161,7 @@ describe("usePreferredSidebarSnapshot", () => {
           initialData={initialData}
           sidebarQueryData={refreshedQuery}
           treeStreamData={caughtUpTreeStream}
+          treeStreamStatus="live"
           onState={onState}
         />,
       );