from pathlib import Path
from unittest.mock import patch

import numpy as np
from backend.data_types import DataField
from backend.execution import ExecutionEngine
import backend.nodes  # noqa: F401

FIXTURES = Path(__file__).parent.parent / "output"


def test_load_npy():
    from backend.nodes.image_demo import ImageDemo

    with patch("backend.nodes.image_demo.DEMO_DIR", FIXTURES):
        result = ImageDemo().load(name="nanoparticles.npy")

    # result[0] is the FILE_PATH string, fields follow
    assert len(result) >= 2
    assert isinstance(result[0], str)
    assert isinstance(result[1], DataField)
    assert result[1].data.ndim == 2


def test_load_ibw_multi_channel():
    from backend.nodes.image_demo import ImageDemo

    with patch("backend.nodes.image_demo.DEMO_DIR", FIXTURES):
        result = ImageDemo().load(name="Bacteria.ibw")

    fields = [v for v in result if isinstance(v, DataField)]
    assert len(fields) == 4
    for field in fields:
        assert field.data.ndim == 2


def test_load_not_found():
    from backend.nodes.image_demo import ImageDemo

    with patch("backend.nodes.image_demo.DEMO_DIR", FIXTURES):
        try:
            ImageDemo().load(name="nonexistent_file.png")
            assert False, "Should have raised FileNotFoundError"
        except FileNotFoundError:
            pass


def test_load_cache():
    from backend.nodes.image import Image
    from backend.nodes.image_demo import ImageDemo

    node = ImageDemo()
    Image._load_fields_cached.cache_clear()

    with patch("backend.nodes.image_demo.DEMO_DIR", FIXTURES):
        import backend.importers.array_image as _ai
        with patch.object(_ai, "load", wraps=_ai.load) as loader:
            _, first = node.load(name="nanoparticles.npy")
            _, second = node.load(name="nanoparticles.npy")
            assert loader.call_count == 1

        assert np.allclose(first.data, second.data)
        assert first is not second
        first.data[0, 0] = -999.0

        _, third = node.load(name="nanoparticles.npy")
        assert third.data[0, 0] != -999.0

    Image._load_fields_cached.cache_clear()


def test_multi_layer_preview_payload():
    previews = []

    with patch("backend.nodes.image_demo.DEMO_DIR", FIXTURES):
        ExecutionEngine().execute(
            {"1": {"class_type": "ImageDemo", "inputs": {"name": "Bacteria.ibw", "colormap": "viridis"}}},
            on_preview=lambda node_id, payload: previews.append((node_id, payload)),
        )

    assert len(previews) == 1
    node_id, payload = previews[0]
    assert node_id == "1"
    assert payload["kind"] == "layer_gallery"
    assert len(payload["layers"]) == 4
    assert all(isinstance(layer["name"], str) and layer["name"] for layer in payload["layers"])
    assert all(layer["image"].startswith("data:image/png;base64,") for layer in payload["layers"])