from __future__ import annotations

from uuid import uuid4
import os
import unittest

from prod.clean_arch.dita_v2 import (
    BackendMode,
    ControlUpdate,
    InMemoryControlPlane,
    ZincControlPlane,
    KernelControlSnapshot,
    KernelMode,
    KernelVerbosity,
    RealZincControlPlane,
    build_control_plane,
)
from prod.clean_arch.dita_v2.real_control_plane import SharedRegion


HAS_REAL_ZINC = SharedRegion is not None


@unittest.skipUnless(HAS_REAL_ZINC, "Real Zinc adapter is unavailable")
class TestDITAv2RealControlPlane(unittest.TestCase):
    def test_build_control_plane_defaults_to_zinc(self) -> None:
        plane = build_control_plane()
        self.assertIsInstance(plane, ZincControlPlane)

    def test_roundtrip_update_and_read(self) -> None:
        prefix = f"dita_v2_control_{uuid4().hex}"
        writer = RealZincControlPlane(prefix=prefix, create=True)
        reader = RealZincControlPlane(prefix=prefix, create=False)
        try:
            snapshot = writer.update(
                ControlUpdate(
                    mode=KernelMode.DEBUG,
                    verbosity=KernelVerbosity.TRACE,
                    backend_mode=BackendMode.BINGX,
                    trace_transitions=True,
                    mirror_to_hazelcast=True,
                )
            )
            self.assertEqual(snapshot.mode, KernelMode.DEBUG)
            self.assertEqual(snapshot.verbosity, KernelVerbosity.TRACE)
            self.assertEqual(snapshot.backend_mode, BackendMode.BINGX)
            self.assertTrue(snapshot.trace_transitions)
            read_back = reader.read()
            self.assertEqual(read_back.mode, KernelMode.DEBUG)
            self.assertEqual(read_back.verbosity, KernelVerbosity.TRACE)
            self.assertEqual(read_back.backend_mode, BackendMode.BINGX)
            self.assertTrue(read_back.trace_transitions)
        finally:
            writer.close()
            reader.close()

    def test_env_can_select_real_control_plane(self) -> None:
        prefix = f"dita_v2_control_{uuid4().hex}"
        previous = os.environ.get("DITA_V2_CONTROL_PLANE")
        os.environ["DITA_V2_CONTROL_PLANE"] = "REAL_ZINC"
        try:
            plane = build_control_plane(prefix=prefix)
            self.assertIsInstance(plane, RealZincControlPlane)
            if isinstance(plane, RealZincControlPlane):
                plane.close()
        finally:
            if previous is None:
                os.environ.pop("DITA_V2_CONTROL_PLANE", None)
            else:
                os.environ["DITA_V2_CONTROL_PLANE"] = previous

    def test_initial_snapshot_is_default(self) -> None:
        prefix = f"dita_v2_control_{uuid4().hex}"
        plane = RealZincControlPlane(prefix=prefix, create=True)
        try:
            snapshot = plane.read()
            self.assertEqual(snapshot, KernelControlSnapshot())
        finally:
            plane.close()


class TestDITAv2InMemoryControlPlane(unittest.TestCase):
    def test_wait_and_notify(self) -> None:
        plane = InMemoryControlPlane()
        self.assertFalse(plane.wait(timeout_ms=1))
        plane.notify()
        self.assertTrue(plane.wait(timeout_ms=1))
        snapshot = plane.update(ControlUpdate(mode=KernelMode.DEBUG, verbosity=KernelVerbosity.TRACE))
        self.assertEqual(snapshot.mode, KernelMode.DEBUG)
        self.assertEqual(snapshot.verbosity, KernelVerbosity.TRACE)


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