gloomberb 0.11.3 → 0.12.0

package/src/plugins/builtin/research-search/pane.test.tsx

@@ -0,0 +1,241 @@
+import { afterEach, describe, expect, test } from "bun:test";
+import { act, useReducer } from "react";
+import { testRender } from "../../../renderers/opentui/test-utils";
+import { apiClient, setCloudApiFetchTransport } from "../../../api-client";
+import {
+  AppContext,
+  PaneInstanceProvider,
+  appReducer,
+  createInitialState,
+} from "../../../state/app/context";
+import { createTestPluginRuntime } from "../../../test-support/plugin-runtime";
+import { createDefaultConfig } from "../../../types/config";
+import { PluginRenderProvider } from "../../runtime";
+import { ResearchSearchPane } from "./pane";
+
+const PANE_ID = "research-search:test";
+
+const HIT = {
+  id: "hit-1",
+  docType: "transcript",
+  sourceId: "call-1",
+  chunkIndex: 2,
+  ticker: "AAPL",
+  publishedAt: "2026-05-02T21:00:00.000Z",
+  title: "Apple FQ2 2026 Earnings Call",
+  url: "https://example.com/call-1",
+  snippet: "we expect <mark>gross margin</mark> to expand",
+  score: 4.2,
+  metadata: { speaker: "Tim Cook", role: "CEO", isQa: false },
+};
+
+const DOCUMENT = {
+  docType: "transcript",
+  sourceId: "call-1",
+  ticker: "AAPL",
+  title: "Apple FQ2 2026 Earnings Call",
+  url: "https://example.com/call-1",
+  publishedAt: "2026-05-02T21:00:00.000Z",
+  chunks: [
+    {
+      id: "chunk-0",
+      chunkIndex: 0,
+      body: "Operator opening remarks.",
+      metadata: { speaker: "Operator" },
+    },
+    {
+      id: "chunk-2",
+      chunkIndex: 2,
+      body: "we expect gross margin to expand next quarter",
+      metadata: { speaker: "Tim Cook", role: "CEO" },
+    },
+  ],
+};
+
+let testSetup: Awaited<ReturnType<typeof testRender>> | undefined;
+
+function jsonResponse(body: unknown): Response {
+  return {
+    ok: true,
+    status: 200,
+    headers: new Headers(),
+    text: async () => JSON.stringify(body),
+  } as Response;
+}
+
+const SAVED_SEARCH = {
+  id: "saved-1",
+  name: "gross margin",
+  query: "gross margin",
+  filters: { tickers: ["AAPL"], docTypes: ["transcript"] },
+  alertEnabled: false,
+  alertChannels: [],
+  lastRunAt: null,
+  lastMatchAt: "2026-05-03T09:00:00.000Z",
+  matchCount: 4,
+  createdAt: "2026-05-01T00:00:00.000Z",
+};
+
+interface RecordedRequest {
+  path: string;
+  method: string;
+  body?: unknown;
+}
+
+function installTransport(): { requests: RecordedRequest[] } {
+  const requests: RecordedRequest[] = [];
+  setCloudApiFetchTransport(async (url, init) => {
+    const parsed = new URL(String(url));
+    requests.push({
+      path: parsed.pathname,
+      method: init?.method ?? "GET",
+      body: init?.body ? JSON.parse(String(init.body)) : undefined,
+    });
+    if (parsed.pathname === "/cloud/search/saved") {
+      return jsonResponse({ searches: [SAVED_SEARCH] });
+    }
+    if (parsed.pathname.startsWith("/cloud/search/saved/")) {
+      return jsonResponse({ search: { ...SAVED_SEARCH, alertEnabled: true } });
+    }
+    if (parsed.pathname === "/cloud/search") {
+      return jsonResponse({
+        hits: [HIT],
+        total: 1,
+        countCapped: false,
+        hasMore: false,
+        nextOffset: 1,
+        tookMs: 12,
+      });
+    }
+    if (parsed.pathname.startsWith("/cloud/search/documents/")) {
+      return jsonResponse({ document: DOCUMENT });
+    }
+    return jsonResponse({});
+  });
+  return { requests };
+}
+
+function signIn(plan: "free" | "pro" = "free"): void {
+  apiClient.setSessionToken("test-session");
+  apiClient.restoreCachedUser({
+    id: "user-1",
+    name: "Test",
+    email: "test@example.com",
+    username: "test",
+    emailVerified: true,
+    plan,
+    effectivePlan: plan,
+  } as never);
+}
+
+function Harness({ mode = "results" }: { mode?: "results" | "saved" }) {
+  const initialState = createInitialState(createDefaultConfig("/tmp/gloomberb-research-search-test"));
+  initialState.focusedPaneId = PANE_ID;
+  initialState.paneState[PANE_ID] = {
+    pluginState: { "research-search": { query: "gross margin", mode } },
+  } as never;
+  const [state, dispatch] = useReducer(appReducer, initialState);
+
+  return (
+    <AppContext value={{ state, dispatch }}>
+      <PaneInstanceProvider paneId={PANE_ID}>
+        <PluginRenderProvider pluginId="research-search" runtime={createTestPluginRuntime()}>
+          <ResearchSearchPane
+            paneId={PANE_ID}
+            paneType="research-search"
+            focused
+            width={110}
+            height={20}
+          />
+        </PluginRenderProvider>
+      </PaneInstanceProvider>
+    </AppContext>
+  );
+}
+
+async function pressKey(name: string) {
+  await act(async () => {
+    testSetup!.renderer.keyInput.emit("keypress", {
+      name,
+      ctrl: false,
+      meta: false,
+      option: false,
+      shift: false,
+      eventType: "press",
+      repeated: false,
+      preventDefault: () => {},
+      stopPropagation: () => {},
+    } as never);
+    await testSetup!.renderOnce();
+  });
+}
+
+async function renderFrames(count = 6) {
+  for (let index = 0; index < count; index += 1) {
+    await act(async () => {
+      await new Promise((resolve) => setTimeout(resolve, 5));
+      await testSetup!.renderOnce();
+    });
+  }
+}
+
+afterEach(async () => {
+  if (testSetup) {
+    await act(async () => {
+      testSetup!.renderer.destroy();
+    });
+    testSetup = undefined;
+  }
+  setCloudApiFetchTransport(null);
+  apiClient.setSessionToken(null);
+});
+
+describe("ResearchSearchPane", () => {
+  test("renders a hit with the matched terms styled instead of tagged", async () => {
+    installTransport();
+    signIn();
+
+    testSetup = await testRender(<Harness />, { width: 110, height: 20 });
+    await renderFrames();
+
+    const frame = testSetup.captureCharFrame();
+    expect(frame).toContain("AAPL");
+    expect(frame).toContain("Apple FQ2 2026 Earnings Call");
+    expect(frame).toContain("gross margin");
+    expect(frame).not.toContain("<mark>");
+  });
+
+  // Searching and reading a hit are free: the pane used to refuse to issue the
+  // request at all without Pro, which was stricter than the server ever was.
+  test("a free account searches and opens the document at the chunk that matched", async () => {
+    installTransport();
+    signIn();
+
+    testSetup = await testRender(<Harness />, { width: 110, height: 20 });
+    await renderFrames();
+
+    await pressKey("return");
+    await renderFrames();
+
+    const frame = testSetup.captureCharFrame();
+    expect(frame).toContain("Tim Cook");
+    expect(frame).toContain("to expand next quarter");
+  });
+
+  test("flips a saved-search alert and persists it", async () => {
+    const { requests } = installTransport();
+    signIn("pro");
+
+    testSetup = await testRender(<Harness mode="saved" />, { width: 110, height: 20 });
+    await renderFrames();
+    expect(testSetup.captureCharFrame()).toContain("off");
+
+    await pressKey("a");
+    await renderFrames();
+
+    const write = requests.find((request) => request.method === "PATCH");
+    expect(write?.path).toBe("/cloud/search/saved/saved-1");
+    expect(write?.body).toEqual({ alertEnabled: true });
+    expect(testSetup.captureCharFrame()).toContain("on");
+  });
+});