refactor(test): dedupe bluebubbles monitor helpers

2026-04-24 23:21:30 +00:00 · 2026-03-22 02:11:18 +00:00
parent 7b344b8a8a
commit 4f210e98a5
4 changed files with 427 additions and 318 deletions
--- a/test/helpers/extensions/bluebubbles-monitor.ts
+++ b/test/helpers/extensions/bluebubbles-monitor.ts
@@ -0,0 +1,143 @@
+import { vi } from "vitest";
+import type { BlueBubblesHistoryFetchResult } from "../../../extensions/bluebubbles/src/history.js";
+import { _resetBlueBubblesShortIdState } from "../../../extensions/bluebubbles/src/monitor.js";
+import type { PluginRuntime } from "../../../extensions/bluebubbles/src/runtime-api.js";
+import { setBlueBubblesRuntime } from "../../../extensions/bluebubbles/src/runtime.js";
+import { createPluginRuntimeMock } from "./plugin-runtime-mock.js";
+
+export type DispatchReplyParams = Parameters<
+  PluginRuntime["channel"]["reply"]["dispatchReplyWithBufferedBlockDispatcher"]
+>[0];
+
+export const EMPTY_DISPATCH_RESULT = {
+  queuedFinal: false,
+  counts: { tool: 0, block: 0, final: 0 },
+} as const;
+
+type BlueBubblesMonitorTestRuntimeMocks = {
+  enqueueSystemEvent: unknown;
+  chunkMarkdownText: unknown;
+  chunkByNewline: unknown;
+  chunkMarkdownTextWithMode: unknown;
+  chunkTextWithMode: unknown;
+  resolveChunkMode: unknown;
+  hasControlCommand: unknown;
+  dispatchReplyWithBufferedBlockDispatcher: unknown;
+  formatAgentEnvelope: unknown;
+  formatInboundEnvelope: unknown;
+  resolveEnvelopeFormatOptions: unknown;
+  resolveAgentRoute: unknown;
+  buildPairingReply: unknown;
+  readAllowFromStore: unknown;
+  upsertPairingRequest: unknown;
+  saveMediaBuffer: unknown;
+  resolveStorePath: unknown;
+  readSessionUpdatedAt: unknown;
+  buildMentionRegexes: unknown;
+  matchesMentionPatterns: unknown;
+  matchesMentionWithExplicit: unknown;
+  resolveGroupPolicy: unknown;
+  resolveRequireMention: unknown;
+  resolveCommandAuthorizedFromAuthorizers: unknown;
+};
+
+export function createBlueBubblesMonitorTestRuntime(
+  mocks: BlueBubblesMonitorTestRuntimeMocks,
+): PluginRuntime {
+  return createPluginRuntimeMock({
+    system: {
+      enqueueSystemEvent: mocks.enqueueSystemEvent as PluginRuntime["system"]["enqueueSystemEvent"],
+    },
+    channel: {
+      text: {
+        chunkMarkdownText:
+          mocks.chunkMarkdownText as PluginRuntime["channel"]["text"]["chunkMarkdownText"],
+        chunkByNewline: mocks.chunkByNewline as PluginRuntime["channel"]["text"]["chunkByNewline"],
+        chunkMarkdownTextWithMode:
+          mocks.chunkMarkdownTextWithMode as PluginRuntime["channel"]["text"]["chunkMarkdownTextWithMode"],
+        chunkTextWithMode:
+          mocks.chunkTextWithMode as PluginRuntime["channel"]["text"]["chunkTextWithMode"],
+        resolveChunkMode:
+          mocks.resolveChunkMode as PluginRuntime["channel"]["text"]["resolveChunkMode"],
+        hasControlCommand:
+          mocks.hasControlCommand as PluginRuntime["channel"]["text"]["hasControlCommand"],
+      },
+      reply: {
+        dispatchReplyWithBufferedBlockDispatcher:
+          mocks.dispatchReplyWithBufferedBlockDispatcher as PluginRuntime["channel"]["reply"]["dispatchReplyWithBufferedBlockDispatcher"],
+        formatAgentEnvelope:
+          mocks.formatAgentEnvelope as PluginRuntime["channel"]["reply"]["formatAgentEnvelope"],
+        formatInboundEnvelope:
+          mocks.formatInboundEnvelope as PluginRuntime["channel"]["reply"]["formatInboundEnvelope"],
+        resolveEnvelopeFormatOptions:
+          mocks.resolveEnvelopeFormatOptions as PluginRuntime["channel"]["reply"]["resolveEnvelopeFormatOptions"],
+      },
+      routing: {
+        resolveAgentRoute:
+          mocks.resolveAgentRoute as PluginRuntime["channel"]["routing"]["resolveAgentRoute"],
+      },
+      pairing: {
+        buildPairingReply:
+          mocks.buildPairingReply as PluginRuntime["channel"]["pairing"]["buildPairingReply"],
+        readAllowFromStore:
+          mocks.readAllowFromStore as PluginRuntime["channel"]["pairing"]["readAllowFromStore"],
+        upsertPairingRequest:
+          mocks.upsertPairingRequest as PluginRuntime["channel"]["pairing"]["upsertPairingRequest"],
+      },
+      media: {
+        saveMediaBuffer:
+          mocks.saveMediaBuffer as PluginRuntime["channel"]["media"]["saveMediaBuffer"],
+      },
+      session: {
+        resolveStorePath:
+          mocks.resolveStorePath as PluginRuntime["channel"]["session"]["resolveStorePath"],
+        readSessionUpdatedAt:
+          mocks.readSessionUpdatedAt as PluginRuntime["channel"]["session"]["readSessionUpdatedAt"],
+      },
+      mentions: {
+        buildMentionRegexes:
+          mocks.buildMentionRegexes as PluginRuntime["channel"]["mentions"]["buildMentionRegexes"],
+        matchesMentionPatterns:
+          mocks.matchesMentionPatterns as PluginRuntime["channel"]["mentions"]["matchesMentionPatterns"],
+        matchesMentionWithExplicit:
+          mocks.matchesMentionWithExplicit as PluginRuntime["channel"]["mentions"]["matchesMentionWithExplicit"],
+      },
+      groups: {
+        resolveGroupPolicy:
+          mocks.resolveGroupPolicy as PluginRuntime["channel"]["groups"]["resolveGroupPolicy"],
+        resolveRequireMention:
+          mocks.resolveRequireMention as PluginRuntime["channel"]["groups"]["resolveRequireMention"],
+      },
+      commands: {
+        resolveCommandAuthorizedFromAuthorizers:
+          mocks.resolveCommandAuthorizedFromAuthorizers as PluginRuntime["channel"]["commands"]["resolveCommandAuthorizedFromAuthorizers"],
+      },
+    },
+  });
+}
+
+export function resetBlueBubblesMonitorTestState(params: {
+  createRuntime: () => PluginRuntime;
+  fetchHistoryMock: { mockResolvedValue: (value: BlueBubblesHistoryFetchResult) => unknown };
+  readAllowFromStoreMock: { mockResolvedValue: (value: string[]) => unknown };
+  upsertPairingRequestMock: {
+    mockResolvedValue: (value: { code: string; created: boolean }) => unknown;
+  };
+  resolveRequireMentionMock: { mockReturnValue: (value: boolean) => unknown };
+  hasControlCommandMock: { mockReturnValue: (value: boolean) => unknown };
+  resolveCommandAuthorizedFromAuthorizersMock: { mockReturnValue: (value: boolean) => unknown };
+  buildMentionRegexesMock: { mockReturnValue: (value: RegExp[]) => unknown };
+  extraReset?: () => void;
+}) {
+  vi.clearAllMocks();
+  _resetBlueBubblesShortIdState();
+  params.extraReset?.();
+  params.fetchHistoryMock.mockResolvedValue({ entries: [], resolved: true });
+  params.readAllowFromStoreMock.mockResolvedValue([]);
+  params.upsertPairingRequestMock.mockResolvedValue({ code: "TESTCODE", created: true });
+  params.resolveRequireMentionMock.mockReturnValue(false);
+  params.hasControlCommandMock.mockReturnValue(false);
+  params.resolveCommandAuthorizedFromAuthorizersMock.mockReturnValue(false);
+  params.buildMentionRegexesMock.mockReturnValue([/\bbert\b/i]);
+  setBlueBubblesRuntime(params.createRuntime());
+}