#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-3.0-or-later
"""Tracked guardrails for Phase-1.0AI."""

from __future__ import annotations

import argparse
import json
from pathlib import Path
import unittest


PARSER = argparse.ArgumentParser()
PARSER.add_argument("--root", type=Path, required=True)
ROOT = PARSER.parse_args().root


class Phase10AIGuardrails(unittest.TestCase):
    @classmethod
    def setUpClass(cls) -> None:
        cls.data = json.loads((ROOT / "manifests/retroarch/phase-1.0ai-mapping-model.json").read_text(encoding="utf-8"))

    def test_failure_never_retains_partial_mapping(self) -> None:
        transaction = self.data["transaction"]
        self.assertTrue(transaction["failure_unmaps_entire_new_child_region"])
        self.assertTrue(transaction["failure_releases_mirror"])
        self.assertFalse(transaction["partial_mapping_retained_on_failure"])

    def test_model_has_no_mapping_capability(self) -> None:
        transaction = self.data["transaction"]
        self.assertFalse(transaction["target_memory_interface_present"])
        self.assertFalse(transaction["host_memory_mapping_present"])
        self.assertFalse(transaction["process_interface_present"])

    def test_no_authority_or_target_claim(self) -> None:
        self.assertFalse(any(self.data["authorizations"].values()))
        self.assertFalse(self.data["decision"]["target_mapping_allowed"])
        self.assertFalse(self.data["decision"]["target_implementation_allowed"])
        self.assertFalse(self.data["decision"]["device_action_allowed"])


if __name__ == "__main__":
    unittest.main(argv=[__file__])