fix: make workspace policy waits selection-safe

2026-08-04 05:55:49 +02:00
parent cddd906719
commit b3ff39fc26
4 changed files with 181 additions and 32 deletions
@@ -289,6 +289,119 @@ test("immediate submit waits for a switched workspace policy before creating a s
  }));
 });

+test("initial restored workspace waits for its delayed policy before creating a session", async () => {
+  let body: unknown;
+  let releasePolicy!: () => void;
+  let policyRequestStarted = false;
+  const policyMayFinish = new Promise<void>((resolve) => { releasePolicy = resolve; });
+  localStorage.setItem("thothii.workspace-registry.v1.preferences", JSON.stringify({
+    workspaceId: "psd-clinical", provider: "deepseek", model: "deepseek-v4-pro", thinking: "medium",
+  }));
+  const revision = (id: string) => ({
+    id, commit: "a".repeat(40), blob: "b".repeat(40), snapshotPath: "/snapshot", state: "operational" as const,
+  });
+  server.use(
+    http.get("http://localhost:8787/settings", () => HttpResponse.json({ workspace: "psd-clinical" })),
+    http.get("http://localhost:8787/workspaces", () => HttpResponse.json([
+      { id: "psd-clinical", name: "psd-clinical", file: "psd-clinical.yaml", displayName: "PSD Clinical", revision: revision("psd-clinical") },
+    ])),
+    http.get("http://localhost:8787/workspaces/psd-clinical", async () => {
+      policyRequestStarted = true;
+      await policyMayFinish;
+      return HttpResponse.json({
+        workspace: { llm_policy: { default: "zai/glm-5.2", allowed: ["zai/glm-5.2"] } },
+        revision: revision("psd-clinical"),
+      });
+    }),
+    http.get("http://localhost:8787/models", () => HttpResponse.json({ models: [
+      { provider: "zai", id: "glm-5.2", name: "GLM-5.2", reasoning: true },
+      { provider: "deepseek", id: "deepseek-v4-pro", name: "DeepSeek V4 Pro", reasoning: true },
+    ] })),
+    http.post("http://localhost:8787/sessions", async ({ request }) => {
+      body = await request.json();
+      return HttpResponse.json({ id: "s1" });
+    }),
+  );
+  const client = new QueryClient({ defaultOptions: { queries: { retry: false } } });
+  render(<QueryClientProvider client={client}><ComposerFooter /><SteerInput sessionId={null} /></QueryClientProvider>);
+
+  await waitFor(() => expect(policyRequestStarted).toBe(true));
+  await userEvent.type(screen.getByRole("textbox", { name: /new question/i }), "q");
+  await userEvent.click(screen.getByRole("button", { name: /send/i }));
+
+  expect(body).toBeUndefined();
+  expect(screen.getByRole("button", { name: /send/i })).toBeDisabled();
+  releasePolicy();
+
+  await waitFor(() => expect(body).toEqual({
+    question: "q", workspaceId: "psd-clinical", provider: "zai", model: "glm-5.2", thinking: "medium",
+  }));
+});
+
+test("submit follows a rapid workspace switch instead of waiting for an abandoned policy", async () => {
+  let body: unknown;
+  let releaseC!: () => void;
+  let bPolicyRequestStarted = false;
+  let cPolicyRequestStarted = false;
+  const cPolicyMayFinish = new Promise<void>((resolve) => { releaseC = resolve; });
+  localStorage.setItem("thothii.workspace-registry.v1.preferences", JSON.stringify({
+    workspaceId: "research", provider: "deepseek", model: "deepseek-v4-pro", thinking: "medium",
+  }));
+  const revision = (id: string) => ({
+    id, commit: "a".repeat(40), blob: "b".repeat(40), snapshotPath: "/snapshot", state: "operational" as const,
+  });
+  server.use(
+    http.get("http://localhost:8787/settings", () => HttpResponse.json({ workspace: "research" })),
+    http.get("http://localhost:8787/workspaces", () => HttpResponse.json([
+      { id: "research", name: "research", file: "research.yaml", displayName: "Research", revision: revision("research") },
+      { id: "workspace-b", name: "workspace-b", file: "workspace-b.yaml", displayName: "Workspace B", revision: revision("workspace-b") },
+      { id: "workspace-c", name: "workspace-c", file: "workspace-c.yaml", displayName: "Workspace C", revision: revision("workspace-c") },
+    ])),
+    http.get("http://localhost:8787/workspaces/research", () => HttpResponse.json({
+      workspace: { llm_policy: { allowed: ["deepseek/deepseek-v4-pro"] } }, revision: revision("research"),
+    })),
+    http.get("http://localhost:8787/workspaces/workspace-b", async () => {
+      bPolicyRequestStarted = true;
+      await new Promise(() => undefined);
+      return HttpResponse.json({});
+    }),
+    http.get("http://localhost:8787/workspaces/workspace-c", async () => {
+      cPolicyRequestStarted = true;
+      await cPolicyMayFinish;
+      return HttpResponse.json({
+        workspace: { llm_policy: { default: "zai/glm-5.2", allowed: ["zai/glm-5.2"] } },
+        revision: revision("workspace-c"),
+      });
+    }),
+    http.get("http://localhost:8787/models", () => HttpResponse.json({ models: [
+      { provider: "zai", id: "glm-5.2", name: "GLM-5.2", reasoning: true },
+      { provider: "deepseek", id: "deepseek-v4-pro", name: "DeepSeek V4 Pro", reasoning: true },
+    ] })),
+    http.post("http://localhost:8787/sessions", async ({ request }) => {
+      body = await request.json();
+      return HttpResponse.json({ id: "s1" });
+    }),
+  );
+  const client = new QueryClient({ defaultOptions: { queries: { retry: false } } });
+  render(<QueryClientProvider client={client}><ComposerFooter /><SteerInput sessionId={null} /></QueryClientProvider>);
+
+  const workspaceSelector = await screen.findByRole("combobox", { name: "Workspace" });
+  await waitFor(() => expect(screen.getByRole("option", { name: "workspace-b" })).toBeInTheDocument());
+  await userEvent.selectOptions(workspaceSelector, "workspace-b");
+  await waitFor(() => expect(bPolicyRequestStarted).toBe(true));
+  await userEvent.type(screen.getByRole("textbox", { name: /new question/i }), "q");
+  await userEvent.click(screen.getByRole("button", { name: /send/i }));
+  await userEvent.selectOptions(workspaceSelector, "workspace-c");
+  await waitFor(() => expect(cPolicyRequestStarted).toBe(true));
+
+  expect(body).toBeUndefined();
+  releaseC();
+
+  await waitFor(() => expect(body).toEqual({
+    question: "q", workspaceId: "workspace-c", provider: "zai", model: "glm-5.2", thinking: "medium",
+  }));
+});
+
 test("pulses the stop dot only while the harness is working", () => {
  const { rerender } = render(<SteerInput sessionId="s1" working />);
  const dot = () =>
@@ -84,8 +84,7 @@ export function SteerInput({
    setLastUserEntry({ kind: "input", text: trimmed });
    onSessionCreating?.(trimmed);
    try {
-      await workspacePolicyGate.waitFor(workspacePreferences.load().workspaceId);
-      const { id } = await createSession({ question: trimmed, ...workspacePreferences.load() });
+      const { id } = await createSession({ question: trimmed });
      onSessionCreated?.(id);
      setText("");
    } catch (error) {
@@ -190,6 +189,11 @@ export function ComposerFooter() {
    ? models.filter((candidate) => allowedModels.includes(`${candidate.provider}/${candidate.id}`))
    : models;

+  useEffect(() => {
+    if (selectedWorkspace?.revision) workspacePolicyGate.select(workspace);
+    else workspacePolicyGate.clear();
+  }, [selectedWorkspace, workspace]);
+
  useEffect(() => {
    if (!allowedModels?.length) return;
    const selected = preferences.provider && preferences.model
@@ -222,7 +226,7 @@ export function ComposerFooter() {
      patch.workspaceId
      && patch.workspaceId !== workspace
      && workspaces.some((candidate) => candidate.id === patch.workspaceId && candidate.revision)
-    ) workspacePolicyGate.begin(patch.workspaceId);
+    ) workspacePolicyGate.select(patch.workspaceId);
    const next = workspacePreferences.save({ ...preferences, ...patch });
    setPreferences(next);
  }