103 lines
4.8 KiB
TypeScript
103 lines
4.8 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import { defaultConfig } from "../src/stores/use-config-store";
|
|
import { createSeedanceTask } from "../src/services/api/video-provider-seedance";
|
|
import type { VideoProviderDeps } from "../src/services/api/video-provider-deps";
|
|
import { unwrapEnvelope, videoResponseTools, videoTaskId } from "../src/services/api/video-response";
|
|
import { isPublicMediaUrl, normalizeVideoResolution, normalizeVideoSeconds, normalizeVideoSize } from "../src/services/api/video-validation";
|
|
|
|
describe("video API response contracts", () => {
|
|
test("解包裸响应和后端 envelope", () => {
|
|
expect(unwrapEnvelope({ id: "task-1" }, "缺少任务")).toEqual({ id: "task-1" });
|
|
expect(unwrapEnvelope({ code: 0, data: { id: "task-2" }, msg: "ok" }, "缺少任务")).toEqual({ id: "task-2" });
|
|
expect(videoTaskId({ request_id: "request-1" })).toBe("request-1");
|
|
});
|
|
|
|
test("业务失败和空数据不会被静默转换为成功", () => {
|
|
expect(() => unwrapEnvelope({ code: 401, data: null, msg: "未授权" }, "缺少任务")).toThrow("未授权");
|
|
expect(() => unwrapEnvelope({ code: 0, data: null, msg: "ok" }, "缺少任务")).toThrow("缺少任务");
|
|
});
|
|
});
|
|
|
|
describe("video request normalization", () => {
|
|
test("规范化时长、比例尺寸和分辨率", () => {
|
|
expect(normalizeVideoSeconds("0")).toBe("6");
|
|
expect(normalizeVideoSeconds("8.9")).toBe("8");
|
|
expect(normalizeVideoSize("16:9")).toBe("1280x720");
|
|
expect(normalizeVideoSize("auto")).toBeNull();
|
|
expect(normalizeVideoResolution("low")).toBe("480p");
|
|
expect(normalizeVideoResolution("2k")).toBe("1440p");
|
|
});
|
|
|
|
test("只把 HTTP(S) 地址视为公网媒体地址", () => {
|
|
expect(isPublicMediaUrl("https://cdn.example.com/video.mp4")).toBe(true);
|
|
expect(isPublicMediaUrl("http://localhost/video.mp4")).toBe(true);
|
|
expect(isPublicMediaUrl("asset://resource-1")).toBe(false);
|
|
expect(isPublicMediaUrl("data:video/mp4;base64,AAAA")).toBe(false);
|
|
expect(isPublicMediaUrl("/resources/video.mp4")).toBe(false);
|
|
});
|
|
});
|
|
|
|
describe("Volcengine Ark full-modal references", () => {
|
|
const model = "seedance-2-0-250824";
|
|
const config = {
|
|
...defaultConfig,
|
|
baseUrl: "https://ark.cn-beijing.volces.com/api/v3",
|
|
apiKey: "test-key",
|
|
interfaceType: "volcengine-ark-video",
|
|
channels: [
|
|
{
|
|
id: "ark",
|
|
name: "Ark",
|
|
baseUrl: "https://ark.cn-beijing.volces.com/api/v3",
|
|
apiKey: "test-key",
|
|
secretKey: "",
|
|
headers: [],
|
|
apiFormat: "openai",
|
|
interfaceType: "volcengine-ark-video",
|
|
models: [model],
|
|
scope: "user",
|
|
enabled: true,
|
|
},
|
|
],
|
|
};
|
|
|
|
test("映射图片、视频和音频参考素材", async () => {
|
|
let requestUrl = "";
|
|
let requestBody: unknown;
|
|
const deps = {
|
|
transport: {
|
|
post: async (url: string, body: unknown) => {
|
|
requestUrl = url;
|
|
requestBody = body;
|
|
return { id: "task-1", status: "queued" };
|
|
},
|
|
},
|
|
response: videoResponseTools,
|
|
} as unknown as VideoProviderDeps;
|
|
|
|
await createSeedanceTask(
|
|
deps,
|
|
config as never,
|
|
model,
|
|
"保持主体一致",
|
|
[{ id: "image-1", name: "image.png", type: "image/png", dataUrl: "", url: "https://cdn.example.com/image.png" }],
|
|
[{ id: "video-1", name: "video.mp4", type: "video/mp4", url: "https://cdn.example.com/video.mp4" }],
|
|
[{ id: "audio-1", name: "audio.mp3", type: "audio/mpeg", url: "https://cdn.example.com/audio.mp3" }],
|
|
);
|
|
|
|
expect(requestUrl).toBe("https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks");
|
|
expect((requestBody as { content: unknown[] }).content).toEqual([
|
|
{ type: "text", text: "保持主体一致" },
|
|
{ type: "image_url", image_url: { url: "https://cdn.example.com/image.png" }, role: "reference_image" },
|
|
{ type: "video_url", video_url: { url: "https://cdn.example.com/video.mp4" }, role: "reference_video" },
|
|
{ type: "audio_url", audio_url: { url: "https://cdn.example.com/audio.mp3" }, role: "reference_audio" },
|
|
]);
|
|
});
|
|
|
|
test("拒绝纯音频和文本加音频", async () => {
|
|
const deps = { transport: {}, response: videoResponseTools } as unknown as VideoProviderDeps;
|
|
await expect(createSeedanceTask(deps, config as never, model, "跟随节奏", [], [], [{ id: "audio-1", name: "audio.mp3", type: "audio/mpeg", url: "https://cdn.example.com/audio.mp3" }])).rejects.toThrow("不支持纯音频或文本+音频");
|
|
});
|
|
});
|