gloomberb 0.5.0 → 0.6.0

package/src/plugins/builtin/ai/screener-pane.test.tsx

@@ -1,14 +1,17 @@
 import { afterEach, describe, expect, test } from "bun:test";
 import { act, useReducer, useState } from "react";
-import { testRender } from "@opentui/react/test-utils";
+import { testRender } from "../../../renderers/opentui/test-utils";
 import { DialogProvider } from "@opentui-ui/dialog/react";
+import { PaneFooterBar, PaneFooterProvider } from "../../../components/layout/pane-footer";
 import { AppContext, PaneInstanceProvider, appReducer, createInitialState } from "../../../state/app-context";
 import { createDefaultConfig } from "../../../types/config";
 import type { Quote, TickerFinancials } from "../../../types/financials";
 import type { TickerRecord } from "../../../types/ticker";
 import { createTestDataProvider } from "../../../test-support/data-provider";
+import { createStatefulTestPluginRuntime } from "../../../test-support/plugin-runtime";
+import { Box } from "../../../ui";
 import { PluginRenderProvider, type PluginRuntimeAccess } from "../../plugin-runtime";
-import { setSharedDataProviderForTests, setSharedRegistryForTests } from "../../registry";
+import { getSharedMarketData, setSharedMarketDataForTests, setSharedRegistryForTests } from "../../registry";
 import { AiScreenerPane } from "./screener-pane";
 import { __setDetectedProvidersForTests, type AiProvider } from "./providers";
 
