import { beforeEach, describe, expect, it, vi } from "vitest";

import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";

import { VIDEO_GENERATION_TASK_KIND } from "../video-generation-task-status.js";

import {

announceDeliveryMocks,

recordVideoGenerationTaskProgress,

wakeVideoGenerationTaskCompletion,

} = await import("./video-generate-background.js");

const { withMediaGenerationTaskKeepalive } = await import("./media-generate-background-shared.js");

describe("video generate background helpers", () => {

beforeEach(() => {

});

});

afterEach(() => {

vi.useRealTimers();

});

it("creates a running task with queued progress text", () => {

taskExecutorMocks.createRunningTaskRun.mockReturnValue({

taskId: "task-123",

});

});

it("keeps long-running media tasks fresh while provider work is pending", async () => {

vi.useFakeTimers();

let resolveRun!: (value: string) => void;

const runPromise = new Promise<string>((resolve) => {

resolveRun = resolve;

});

const task = withMediaGenerationTaskKeepalive({

handle: {

taskId: "task-123",

runId: "tool:video_generate:abc",

requesterSessionKey: "agent:main:discord:direct:123",

taskLabel: "friendly lobster surfing",

},

progressSummary: "Generating video",

run: () => runPromise,

});

await vi.advanceTimersByTimeAsync(60_000);

expectRecordedTaskProgress({

taskExecutorMocks,

runId: "tool:video_generate:abc",

progressSummary: "Generating video",

});

resolveRun("done");

await expect(task).resolves.toBe("done");

const callsAfterCompletion = taskExecutorMocks.recordTaskRunProgressByRunId.mock.calls.length;

await vi.advanceTimersByTimeAsync(60_000);

expect(taskExecutorMocks.recordTaskRunProgressByRunId).toHaveBeenCalledTimes(

callsAfterCompletion,

);

});

it("queues a completion event by default when direct send is disabled", async () => {

announceDeliveryMocks.deliverSubagentAnnouncement.mockResolvedValue({

delivered: true,