import importlib.util
import copy
from pathlib import Path
import unittest


class ModelContracts(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        path = Path(__file__).resolve().parents[1] / "tools/validate_models.py"
        spec = importlib.util.spec_from_file_location("validate_vortex_models", path)
        cls.module = importlib.util.module_from_spec(spec)
        spec.loader.exec_module(cls.module)

    def test_retained_reference_and_architecture_models(self):
        self.module.validate()

    def test_rejects_same_warp_state_ownership(self):
        inventory, model = self.module.load_models()
        changed = copy.deepcopy(model)
        changed["candidates"][0]["handoff"]["same_warp"] = True
        with self.assertRaisesRegex(RuntimeError, "one warp"):
            self.module.validate_documents(inventory, changed)

    def test_rejects_inconsistent_shared_memory(self):
        inventory, model = self.module.load_models()
        changed = copy.deepcopy(model)
        changed["candidates"][1]["shared_kib"]["total"] = [52, 56]
        with self.assertRaisesRegex(RuntimeError, "VEA-B shared-memory"):
            self.module.validate_documents(inventory, changed)

    def test_rejects_inconsistent_screening_arithmetic(self):
        inventory, model = self.module.load_models()
        changed = copy.deepcopy(model)
        changed["candidates"][0]["mainloop_ms_projected"] = [192, 220]
        with self.assertRaisesRegex(RuntimeError, "screening arithmetic"):
            self.module.validate_documents(inventory, changed)

    def test_rejects_measured_result_label(self):
        inventory, model = self.module.load_models()
        changed = copy.deepcopy(model)
        changed["status"] = "achieved"
        with self.assertRaisesRegex(RuntimeError, "measurements"):
            self.module.validate_documents(inventory, changed)


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