Files
UHD_Media_Tool/apps/server/test/sources-identity/mteam-tasks.test.ts
T
rootandClaude Opus 5.5 05a7706dde feat(sources): site center backend (credentials, refresh login, user data, health check, delete)
PUT /sites/{id}/credentials by kind (moved from auth-settings, legacy body
accepted), POST refresh-login (async task or wait), POST health-check,
GET user-data history, DELETE /sites/{id} (soft delete, releases kept),
DELETE /site-accounts/{id}, GET /sites?include=accounts; refreshSiteAccount
task with login-failure decisions and auto re-login on auth_expired;
scheduled jobs sites.user_data (6h) and sites.health_check (1h);
settings.sites defaults for rate limit, snapshot interval, UA and proxy.

Co-Authored-By: Claude Opus 5.5 <[email protected]>
2026-09-28 15:18:40 +08:00

105 lines
3.8 KiB
TypeScript

import { TaskRegistry } from "@umt/tasks";
import { createMockMteamApi, type MockMteamApi } from "@umt/testkit";
import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { snapshotSite } from "../../src/modules/sources-identity/service/snapshot.ts";
import { createHarness, type Harness, pgAvailable } from "./harness.ts";
const up = await pgAvailable();
describe.skipIf(!up)("M-Team 快照与任务注册", () => {
let h: Harness;
let api: MockMteamApi;
let siteId: string;
beforeAll(async () => {
api = await createMockMteamApi();
h = await createHarness("mteam");
const site = await h.db
.insertInto("site")
.values({
key: "mteam",
name: "馒头",
adapter: "mteam_api",
base_url: "https://kp.m-team.cc/",
rate_limit: JSON.stringify({ minIntervalMs: 0 }),
config: JSON.stringify({ apiBaseUrl: api.baseUrl }),
})
.returningAll()
.executeTakeFirstOrThrow();
siteId = site.id;
const ref = await h.secrets.create({ purpose: "site_api_key", plaintext: api.apiKey });
await h.db
.insertInto("site_account")
.values({ site_id: siteId, label: "api", auth_kind: "api_key", secret_ref: ref })
.execute();
});
afterAll(async () => {
await h?.close();
await api?.close();
});
it("API Key 站点快照:原始分类保留,促销带观察/过期时间;key 不入库", async () => {
const r = await snapshotSite(h.rt, { siteId, kind: "periodic" });
expect(r.status).toBe("ok");
expect(r.itemCount).toBeGreaterThan(0);
const rows = await h.db.selectFrom("source_release").selectAll().where("site_id", "=", siteId).execute();
const sports = rows.find((x) => x.site_release_id === "912237");
expect(sports?.category_raw).toBe("407");
const promo = rows.find((x) => x.site_release_id === "912233")!;
expect((promo.promotion as { observedAt?: string }).observedAt).toBeTruthy();
expect(JSON.stringify(rows)).not.toContain(api.apiKey);
});
it("key 失效:账号标失效 + 待处理,快照结果 failed", async () => {
api.setFailMode("banned_key");
const r = await snapshotSite(h.rt, { siteId, kind: "manual" });
expect(r.status).toBe("failed");
const acc = await h.db
.selectFrom("site_account")
.select(["status"])
.where("site_id", "=", siteId)
.executeTakeFirstOrThrow();
expect(["invalid", "expired"]).toContain(acc.status);
const d = await h.db
.selectFrom("decision")
.selectAll()
.where("dedupe_key", "=", `site-auth:${siteId}`)
.executeTakeFirstOrThrow();
expect(d.status).toBe("open");
api.setFailMode(null);
});
it("模块注册 6 个任务与 3 条 outbox 路由;任务处理器经 step.run 调服务", async () => {
const reg = new TaskRegistry();
h.module.registerTasks(reg, h.ctx);
expect(reg.names()).toEqual([
"ingestFeed",
"refreshMetadata",
"refreshSiteAccount",
"resolveObservation",
"searchSubscription",
"snapshotSite",
]);
expect(Object.keys(h.module.outboxRoutes!(h.ctx)).sort()).toEqual([
"feed.due",
"feed.gap_suspected",
"release.observed",
]);
const def = reg.latest("resolveObservation");
expect(def.inputSchema.safeParse({}).success).toBe(false);
expect(def.inputSchema.safeParse({ observationId: "00000000-0000-4000-8000-000000000000" }).success).toBe(true);
const steps: string[] = [];
const out = await reg.latest("snapshotSite").handler({
input: { siteId, kind: "periodic" },
step: {
run: async (name: string, fn: () => unknown) => {
steps.push(name);
return fn();
},
},
} as never);
expect(steps).toEqual(["snapshot"]);
expect(out).toMatchObject({ status: "skipped", reason: "auth_invalid" });
});
});