"""Unit tests for nodes.debug_node.WPDebug."""

import json

from wp_nodes.debug_node import WPDebug
from wp_nodes.types import ContextPayload


class TestWPDebugSchema:
    def test_schema_basic(self):
        schema = WPDebug.define_schema()
        assert schema.node_id == "WP_Debug"
        assert schema.display_name == "WP Debug"
        assert schema.category == "wildcard-pipeline"
        assert schema.is_output_node is True
        assert schema.not_idempotent is True

    def test_inputs(self):
        schema = WPDebug.define_schema()
        names = [s.name for s in schema.inputs]
        assert names == ["context", "wp_viewer"]

        ctx_in, viewer_in = schema.inputs
        assert ctx_in.type_name == "PIPELINE_CONTEXT"
        assert viewer_in.type_name == "WP_DEBUG_VIEWER"
        assert viewer_in.socketless is True

    def test_no_outputs(self):
        schema = WPDebug.define_schema()
        assert schema.outputs == []


class TestWPDebugExecute:
    def _parse_snapshot(self, out):
        snapshot_list = out.ui["wp_debug_snapshot"]
        assert len(snapshot_list) == 1
        return json.loads(snapshot_list[0])

    def test_emits_ui_payload(self):
        # User-facing variables sit at top level of the snapshot.
        # Engine internals (`__wp_*`) are also top-level so the
        # frontend's per-tab readers (snapshot/trace/picks/warnings)
        # can pull them directly. `node_seed` is surfaced as
        # `__wp_node_seed__` so the snapshot tab's `__`-prefix filter
        # hides it (matching trace/picks/warnings).
        payload = ContextPayload(
            context={"style": "photo"},
            debug={"node_seed": 42, "__wp_trace__": [], "__wp_warnings__": []},
        )
        out = WPDebug.execute(context=payload, wp_viewer=None)

        assert out.ui is not None
        assert "wp_debug_snapshot" in out.ui
        snapshot = self._parse_snapshot(out)
        assert snapshot == {
            "style": "photo",
            "__wp_node_seed__": 42,
        }

    def test_trace_warnings_picks_surface_at_top_level(self):
        payload = ContextPayload(
            context={"style": "photo"},
            debug={
                "node_seed": 7,
                "__wp_trace__": [{"id": "abc", "type": "wildcard"}],
                "__wp_warnings__": [{"type": "duplicate_variable"}],
            },
            internals={"__wp_picks__": {"abc": {"value": "v1"}}},
        )
        out = WPDebug.execute(context=payload, wp_viewer=None)
        snapshot = self._parse_snapshot(out)
        assert snapshot["style"] == "photo"
        assert snapshot["__wp_trace__"] == [{"id": "abc", "type": "wildcard"}]
        assert snapshot["__wp_warnings__"] == [{"type": "duplicate_variable"}]
        assert snapshot["__wp_picks__"] == {"abc": {"value": "v1"}}
        assert snapshot["__wp_node_seed__"] == 7

    def test_complex_values_serialized_via_str(self):
        class Weird:
            def __str__(self): return "weird"

        payload = ContextPayload(
            context={"obj": Weird()},
            debug={},
        )
        out = WPDebug.execute(context=payload, wp_viewer=None)
        snapshot = self._parse_snapshot(out)
        assert snapshot["obj"] == "weird"

    def test_empty_context(self):
        payload = ContextPayload()
        out = WPDebug.execute(context=payload, wp_viewer=None)
        snapshot = self._parse_snapshot(out)
        assert snapshot == {}