Keep git review state fresh after reconnect (#1491)

packages/app/src/git/query-keys.test.ts

@@ -5,6 +5,7 @@ import {
   checkoutPrStatusQueryKey,
   checkoutStatusQueryKey,
   invalidateCheckoutGitQueriesForClient,
+  invalidateCheckoutGitQueriesForServer,
 } from "@/git/query-keys";
 import { prPaneTimelineQueryKey } from "@/git/pull-request-panel/query-keys";
 
@@ -59,4 +60,47 @@ describe("checkout query keys", () => {
 
     queryClient.clear();
   });
+
+  it("invalidates fetch-based checkout queries server-wide without touching other servers", async () => {
+    const queryClient = new QueryClient();
+    const otherServerId = "server-2";
+    const otherCwd = "/tmp/repo-2";
+
+    queryClient.setQueryData(checkoutStatusQueryKey(serverId, cwd), { isGit: true });
+    queryClient.setQueryData(checkoutStatusQueryKey(serverId, otherCwd), { isGit: true });
+    queryClient.setQueryData(checkoutPrStatusQueryKey(serverId, cwd), { status: { number: 12 } });
+    queryClient.setQueryData(prPaneTimelineQueryKey({ serverId, cwd, prNumber: 12 }), {
+      items: [],
+    });
+    // Subscription-fed diff queries are deliberately not part of the server-wide sweep.
+    queryClient.setQueryData(checkoutDiffQueryKey(serverId, cwd, "base", "main", true), {
+      files: [],
+    });
+    queryClient.setQueryData(checkoutStatusQueryKey(otherServerId, cwd), { isGit: true });
+
+    await invalidateCheckoutGitQueriesForServer(queryClient, serverId);
+
+    expect(queryClient.getQueryState(checkoutStatusQueryKey(serverId, cwd))?.isInvalidated).toBe(
+      true,
+    );
+    expect(
+      queryClient.getQueryState(checkoutStatusQueryKey(serverId, otherCwd))?.isInvalidated,
+    ).toBe(true);
+    expect(queryClient.getQueryState(checkoutPrStatusQueryKey(serverId, cwd))?.isInvalidated).toBe(
+      true,
+    );
+    expect(
+      queryClient.getQueryState(prPaneTimelineQueryKey({ serverId, cwd, prNumber: 12 }))
+        ?.isInvalidated,
+    ).toBe(true);
+    expect(
+      queryClient.getQueryState(checkoutDiffQueryKey(serverId, cwd, "base", "main", true))
+        ?.isInvalidated,
+    ).toBe(false);
+    expect(
+      queryClient.getQueryState(checkoutStatusQueryKey(otherServerId, cwd))?.isInvalidated,
+    ).toBe(false);
+
+    queryClient.clear();
+  });
 });