from __future__ import annotations

import unittest

from app.main import (
    ReactFlowProjectPayload,
    build_reactflow_system_xml,
    compile_reactflow_network,
    run_system_xml_simulation,
)
from app.system_xml import validate_system_xml_document
from tests.test_amesim_pnvo001_signal_xml import signal_edge, signal_port
from tests.test_generic_system_xml_simulation import component_node, physical_edge
from tests.test_system_xml_protocol import physical_port


def amesim_ud00_signal_project() -> ReactFlowProjectPayload:
    return ReactFlowProjectPayload(
        name="amesim-ud00-signal-smoke",
        nodes=[
            component_node(
                "piecewise_1",
                "amesim_ud00",
                [signal_port("out", "output", "right")],
                {
                    "tstart": 0.0,
                    "start1": 1.0,
                    "end1": 1.0,
                    "t1": 1.0,
                    "start2": 1.0,
                    "end2": 1.0,
                    "t2": 0.0,
                    "start3": 1.0,
                    "end3": 1.0,
                    "t3": 0.0,
                    "start4": 1.0,
                    "end4": 1.0,
                    "t4": 0.0,
                    "start5": 1.0,
                    "end5": 1.0,
                    "t5": 0.0,
                    "start6": 1.0,
                    "end6": 1.0,
                    "t6": 0.0,
                    "start7": 1.0,
                    "end7": 1.0,
                    "t7": 0.0,
                    "start8": 1.0,
                    "end8": 1.0,
                    "t8": 0.0,
                    "nstages": 1.0,
                    "iscyclic": 0.0,
                },
            ),
            component_node(
                "cylinder_1",
                "cylinder",
                [physical_port("port_b", "outlet", "right")],
                {"volume": 0.01, "p0": 300000.0, "T0": 300.0},
            ),
            component_node(
                "valve_1",
                "amesim_pnvo001",
                [
                    signal_port("res", "input", "left"),
                    physical_port("port_2", "bidirectional", "left"),
                    physical_port("port_3", "bidirectional", "right"),
                ],
                {
                    "cq": 0.72,
                    "area0": 5.0e-6,
                    "Cv": 0.5,
                    "Kv": 0.4,
                    "gi": 1.0,
                    "flowset": 1.0,
                    "opening0": 0.0,
                },
            ),
            component_node(
                "tank_1",
                "tank",
                [physical_port("port_a", "inlet", "left")],
                {"volume": 0.1, "p0": 100000.0, "T0": 300.0},
            ),
        ],
        edges=[
            signal_edge("signal-1", "piecewise_1", "out", "valve_1", "res"),
            physical_edge("edge-1", "cylinder_1", "port_b", "valve_1", "port_2"),
            physical_edge("edge-2", "valve_1", "port_3", "tank_1", "port_a"),
        ],
        simulation={"t_start": 0.0, "t_stop": 0.002, "step": 0.001, "max_step": 0.001, "method": "BDF"},
    )


class AmesimUd00XmlTests(unittest.TestCase):
    def test_signal_project_compiles_with_signal_connection(self) -> None:
        network = compile_reactflow_network(amesim_ud00_signal_project())

        self.assertIn("piecewise_1", network.components)
        self.assertEqual(network.components["piecewise_1"].model_type, "amesim_ud00")
        self.assertEqual(network.components["valve_1"].model_type, "amesim_pnvo001")
        self.assertTrue(network.pressure_flow_structure_dict()["isSquare"])

    def test_signal_system_xml_validates_and_simulates(self) -> None:
        xml = build_reactflow_system_xml(amesim_ud00_signal_project())
        report = validate_system_xml_document(xml)

        self.assertTrue(report.valid, report.as_dict())
        result = run_system_xml_simulation(xml)

        self.assertTrue(result["success"], result["message"])
        self.assertEqual(result["series"]["time"], [0.0, 0.001, 0.002])
        self.assertEqual(result["series"]["piecewise_1.out.signal"], [1.0, 1.0, 1.0])
        self.assertEqual(result["series"]["valve_1.res.signal"], [1.0, 1.0, 1.0])
        self.assertIn("valve_1.xv", result["series"])
        self.assertGreater(result["diagnostics"]["signal"]["propagations"], 0)


if __name__ == "__main__":
    unittest.main()