79 lines
2.4 KiB
TypeScript
79 lines
2.4 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { NextRequest } from "next/server";
|
|
|
|
let runtimeRoot = "";
|
|
let postClaim: (typeof import("../src/app/api/v1/master-agent/tasks/claim/route.ts"))["POST"];
|
|
let queueMasterAgentTask: (typeof import("../src/lib/boss-data.ts"))["queueMasterAgentTask"];
|
|
|
|
async function setup() {
|
|
if (runtimeRoot) return;
|
|
|
|
runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-master-task-claim-route-"));
|
|
process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
|
|
process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");
|
|
|
|
const [route, data] = await Promise.all([
|
|
import("../src/app/api/v1/master-agent/tasks/claim/route.ts"),
|
|
import("../src/lib/boss-data.ts"),
|
|
]);
|
|
postClaim = route.POST;
|
|
queueMasterAgentTask = data.queueMasterAgentTask;
|
|
}
|
|
|
|
test.beforeEach(async () => {
|
|
await setup();
|
|
await rm(runtimeRoot, { recursive: true, force: true });
|
|
});
|
|
|
|
test.after(async () => {
|
|
if (runtimeRoot) {
|
|
await rm(runtimeRoot, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("master task claim waits for a matching queued task event before returning", async () => {
|
|
const claimStartedAt = Date.now();
|
|
const claim = postClaim(
|
|
new NextRequest("http://127.0.0.1:3000/api/v1/master-agent/tasks/claim", {
|
|
method: "POST",
|
|
headers: {
|
|
"content-type": "application/json",
|
|
"x-boss-device-token": "boss-mac-studio-token",
|
|
},
|
|
body: JSON.stringify({
|
|
deviceId: "mac-studio",
|
|
waitMs: 1_000,
|
|
}),
|
|
}),
|
|
);
|
|
|
|
setTimeout(() => {
|
|
void queueMasterAgentTask({
|
|
projectId: "master-agent",
|
|
requestMessageId: "msg-long-poll",
|
|
requestText: "打开 Finder",
|
|
executionPrompt: "打开 Finder",
|
|
requestedBy: "Boss 超级管理员",
|
|
requestedByAccount: "krisolo",
|
|
deviceId: "mac-studio",
|
|
taskType: "desktop_control",
|
|
});
|
|
}, 20);
|
|
|
|
const response = await claim;
|
|
const payload = (await response.json()) as {
|
|
ok: boolean;
|
|
task?: { taskId: string; taskType: string; status: string } | null;
|
|
};
|
|
|
|
assert.equal(response.status, 200);
|
|
assert.equal(payload.ok, true);
|
|
assert.equal(payload.task?.taskType, "desktop_control");
|
|
assert.equal(payload.task?.status, "running");
|
|
assert.ok(Date.now() - claimStartedAt < 500);
|
|
});
|