test: trim agent and discord harness partial mocks

This commit is contained in:
Peter Steinberger
2026-04-03 20:04:43 +01:00
parent 811efa2db0
commit 4578351488
4 changed files with 17 additions and 12 deletions

View File

@@ -45,9 +45,9 @@ export const DEFAULT_DISCORD_SEND_RESULT = {
export async function createDiscordSendModuleMock(
hoisted: DiscordOutboundHoisted,
importOriginal: () => Promise<DiscordSendModule>,
loadActual: () => Promise<DiscordSendModule>,
): Promise<DiscordSendModule> {
const actual = await importOriginal();
const actual = await loadActual();
return {
...actual,
sendMessageDiscord: (...args: Parameters<DiscordSendModule["sendMessageDiscord"]>) =>
@@ -72,9 +72,9 @@ export async function createDiscordSendModuleMock(
export async function createDiscordSendComponentsModuleMock(
hoisted: DiscordOutboundHoisted,
importOriginal: () => Promise<DiscordSendComponentsModule>,
loadActual: () => Promise<DiscordSendComponentsModule>,
): Promise<DiscordSendComponentsModule> {
const actual = await importOriginal();
const actual = await loadActual();
return {
...actual,
sendDiscordComponentMessage: (
@@ -89,9 +89,9 @@ export async function createDiscordSendComponentsModuleMock(
export async function createDiscordThreadBindingsModuleMock(
hoisted: DiscordOutboundHoisted,
importOriginal: () => Promise<DiscordThreadBindingsModule>,
loadActual: () => Promise<DiscordThreadBindingsModule>,
): Promise<DiscordThreadBindingsModule> {
const actual = await importOriginal();
const actual = await loadActual();
return {
...actual,
getThreadBindingManager: (

View File

@@ -3,8 +3,9 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { __testing as extraParamsTesting } from "./extra-params.js";
import { runExtraParamsCase } from "./extra-params.test-support.js";
vi.mock("@mariozechner/pi-ai", async (importOriginal) => {
const original = await importOriginal<typeof import("@mariozechner/pi-ai")>();
vi.mock("@mariozechner/pi-ai", async () => {
const original =
await vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai");
return {
...original,
streamSimple: vi.fn(() => ({

View File

@@ -3,8 +3,10 @@ import { describe, expect, it, vi } from "vitest";
import { createPiAiStreamSimpleMock } from "./extra-params.pi-ai-mock.js";
import { runExtraParamsCase } from "./extra-params.test-support.js";
vi.mock("@mariozechner/pi-ai", async (importOriginal) =>
createPiAiStreamSimpleMock(() => importOriginal<typeof import("@mariozechner/pi-ai")>()),
vi.mock("@mariozechner/pi-ai", async () =>
createPiAiStreamSimpleMock(() =>
vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai"),
),
);
function runToolPayloadCase(provider: "openai" | "xai", modelId: string) {

View File

@@ -4,8 +4,10 @@ import type { OpenClawConfig } from "../../config/config.js";
import { createPiAiStreamSimpleMock } from "./extra-params.pi-ai-mock.js";
import { runExtraParamsCase } from "./extra-params.test-support.js";
vi.mock("@mariozechner/pi-ai", async (importOriginal) =>
createPiAiStreamSimpleMock(() => importOriginal<typeof import("@mariozechner/pi-ai")>()),
vi.mock("@mariozechner/pi-ai", async () =>
createPiAiStreamSimpleMock(() =>
vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai"),
),
);
type ToolStreamCase = {