mirror of https://github.com/openclaw/openclaw.git
82 lines
2.3 KiB
TypeScript
82 lines
2.3 KiB
TypeScript
import os from "node:os";
|
|
import path from "node:path";
|
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
import type { CliDeps } from "../cli/deps.js";
|
|
import type { OpenClawConfig } from "../config/config.js";
|
|
|
|
const enqueueSystemEventMock = vi.fn();
|
|
const requestHeartbeatNowMock = vi.fn();
|
|
const loadConfigMock = vi.fn();
|
|
|
|
vi.mock("../infra/system-events.js", () => ({
|
|
enqueueSystemEvent: (...args: unknown[]) => enqueueSystemEventMock(...args),
|
|
}));
|
|
|
|
vi.mock("../infra/heartbeat-wake.js", () => ({
|
|
requestHeartbeatNow: (...args: unknown[]) => requestHeartbeatNowMock(...args),
|
|
}));
|
|
|
|
vi.mock("../config/config.js", async () => {
|
|
const actual = await vi.importActual<typeof import("../config/config.js")>("../config/config.js");
|
|
return {
|
|
...actual,
|
|
loadConfig: () => loadConfigMock(),
|
|
};
|
|
});
|
|
|
|
import { buildGatewayCronService } from "./server-cron.js";
|
|
|
|
describe("buildGatewayCronService", () => {
|
|
beforeEach(() => {
|
|
enqueueSystemEventMock.mockReset();
|
|
requestHeartbeatNowMock.mockReset();
|
|
loadConfigMock.mockReset();
|
|
});
|
|
|
|
it("canonicalizes non-agent sessionKey to agent store key for enqueue + wake", async () => {
|
|
const tmpDir = path.join(os.tmpdir(), `server-cron-${Date.now()}`);
|
|
const cfg = {
|
|
session: {
|
|
mainKey: "main",
|
|
},
|
|
cron: {
|
|
store: path.join(tmpDir, "cron.json"),
|
|
},
|
|
} as OpenClawConfig;
|
|
loadConfigMock.mockReturnValue(cfg);
|
|
|
|
const state = buildGatewayCronService({
|
|
cfg,
|
|
deps: {} as CliDeps,
|
|
broadcast: () => {},
|
|
});
|
|
try {
|
|
const job = await state.cron.add({
|
|
name: "canonicalize-session-key",
|
|
enabled: true,
|
|
schedule: { kind: "at", at: new Date(1).toISOString() },
|
|
sessionTarget: "main",
|
|
wakeMode: "next-heartbeat",
|
|
sessionKey: "discord:channel:ops",
|
|
payload: { kind: "systemEvent", text: "hello" },
|
|
});
|
|
|
|
await state.cron.run(job.id, "force");
|
|
|
|
expect(enqueueSystemEventMock).toHaveBeenCalledWith(
|
|
"hello",
|
|
expect.objectContaining({
|
|
sessionKey: "agent:main:discord:channel:ops",
|
|
}),
|
|
);
|
|
expect(requestHeartbeatNowMock).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
sessionKey: "agent:main:discord:channel:ops",
|
|
}),
|
|
);
|
|
} finally {
|
|
state.cron.stop();
|
|
}
|
|
});
|
|
});
|