mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-04 19:10:23 +00:00
136 lines
4.2 KiB
TypeScript
136 lines
4.2 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { CronService } from "./service.js";
|
|
import { createCronStoreHarness, createNoopLogger } from "./service.test-harness.js";
|
|
import type { CronJob } from "./types.js";
|
|
|
|
const noopLogger = createNoopLogger();
|
|
const { makeStorePath } = createCronStoreHarness();
|
|
|
|
async function waitForFirstJob(
|
|
cron: CronService,
|
|
predicate: (job: CronJob | undefined) => boolean,
|
|
) {
|
|
let latest: CronJob | undefined;
|
|
for (let i = 0; i < 30; i++) {
|
|
const jobs = await cron.list({ includeDisabled: true });
|
|
latest = jobs[0];
|
|
if (predicate(latest)) {
|
|
return latest;
|
|
}
|
|
await vi.runOnlyPendingTimersAsync();
|
|
}
|
|
return latest;
|
|
}
|
|
|
|
async function withCronService(
|
|
cronEnabled: boolean,
|
|
run: (params: {
|
|
cron: CronService;
|
|
enqueueSystemEvent: ReturnType<typeof vi.fn>;
|
|
requestHeartbeatNow: ReturnType<typeof vi.fn>;
|
|
}) => Promise<void>,
|
|
) {
|
|
const store = await makeStorePath();
|
|
const enqueueSystemEvent = vi.fn();
|
|
const requestHeartbeatNow = vi.fn();
|
|
const cron = new CronService({
|
|
storePath: store.storePath,
|
|
cronEnabled,
|
|
log: noopLogger,
|
|
enqueueSystemEvent,
|
|
requestHeartbeatNow,
|
|
runIsolatedAgentJob: vi.fn(async () => ({ status: "ok" as const })),
|
|
});
|
|
|
|
await cron.start();
|
|
try {
|
|
await run({ cron, enqueueSystemEvent, requestHeartbeatNow });
|
|
} finally {
|
|
cron.stop();
|
|
await store.cleanup();
|
|
}
|
|
}
|
|
|
|
describe("CronService", () => {
|
|
beforeEach(() => {
|
|
vi.useFakeTimers();
|
|
vi.setSystemTime(new Date("2025-12-13T00:00:00.000Z"));
|
|
noopLogger.debug.mockClear();
|
|
noopLogger.info.mockClear();
|
|
noopLogger.warn.mockClear();
|
|
noopLogger.error.mockClear();
|
|
});
|
|
|
|
afterEach(() => {
|
|
vi.useRealTimers();
|
|
});
|
|
|
|
it("skips main jobs with empty systemEvent text", async () => {
|
|
await withCronService(true, async ({ cron, enqueueSystemEvent, requestHeartbeatNow }) => {
|
|
const atMs = Date.parse("2025-12-13T00:00:01.000Z");
|
|
await cron.add({
|
|
name: "empty systemEvent test",
|
|
enabled: true,
|
|
schedule: { kind: "at", at: new Date(atMs).toISOString() },
|
|
sessionTarget: "main",
|
|
wakeMode: "now",
|
|
payload: { kind: "systemEvent", text: " " },
|
|
});
|
|
|
|
vi.setSystemTime(new Date("2025-12-13T00:00:01.000Z"));
|
|
await vi.runOnlyPendingTimersAsync();
|
|
|
|
expect(enqueueSystemEvent).not.toHaveBeenCalled();
|
|
expect(requestHeartbeatNow).not.toHaveBeenCalled();
|
|
|
|
const job = await waitForFirstJob(cron, (current) => current?.state.lastStatus === "skipped");
|
|
expect(job?.state.lastStatus).toBe("skipped");
|
|
expect(job?.state.lastError).toMatch(/non-empty/i);
|
|
});
|
|
});
|
|
|
|
it("does not schedule timers when cron is disabled", async () => {
|
|
await withCronService(false, async ({ cron, enqueueSystemEvent, requestHeartbeatNow }) => {
|
|
const atMs = Date.parse("2025-12-13T00:00:01.000Z");
|
|
await cron.add({
|
|
name: "disabled cron job",
|
|
enabled: true,
|
|
schedule: { kind: "at", at: new Date(atMs).toISOString() },
|
|
sessionTarget: "main",
|
|
wakeMode: "now",
|
|
payload: { kind: "systemEvent", text: "hello" },
|
|
});
|
|
|
|
const status = await cron.status();
|
|
expect(status.enabled).toBe(false);
|
|
expect(status.nextWakeAtMs).toBeNull();
|
|
|
|
vi.setSystemTime(new Date("2025-12-13T00:00:01.000Z"));
|
|
await vi.runOnlyPendingTimersAsync();
|
|
|
|
expect(enqueueSystemEvent).not.toHaveBeenCalled();
|
|
expect(requestHeartbeatNow).not.toHaveBeenCalled();
|
|
expect(noopLogger.warn).toHaveBeenCalled();
|
|
});
|
|
});
|
|
|
|
it("status reports next wake when enabled", async () => {
|
|
await withCronService(true, async ({ cron }) => {
|
|
const atMs = Date.parse("2025-12-13T00:00:05.000Z");
|
|
await cron.add({
|
|
name: "status next wake",
|
|
enabled: true,
|
|
schedule: { kind: "at", at: new Date(atMs).toISOString() },
|
|
sessionTarget: "main",
|
|
wakeMode: "next-heartbeat",
|
|
payload: { kind: "systemEvent", text: "hello" },
|
|
});
|
|
|
|
const status = await cron.status();
|
|
expect(status.enabled).toBe(true);
|
|
expect(status.jobs).toBe(1);
|
|
expect(status.nextWakeAtMs).toBe(atMs);
|
|
});
|
|
});
|
|
});
|