@@ -49,26 +49,46 @@ describe("deepinfra video generation provider", () => {
4949},
5050});
515152-expect(resolveProviderHttpRequestConfigMock).toHaveBeenCalledWith(
53-expect.objectContaining({
54-provider: "deepinfra",
55-capability: "video",
56-baseUrl: "https://api.deepinfra.com/v1/inference",
57-}),
58-);
59-expect(postJsonRequestMock).toHaveBeenCalledWith(
60-expect.objectContaining({
61-url: "https://api.deepinfra.com/v1/inference/Pixverse/Pixverse-T2V",
62-body: {
63-prompt: "A bicycle weaving through a rainy neon street",
64-aspect_ratio: "16:9",
65-duration: 8,
66-seed: 42,
67-negative_prompt: "blur",
68-style: "anime",
52+expect(resolveProviderHttpRequestConfigMock.mock.calls).toEqual([
53+[
54+{
55+baseUrl: "https://api.deepinfra.com/v1/inference",
56+defaultBaseUrl: "https://api.deepinfra.com/v1/inference",
57+allowPrivateNetwork: false,
58+defaultHeaders: {
59+Authorization: "Bearer provider-key",
60+"Content-Type": "application/json",
61+},
62+provider: "deepinfra",
63+capability: "video",
64+transport: "http",
6965},
70-}),
71-);
66+],
67+]);
68+expect(postJsonRequestMock).toHaveBeenCalledOnce();
69+const [postRequest] = postJsonRequestMock.mock.calls[0] ?? [];
70+const postRequestHeaders = Reflect.get(postRequest ?? {}, "headers");
71+expect(postRequestHeaders).toBeInstanceOf(Headers);
72+expect(Object.fromEntries((postRequestHeaders as Headers).entries())).toEqual({
73+authorization: "Bearer provider-key",
74+"content-type": "application/json",
75+});
76+expect(postRequest).toEqual({
77+url: "https://api.deepinfra.com/v1/inference/Pixverse/Pixverse-T2V",
78+headers: postRequestHeaders,
79+body: {
80+prompt: "A bicycle weaving through a rainy neon street",
81+aspect_ratio: "16:9",
82+duration: 8,
83+seed: 42,
84+negative_prompt: "blur",
85+style: "anime",
86+},
87+timeoutMs: undefined,
88+fetchFn: fetch,
89+allowPrivateNetwork: false,
90+dispatcherPolicy: undefined,
91+});
7292expect(result.videos).toEqual([
7393{
7494url: "https://api.deepinfra.com/generated/video.mp4",
@@ -109,10 +129,10 @@ describe("deepinfra video generation provider", () => {
109129if (!video) {
110130throw new Error("Expected generated DeepInfra video");
111131}
112-expect(video).toMatchObject({
132+expect(video).toEqual({
133+buffer: Buffer.from("webm-data"),
113134mimeType: "video/webm",
114135fileName: "video-1.webm",
115136});
116-expect(video.buffer).toEqual(Buffer.from("webm-data"));
117137});
118138});
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。