vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-04 14:40:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ci): restore local check suite

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-17 08:13:53 +00:00
parent 990d0d7261
commit 6bf07b5075
13 changed files with 61 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/agents/pi-embedded-runner/run/compaction-retry-aggregate-timeout.test.ts
View File

@@ -25,14 +25,18 @@ function expectClearedTimeoutState(onTimeout: ReturnType<typeof vi.fn>, timedOut
function buildAggregateTimeoutParams(
overrides: Partial<AggregateTimeoutParams> &
Pick<AggregateTimeoutParams, "waitForCompactionRetry">,
): AggregateTimeoutParams & { onTimeout: ReturnType<typeof vi.fn> } {
const onTimeout = overrides.onTimeout ?? vi.fn();
): { params: AggregateTimeoutParams; onTimeoutSpy: ReturnType<typeof vi.fn> } {
const onTimeoutSpy = vi.fn();
const onTimeout = overrides.onTimeout ?? (() => onTimeoutSpy());
return {
waitForCompactionRetry: overrides.waitForCompactionRetry,
abortable: overrides.abortable ?? (async (promise) => await promise),
aggregateTimeoutMs: overrides.aggregateTimeoutMs ?? 60_000,
isCompactionStillInFlight: overrides.isCompactionStillInFlight,
onTimeout,
params: {
waitForCompactionRetry: overrides.waitForCompactionRetry,
abortable: overrides.abortable ?? (async (promise) => await promise),
aggregateTimeoutMs: overrides.aggregateTimeoutMs ?? 60_000,
isCompactionStillInFlight: overrides.isCompactionStillInFlight,
onTimeout,
},
onTimeoutSpy,
};
}
@@ -40,7 +44,7 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
it("times out and fires callback when compaction retry never resolves", async () => {
await withFakeTimers(async () => {
const waitForCompactionRetry = vi.fn(async () => await new Promise<void>(() => {}));
const params = buildAggregateTimeoutParams({ waitForCompactionRetry });
const { params, onTimeoutSpy } = buildAggregateTimeoutParams({ waitForCompactionRetry });
const resultPromise = waitForCompactionRetryWithAggregateTimeout(params);
@@ -48,7 +52,7 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
const result = await resultPromise;
expect(result.timedOut).toBe(true);
expectClearedTimeoutState(params.onTimeout, true);
expectClearedTimeoutState(onTimeoutSpy, true);
});
});
@@ -68,14 +72,15 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
waitForCompactionRetry,
isCompactionStillInFlight: () => compactionInFlight,
});
const { params: aggregateTimeoutParams, onTimeoutSpy } = params;
const resultPromise = waitForCompactionRetryWithAggregateTimeout(params);
const resultPromise = waitForCompactionRetryWithAggregateTimeout(aggregateTimeoutParams);
await vi.advanceTimersByTimeAsync(170_000);
const result = await resultPromise;
expect(result.timedOut).toBe(false);
expectClearedTimeoutState(params.onTimeout, false);
expectClearedTimeoutState(onTimeoutSpy, false);
});
});
@@ -86,7 +91,7 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
setTimeout(() => {
compactionInFlight = false;
}, 90_000);
const params = buildAggregateTimeoutParams({
const { params, onTimeoutSpy } = buildAggregateTimeoutParams({
waitForCompactionRetry,
isCompactionStillInFlight: () => compactionInFlight,
});
@@ -97,19 +102,19 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
const result = await resultPromise;
expect(result.timedOut).toBe(true);
expectClearedTimeoutState(params.onTimeout, true);
expectClearedTimeoutState(onTimeoutSpy, true);
});
});
it("does not time out when compaction retry resolves", async () => {
await withFakeTimers(async () => {
const waitForCompactionRetry = vi.fn(async () => {});
const params = buildAggregateTimeoutParams({ waitForCompactionRetry });
const { params, onTimeoutSpy } = buildAggregateTimeoutParams({ waitForCompactionRetry });
const result = await waitForCompactionRetryWithAggregateTimeout(params);
expect(result.timedOut).toBe(false);
expectClearedTimeoutState(params.onTimeout, false);
expectClearedTimeoutState(onTimeoutSpy, false);
});
});
@@ -118,7 +123,7 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
const abortError = new Error("aborted");
abortError.name = "AbortError";
const waitForCompactionRetry = vi.fn(async () => await new Promise<void>(() => {}));
const params = buildAggregateTimeoutParams({
const { params, onTimeoutSpy } = buildAggregateTimeoutParams({
waitForCompactionRetry,
abortable: async () => {
throw abortError;
@@ -127,7 +132,7 @@ describe("waitForCompactionRetryWithAggregateTimeout", () => {
await expect(waitForCompactionRetryWithAggregateTimeout(params)).rejects.toThrow("aborted");
expectClearedTimeoutState(params.onTimeout, false);
expectClearedTimeoutState(onTimeoutSpy, false);
});
});
});

2
src/agents/pi-embedded-runner/run/history-image-prune.test.ts
View File

@@ -8,7 +8,7 @@ function expectArrayMessageContent(
message: AgentMessage | undefined,
errorMessage: string,
): Array<{ type: string; text?: string; data?: string }> {
if (!message || !Array.isArray(message.content)) {
if (!message || !("content" in message) || !Array.isArray(message.content)) {
throw new Error(errorMessage);
}
return message.content as Array<{ type: string; text?: string; data?: string }>;

9
src/agents/pi-embedded-runner/runs.test.ts
View File

@@ -10,14 +10,17 @@ import {
waitForActiveEmbeddedRuns,
} from "./runs.js";
type RunHandle = Parameters<typeof setActiveEmbeddedRun>[1];
function createRunHandle(
overrides: { isCompacting?: boolean; abort?: ReturnType<typeof vi.fn> } = {},
) {
overrides: { isCompacting?: boolean; abort?: () => void } = {},
): RunHandle {
const abort = overrides.abort ?? (() => {});
return {
queueMessage: async () => {},
isStreaming: () => true,
isCompacting: () => overrides.isCompacting ?? false,
abort: overrides.abort ?? vi.fn(),
abort,
};
}

4
src/agents/pi-embedded-runner/system-prompt.test.ts
View File

@@ -2,7 +2,7 @@ import type { AgentSession } from "@mariozechner/pi-coding-agent";
import { describe, expect, it, vi } from "vitest";
import { applySystemPromptOverrideToSession, createSystemPromptOverride } from "./system-prompt.js";
type MutableSession = AgentSession & {
type MutableSystemPromptFields = {
_baseSystemPrompt?: string;
_rebuildSystemPrompt?: (toolNames: string[]) => string;
};
@@ -21,7 +21,7 @@ function applyAndGetMutableSession(
const { session, setSystemPrompt } = createMockSession();
applySystemPromptOverrideToSession(session, prompt);
return {
mutable: session as MutableSession,
mutable: session as unknown as MutableSystemPromptFields,
setSystemPrompt,
};
}