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

import {
    buildPromptRequestBody,
    enrichPromptDataForMfv,
    graphContainsApiNode,
    resolveMfvPreviewMethod,
} from "../features/viewer/workflowSidebar/sidebarRunButton.js";

describe("sidebarRunButton MFV preview forcing", () => {
    it("normalizes configured preview methods to supported values", () => {
        expect(resolveMfvPreviewMethod("TAESD")).toBe("taesd");
        expect(resolveMfvPreviewMethod(" latent2rgb ")).toBe("latent2rgb");
        expect(resolveMfvPreviewMethod("bogus")).toBe("auto");
    });

    it("injects preview_method and workflow metadata into queue payload", () => {
        const promptData = enrichPromptDataForMfv(
            {
                output: { 1: { class_type: "KSampler" } },
                workflow: { nodes: [{ id: 1 }] },
            },
            "taesd",
        );

        expect(promptData.extra_data.preview_method).toBe("taesd");
        expect(promptData.extra_data.extra_pnginfo.workflow).toEqual({ nodes: [{ id: 1 }] });
    });

    it("builds POST /prompt body with client_id when available", () => {
        const body = buildPromptRequestBody(
            {
                output: { 1: { class_type: "SamplerCustomAdvanced" } },
                workflow: { nodes: [{ id: 1 }] },
            },
            { previewMethod: "default", clientId: "client-42" },
        );

        expect(body.prompt).toEqual({ 1: { class_type: "SamplerCustomAdvanced" } });
        expect(body.client_id).toBe("client-42");
        expect(body.extra_data.extra_pnginfo.workflow).toEqual({ nodes: [{ id: 1 }] });
        expect(body.extra_data.preview_method).toBeUndefined();
    });

    it("detects API nodes in graph shapes using _nodes or nodes_by_id", () => {
        expect(
            graphContainsApiNode({
                _nodes: [{ type: "HappyHorseImageToVideoApi" }],
            }),
        ).toBe(true);

        expect(
            graphContainsApiNode({
                nodes_by_id: {
                    12: { class_type: "SamplerCustom" },
                    13: { type: "SomeCloudApi" },
                },
            }),
        ).toBe(true);
    });

    it("detects API nodes inside alternate subgraph containers", () => {
        expect(
            graphContainsApiNode({
                nodes: [
                    {
                        type: "SubgraphContainer",
                        _subgraph: {
                            _nodes_by_id: new Map([
                                [1, { type: "InnerPreviewNode" }],
                                [2, { class_type: "NestedVideoApi" }],
                            ]),
                        },
                    },
                ],
            }),
        ).toBe(true);

        expect(
            graphContainsApiNode({
                nodes: [
                    {
                        type: "GroupNode",
                        nodes: [{ type: "AnotherInnerApi" }],
                    },
                ],
            }),
        ).toBe(true);
    });
});