gloomberb 0.1.0 → 0.3.1

package/src/plugins/prediction-markets/plugin.test.tsx

@@ -0,0 +1,570 @@
+import { afterEach, describe, expect, test } from "bun:test";
+import { cleanupPredictionTest, createConfig, installPredictionMarketMocks } from "./test-helpers";
+import { colors } from "../../theme/colors";
+import { predictionMarketsPlugin } from "./index";
+import { resolvePredictionKeyboardCommand } from "./keyboard";
+import { buildPredictionListRows } from "./rows";
+import { getPredictionColumnValue } from "./metrics";
+import {
+  loadKalshiCatalog,
+  loadKalshiHistory,
+  normalizeKalshiMarket,
+} from "./services/kalshi-adapter";
+import {
+  loadPolymarketCatalog,
+  loadPolymarketDetail,
+  normalizePolymarketMarket,
+} from "./services/polymarket-adapter";
+
+afterEach(async () => {
+  await cleanupPredictionTest();
+});
+
+describe("prediction markets plugin registration and services", () => {
+  test("exposes the pane template and search command", () => {
+    const commands: string[] = [];
+    const ctx = {
+      persistence: {
+        getState: () => null,
+        setState: () => {},
+        deleteState: () => {},
+        getResource: () => null,
+        setResource: (_kind: string, _key: string, value: unknown) => ({
+          value,
+          fetchedAt: Date.now(),
+          staleAt: Date.now(),
+          expiresAt: Date.now(),
+          sourceKey: "test",
+          schemaVersion: 1,
+          provenance: null,
+          stale: false,
+          expired: false,
+        }),
+        deleteResource: () => {},
+      },
+      registerCommand: (command: { id: string }) => {
+        commands.push(command.id);
+      },
+      resume: {
+        getState: () => null,
+        setState: () => {},
+        deleteState: () => {},
+        getPaneState: () => null,
+        setPaneState: () => {},
+        deletePaneState: () => {},
+      },
+      focusPane: () => {},
+      getConfig: () => createConfig(),
+    } as any;
+
+    predictionMarketsPlugin.setup?.(ctx);
+
+    expect(predictionMarketsPlugin.toggleable).toBe(true);
+    expect(predictionMarketsPlugin.panes?.[0]?.defaultMode).toBe("floating");
+    expect(predictionMarketsPlugin.paneTemplates?.[0]?.shortcut?.prefix).toBe(
+      "PM",
+    );
+    expect(commands).toContain("prediction-markets-open");
+    expect(commands).toContain("prediction-markets-search");
+  });
+
+  test("normalizes venue payloads", () => {
+    expect(
+      normalizePolymarketMarket({
+        id: "pm-1",
+        question: "Will inflation fall?",
+        conditionId: "cond-1",
+        outcomes: '["Yes","No"]',
+        outcomePrices: '["0.62","0.38"]',
+        clobTokenIds: '["yes-1","no-1"]',
+        events: [
+          {
+            id: "event-1",
+            title: "US inflation",
+            openInterest: 12,
+            tags: [{ label: "Macro" }],
+          } as any,
+        ],
+      } as any)?.yesPrice,
+    ).toBe(0.62);
+
+    expect(
+      normalizeKalshiMarket({
+        ticker: "KAL-1",
+        title: "Will the Fed cut rates?",
+        yes_sub_title: "Yes",
+        status: "open",
+        market_type: "binary",
+        yes_bid_dollars: "0.47",
+        yes_ask_dollars: "0.49",
+        no_bid_dollars: "0.51",
+        no_ask_dollars: "0.53",
+        last_price_dollars: "0.48",
+        volume_24h_fp: "15000",
+        volume_fp: "90000",
+        open_interest_fp: "45000",
+        liquidity_dollars: "250000",
+      } as any)?.volume24h,
+    ).toBe(15000);
+  });
+
+  test("uses Kalshi target subtitles to disambiguate strike ladders", () => {
+    const summary = normalizeKalshiMarket(
+      {
+        ticker: "KXFED-27APR-T0.25",
+        title:
+          "Will the upper bound of the federal funds target rate be above 0.25%?",
+        yes_sub_title: "Above 0.25%",
+        event_ticker: "KXFED-27APR",
+        status: "open",
+        market_type: "binary",
+        last_price_dollars: "0.48",
+        strike_type: "greater",
+        floor_strike: "0.25",
+      } as any,
+      {
+        title: "Federal funds target rate after April 2026 FOMC",
+        sub_title: "Upper bound",
+        category: "Economics",
+        series_ticker: "KXFED",
+      },
+    );
+
+    expect(summary?.marketLabel).toBe("Above 0.25%");
+    expect(summary?.eventLabel).toBe(
+      "Federal funds target rate after April 2026 FOMC · Upper bound",
+    );
+  });
+
+  test("collapses multi-market venue events into grouped list rows", () => {
+    const rows = buildPredictionListRows([
+      normalizeKalshiMarket(
+        {
+          ticker: "KXFED-27APR-T4.25",
+          title:
+            "Will the upper bound of the federal funds target rate be above 4.25%?",
+          yes_sub_title: "Above 4.25%",
+          event_ticker: "KXFED-27APR",
+          status: "open",
+          market_type: "binary",
+          last_price_dollars: "0.48",
+          volume_24h_fp: "15000",
+          strike_type: "greater",
+          floor_strike: "4.25",
+        } as any,
+        {
+          title: "Federal funds target rate after April 2026 FOMC",
+          sub_title: "Upper bound",
+          category: "Economics",
+          series_ticker: "KXFED",
+        },
+      )!,
+      normalizeKalshiMarket(
+        {
+          ticker: "KXFED-27APR-T4.50",
+          title:
+            "Will the upper bound of the federal funds target rate be above 4.50%?",
+          yes_sub_title: "Above 4.50%",
+          event_ticker: "KXFED-27APR",
+          status: "open",
+          market_type: "binary",
+          last_price_dollars: "0.31",
+          volume_24h_fp: "12000",
+          strike_type: "greater",
+          floor_strike: "4.50",
+        } as any,
+        {
+          title: "Federal funds target rate after April 2026 FOMC",
+          sub_title: "Upper bound",
+          category: "Economics",
+          series_ticker: "KXFED",
+        },
+      )!,
+    ]);
+
+    expect(rows).toHaveLength(1);
+    expect(rows[0]?.kind).toBe("group");
+    expect(rows[0]?.focusMarketKey).toBe("kalshi:KXFED-27APR-T4.25");
+    expect(rows[0]?.focusYesPrice).toBe(0.48);
+  });
+
+  test("styles YES consistently and uses the lead contract quote on grouped rows", () => {
+    const [row] = buildPredictionListRows([
+      normalizeKalshiMarket(
+        {
+          ticker: "KXFED-27APR-T4.25",
+          title:
+            "Will the upper bound of the federal funds target rate be above 4.25%?",
+          yes_sub_title: "Above 4.25%",
+          event_ticker: "KXFED-27APR",
+          status: "open",
+          market_type: "binary",
+          last_price_dollars: "0.48",
+          volume_24h_fp: "15000",
+          strike_type: "greater",
+          floor_strike: "4.25",
+        } as any,
+        {
+          title: "Federal funds target rate after April 2026 FOMC",
+          sub_title: "Upper bound",
+          category: "Economics",
+          series_ticker: "KXFED",
+        },
+      )!,
+      normalizeKalshiMarket(
+        {
+          ticker: "KXFED-27APR-T4.50",
+          title:
+            "Will the upper bound of the federal funds target rate be above 4.50%?",
+          yes_sub_title: "Above 4.50%",
+          event_ticker: "KXFED-27APR",
+          status: "open",
+          market_type: "binary",
+          last_price_dollars: "0.31",
+          volume_24h_fp: "12000",
+          strike_type: "greater",
+          floor_strike: "4.50",
+        } as any,
+        {
+          title: "Federal funds target rate after April 2026 FOMC",
+          sub_title: "Upper bound",
+          category: "Economics",
+          series_ticker: "KXFED",
+        },
+      )!,
+    ]);
+
+    expect(
+      getPredictionColumnValue(
+        {
+          id: "yes",
+          label: "TOP ODDS",
+          width: 20,
+          align: "left",
+          description: "",
+        },
+        row!,
+        false,
+      ),
+    ).toMatchObject({
+      text: "48% Above 4.25%",
+      color: colors.negative,
+    });
+  });
+
+  test("uses an allowed Kalshi candlestick interval for 1W history", async () => {
+    let candlestickUrl: string | null = null;
+
+    globalThis.fetch = (async (input: Request | string | URL) => {
+      const url = String(input);
+      if (url.includes("/trade-api/v2/events/FED-1")) {
+        return new Response(
+          JSON.stringify({
+            event: {
+              title: "Fed series",
+              category: "Macro",
+              event_ticker: "FED-1",
+              series_ticker: "FED",
+            },
+            markets: [],
+          }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("/trade-api/v2/series/FED/markets/KAL-1/candlesticks")) {
+        candlestickUrl = url;
+        return new Response(JSON.stringify({ candlesticks: [] }), {
+          status: 200,
+        });
+      }
+      return new Response(JSON.stringify({}), { status: 200 });
+    }) as unknown as typeof fetch;
+
+    const summary = normalizeKalshiMarket({
+      ticker: "KAL-1",
+      title: "Will the Fed cut rates?",
+      yes_sub_title: "Yes",
+      event_ticker: "FED-1",
+      status: "open",
+      market_type: "binary",
+      last_price_dollars: "0.48",
+    } as any);
+
+    const history = await loadKalshiHistory(summary!, "1W");
+    expect(history).toEqual([]);
+    expect(candlestickUrl).toContain("period_interval=60");
+  });
+
+  test("normalizes keyboard commands from terminal key sequences", () => {
+    expect(resolvePredictionKeyboardCommand({ name: "esc" })).toBe("escape");
+    expect(resolvePredictionKeyboardCommand({ sequence: "j" })).toBe(
+      "move-down",
+    );
+    expect(resolvePredictionKeyboardCommand({ sequence: "\u001bOC" })).toBe(
+      "next-category",
+    );
+    expect(
+      resolvePredictionKeyboardCommand({
+        name: "right",
+        sequence: "\u001b[1;2C",
+        shift: true,
+      }),
+    ).toBe("next-venue-tab");
+  });
+
+  test("filters closed Polymarket child markets from the catalog", async () => {
+    globalThis.fetch = (async (input: Request | string | URL) => {
+      const url = String(input);
+      if (url.includes("gamma-api.polymarket.com/events?")) {
+        return new Response(
+          JSON.stringify([
+            {
+              id: "event-iran",
+              title: "US forces enter Iran by..?",
+              tags: [{ label: "Geopolitics", slug: "geopolitics" }],
+              markets: [
+                {
+                  id: "open-market",
+                  question: "US forces enter Iran by April 30?",
+                  groupItemTitle: "April 30",
+                  conditionId: "cond-open",
+                  outcomes: '["Yes","No"]',
+                  outcomePrices: '["0.55","0.45"]',
+                  clobTokenIds: '["yes-open","no-open"]',
+                  volume24hr: 2500000,
+                  spread: 0.01,
+                  active: true,
+                  closed: false,
+                },
+                {
+                  id: "closed-market",
+                  question: "US forces enter Iran by March 15?",
+                  groupItemTitle: "March 15",
+                  conditionId: "cond-closed",
+                  outcomes: '["Yes","No"]',
+                  outcomePrices: '["0.01","0.99"]',
+                  clobTokenIds: '["yes-closed","no-closed"]',
+                  volume24hr: 9999999,
+                  spread: 0.01,
+                  active: false,
+                  closed: true,
+                },
+              ],
+            },
+          ]),
+          { status: 200 },
+        );
+      }
+      return new Response(JSON.stringify({}), { status: 200 });
+    }) as unknown as typeof fetch;
+
+    const markets = await loadPolymarketCatalog("", "all");
+    expect(markets).toHaveLength(1);
+    expect(markets[0]?.marketLabel).toBe("April 30");
+  });
+
+  test("uses remote catalog endpoints for search and category changes", async () => {
+    const { fetchUrls } = installPredictionMarketMocks();
+
+    await loadPolymarketCatalog("inflation", "macro");
+    await loadKalshiCatalog("", "macro");
+
+    expect(
+      fetchUrls.some((url) =>
+        url.includes("gamma-api.polymarket.com/public-search?q=inflation"),
+      ),
+    ).toBe(true);
+    expect(
+      fetchUrls.some(
+        (url) =>
+          url.includes("/trade-api/v2/events?") &&
+          url.includes("category=Economics"),
+      ),
+    ).toBe(true);
+  });
+
+  test("falls back to thin Polymarket search results when event hydration fails", async () => {
+    let eventFetchCount = 0;
+
+    globalThis.fetch = (async (input: Request | string | URL) => {
+      const url = String(input);
+      if (url.includes("gamma-api.polymarket.com/public-search")) {
+        return new Response(
+          JSON.stringify({
+            events: [
+              {
+                id: "event-1",
+                title: "Fed decision in April?",
+                endDate: "2026-04-29T00:00:00Z",
+                markets: [
+                  {
+                    question:
+                      "Will the Fed decrease interest rates by 25 bps after the April 2026 meeting?",
+                    groupItemTitle: "25 bps decrease",
+                    slug: "will-the-fed-decrease-interest-rates-by-25-bps-after-the-april-2026-meeting",
+                    outcomes: ["Yes", "No"],
+                    outcomePrices: ["0.22", "0.78"],
+                    bestBid: 0.21,
+                    bestAsk: 0.23,
+                    lastTradePrice: 0.22,
+                    spread: 0.02,
+                    active: true,
+                    closed: false,
+                  },
+                ],
+              },
+            ],
+          }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("gamma-api.polymarket.com/events/event-1")) {
+        eventFetchCount += 1;
+        if (eventFetchCount === 1) {
+          return new Response("{}", { status: 500 });
+        }
+        return new Response(
+          JSON.stringify({
+            id: "event-1",
+            title: "Fed decision in April?",
+            description: "Canonical event",
+            resolutionSource: "FOMC",
+            openInterest: 500000,
+            tags: [{ label: "Macro", slug: "economy" }],
+            markets: [
+              {
+                id: "pm-1",
+                question:
+                  "Will the Fed decrease interest rates by 25 bps after the April 2026 meeting?",
+                groupItemTitle: "25 bps decrease",
+                conditionId: "cond-1",
+                slug: "will-the-fed-decrease-interest-rates-by-25-bps-after-the-april-2026-meeting",
+                outcomes: '["Yes","No"]',
+                outcomePrices: '["0.22","0.78"]',
+                clobTokenIds: '["yes-1","no-1"]',
+                bestBid: 0.21,
+                bestAsk: 0.23,
+                lastTradePrice: 0.22,
+                spread: 0.02,
+                volume24hr: 120000,
+                volumeNum: 900000,
+                liquidityNum: 150000,
+                active: true,
+                closed: false,
+              },
+            ],
+          }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("clob.polymarket.com/prices-history")) {
+        return new Response(
+          JSON.stringify({ history: [{ t: 1711929600, p: 0.22 }] }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("clob.polymarket.com/book?token_id=yes-1")) {
+        return new Response(
+          JSON.stringify({
+            bids: [{ price: "0.21", size: "100" }],
+            asks: [{ price: "0.23", size: "120" }],
+            last_trade_price: "0.22",
+          }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("clob.polymarket.com/book?token_id=no-1")) {
+        return new Response(
+          JSON.stringify({
+            bids: [{ price: "0.77", size: "80" }],
+            asks: [{ price: "0.79", size: "90" }],
+            last_trade_price: "0.78",
+          }),
+          { status: 200 },
+        );
+      }
+      if (url.includes("data-api.polymarket.com/trades?market=cond-1")) {
+        return new Response(
+          JSON.stringify([
+            {
+              transactionHash: "0xabc",
+              side: "BUY",
+              size: 25,
+              price: 0.22,
+              timestamp: 1712102400,
+              outcome: "Yes",
+            },
+          ]),
+          { status: 200 },
+        );
+      }
+      return new Response(JSON.stringify({}), { status: 200 });
+    }) as unknown as typeof fetch;
+
+    const searchResults = await loadPolymarketCatalog("fed", "all");
+    const detail = await loadPolymarketDetail(searchResults[0]!, "1M");
+
+    expect(searchResults[0]?.conditionId).toBeUndefined();
+    expect(detail.summary.conditionId).toBe("cond-1");
+  });
+
+  test("filters Kalshi markets locally when venue category responses bleed across buckets", async () => {
+    globalThis.fetch = (async (input: Request | string | URL) => {
+      const url = String(input);
+      if (url.includes("/trade-api/v2/events?")) {
+        return new Response(
+          JSON.stringify({
+            events: [
+              {
+                title: "Fed series",
+                sub_title: "Upper bound",
+                category: "Economics",
+                event_ticker: "FED-1",
+                series_ticker: "FED",
+                markets: [
+                  {
+                    ticker: "KAL-MACRO",
+                    title:
+                      "Will the upper bound of the federal funds target rate be above 4.25%?",
+                    yes_sub_title: "Above 4.25%",
+                    event_ticker: "FED-1",
+                    status: "open",
+                    market_type: "binary",
+                    last_price_dollars: "0.48",
+                    volume_24h_fp: "15000",
+                    strike_type: "greater",
+                    floor_strike: "4.25",
+                  },
+                ],
+              },
+              {
+                title: "NBA Finals winner",
+                category: "Sports",
+                event_ticker: "NBA-1",
+                series_ticker: "NBA",
+                markets: [
+                  {
+                    ticker: "KAL-SPORTS",
+                    title: "Will the Knicks win the title?",
+                    event_ticker: "NBA-1",
+                    status: "open",
+                    market_type: "binary",
+                    last_price_dollars: "0.35",
+                    volume_24h_fp: "12000",
+                  },
+                ],
+              },
+            ],
+          }),
+          { status: 200 },
+        );
+      }
+      return new Response(JSON.stringify({}), { status: 200 });
+    }) as unknown as typeof fetch;
+
+    const macroMarkets = await loadKalshiCatalog("", "macro");
+    expect(macroMarkets.map((market) => market.marketId)).toEqual([
+      "KAL-MACRO",
+    ]);
+  });
+});