@@ -17,40 +20,9 @@ const PANE_ID = "ai-screener:test";
 let testSetup: Awaited<ReturnType<typeof testRender>> | undefined;
 
 function makeRuntime(): PluginRuntimeAccess {
-  const resumeState = new Map<string, unknown>();
-  const listeners = new Map<string, Set<() => void>>();
-
-  return {
-    pinTicker() {},
-    navigateTicker() {},
-    subscribeResumeState(pluginId, key, listener) {
-      const storeKey = `${pluginId}:${key}`;
-      if (!listeners.has(storeKey)) listeners.set(storeKey, new Set());
-      listeners.get(storeKey)!.add(listener);
-      return () => listeners.get(storeKey)?.delete(listener);
-    },
-    getResumeState(pluginId, key) {
-      return (resumeState.get(`${pluginId}:${key}`) as any) ?? null;
-    },
-    setResumeState(pluginId, key, value) {
-      const storeKey = `${pluginId}:${key}`;
-      resumeState.set(storeKey, value);
-      for (const listener of listeners.get(storeKey) ?? []) listener();
-    },
-    deleteResumeState(pluginId, key) {
-      const storeKey = `${pluginId}:${key}`;
-      resumeState.delete(storeKey);
-      for (const listener of listeners.get(storeKey) ?? []) listener();
-    },
-    getConfigState() {
-      return null;
-    },
-    async setConfigState() {},
-    async deleteConfigState() {},
-    getConfigStateKeys() {
-      return [];
-    },
-  };
+  return createStatefulTestPluginRuntime({
+    getMarketData: () => getSharedMarketData() ?? null,
+  });
 }
 
 function makeTicker(symbol: string, name: string): TickerRecord {
@@ -152,7 +124,14 @@ function ScreenerHarness({
       <DialogProvider dialogOptions={{ style: { backgroundColor: "#000000", borderColor: "#ffffff", borderStyle: "single" } }}>
         <PaneInstanceProvider paneId={PANE_ID}>
           <PluginRenderProvider pluginId="ai" runtime={runtime}>
-            <AiScreenerPane paneId={PANE_ID} paneType="ai-screener" focused width={96} height={18} />
+            <PaneFooterProvider>
+              {(footer) => (
+                <Box flexDirection="column" width={96} height={18}>
+                  <AiScreenerPane paneId={PANE_ID} paneType="ai-screener" focused width={96} height={17} />
+                  <PaneFooterBar footer={footer} focused width={96} />
+                </Box>
+              )}
+            </PaneFooterProvider>
           </PluginRenderProvider>
         </PaneInstanceProvider>
       </DialogProvider>
@@ -177,7 +156,7 @@ async function waitForFrameToContain(text: string, attempts = 30): Promise<strin
 afterEach(() => {
   __setDetectedProvidersForTests(null);
   setSharedRegistryForTests(undefined);
-  setSharedDataProviderForTests(undefined);
+  setSharedMarketDataForTests(undefined);
   if (testSetup) {
     testSetup.renderer.destroy();
     testSetup = undefined;
@@ -193,7 +172,7 @@ describe("AiScreenerPane", () => {
     __setDetectedProvidersForTests([provider]);
 
     const dataProvider = createTestDataProvider();
-    setSharedDataProviderForTests(dataProvider);
+    setSharedMarketDataForTests(dataProvider);
     setSharedRegistryForTests({
       dataProvider,
       tickerRepository: {
@@ -202,7 +181,7 @@ describe("AiScreenerPane", () => {
         saveTicker: async () => {},
       },
       events: { emit() {} },
-      pinTickerFn() {},
+      pinTicker() {},
     } as any);
 
     testSetup = await testRender(<ScreenerHarness prompt="Find quality compounders." providerId="shell" />, {
@@ -210,10 +189,11 @@ describe("AiScreenerPane", () => {
       height: 18,
     });
 
-    const frame = await waitForFrameToContain("AAPL");
+    await waitForFrameToContain("AAPL");
+    const frame = await waitForFrameToContain("Initial pass");
     expect(frame).toContain("Compounders");
     expect(frame).toContain("Initial pass");
-    expect(frame).toContain("1 tickers");
+    expect(frame).toContain("Strong cash flow durability");
   });
 
   test("merges new unique results on a normal refresh", async () => {
@@ -224,7 +204,7 @@ describe("AiScreenerPane", () => {
     __setDetectedProvidersForTests([provider]);
 
     const dataProvider = createTestDataProvider();
-    setSharedDataProviderForTests(dataProvider);
+    setSharedMarketDataForTests(dataProvider);
     setSharedRegistryForTests({
       dataProvider,
       tickerRepository: {
@@ -233,7 +213,7 @@ describe("AiScreenerPane", () => {
         saveTicker: async () => {},
       },
       events: { emit() {} },
-      pinTickerFn() {},
+      pinTicker() {},
     } as any);
 
     testSetup = await testRender(<ScreenerHarness prompt="Find quality compounders." providerId="shell" />, {
@@ -248,9 +228,10 @@ describe("AiScreenerPane", () => {
       await testSetup!.renderOnce();
     });
 
-    const frame = await waitForFrameToContain("MSFT");
+    await waitForFrameToContain("MSFT");
+    const frame = await waitForFrameToContain("Second pass");
     expect(frame).toContain("AAPL");
-    expect(frame).toContain("2 tickers");
+    expect(frame).toContain("MSFT");
   });
 
   test("opens the inline prompt editor without using the dialog flow", async () => {
@@ -261,7 +242,7 @@ describe("AiScreenerPane", () => {
     __setDetectedProvidersForTests([provider]);
 
     const dataProvider = createTestDataProvider();
-    setSharedDataProviderForTests(dataProvider);
+    setSharedMarketDataForTests(dataProvider);
     setSharedRegistryForTests({
       dataProvider,
       tickerRepository: {
@@ -270,7 +251,7 @@ describe("AiScreenerPane", () => {
         saveTicker: async () => {},
       },
       events: { emit() {} },
-      pinTickerFn() {},
+      pinTicker() {},
     } as any);
 
     testSetup = await testRender(<ScreenerHarness prompt="Find quality compounders." providerId="shell" />, {
@@ -285,7 +266,7 @@ describe("AiScreenerPane", () => {
       await testSetup!.renderOnce();
     });
 
-    const frame = testSetup.captureCharFrame();
+    const frame = await waitForFrameToContain("Save");
     expect(frame).toContain("Save");
     expect(frame).toContain("Cancel");
     expect(frame).toContain("Find quality compounders.");
@@ -299,7 +280,7 @@ describe("AiScreenerPane", () => {
     __setDetectedProvidersForTests([provider]);
 
     const dataProvider = createTestDataProvider();
-    setSharedDataProviderForTests(dataProvider);
+    setSharedMarketDataForTests(dataProvider);
     setSharedRegistryForTests({
       dataProvider,
       tickerRepository: {
@@ -308,7 +289,7 @@ describe("AiScreenerPane", () => {
         saveTicker: async () => {},
       },
       events: { emit() {} },
-      pinTickerFn() {},
+      pinTicker() {},
     } as any);
 
     testSetup = await testRender(<ScreenerHarness prompt="Find quality compounders." providerId="shell" />, {
@@ -323,9 +304,10 @@ describe("AiScreenerPane", () => {
       await testSetup!.renderOnce();
     });
 
-    const frame = await waitForFrameToContain("MSFT");
+    await waitForFrameToContain("MSFT");
+    const frame = await waitForFrameToContain("Second pass");
     expect(frame).not.toContain("AAPL");
-    expect(frame).toContain("1 tickers");
+    expect(frame).toContain("MSFT");
   });
 
   test("keeps the last good results visible when a rerun returns invalid JSON", async () => {
@@ -336,7 +318,7 @@ describe("AiScreenerPane", () => {
     __setDetectedProvidersForTests([provider]);
 
     const dataProvider = createTestDataProvider();
-    setSharedDataProviderForTests(dataProvider);
+    setSharedMarketDataForTests(dataProvider);
     setSharedRegistryForTests({
       dataProvider,
       tickerRepository: {
@@ -345,7 +327,7 @@ describe("AiScreenerPane", () => {
         saveTicker: async () => {},
       },
       events: { emit() {} },
-      pinTickerFn() {},
+      pinTicker() {},
     } as any);
 
     testSetup = await testRender(<ScreenerHarness prompt="Find quality compounders." providerId="shell" />, {