import sys
import unittest
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parents[1] / "tools"))
from control_cycle import compare


def record(kind="forecast", resource_class="storage"):
    result = {
        "record_id": f"test:{kind}", "record_type": kind,
        "resource_id": "resource:test", "resource_class": resource_class,
        "period": "2026-09", "usage_proxies": {"stored_gb": {"value": 100, "unit": "GB-month"}},
        "costs": {"currency": "EUR", "infrastructure": 10, "internal_labor": 60, "external_labor": 0, "total": 70},
        "allocation": {"method": "direct", "driver": "stored_gb", "method_version": "1", "unattributed_eur": 0},
    }
    if kind == "actual":
        result["forecast_ref"] = "test:forecast"
    return result


class ControlCycleTest(unittest.TestCase):
    def test_same_mechanism_supports_required_resource_classes(self):
        for resource_class in ("storage", "cluster_compute", "shared_platform_service"):
            forecast = record(resource_class=resource_class)
            actual = record("actual", resource_class)
            actual["usage_proxies"]["stored_gb"]["value"] = 120
            result = compare(forecast, actual)
            self.assertEqual(resource_class, result["resource_class"])
            self.assertEqual("demand", result["usage_proxies"]["stored_gb"]["category"])

    def test_explicit_attribution_and_cost_split_are_preserved(self):
        forecast, actual = record(), record("actual")
        actual["costs"].update(infrastructure=12, internal_labor=90, total=102)
        actual["variance_attribution"] = {"costs.infrastructure": "model", "costs.internal_labor": "labor"}
        result = compare(forecast, actual)
        self.assertEqual("model", result["costs"]["infrastructure"]["category"])
        self.assertEqual(30, result["costs"]["internal_labor"]["error"])

    def test_rejects_wrong_forecast_reference(self):
        forecast, actual = record(), record("actual")
        actual["forecast_ref"] = "test:other"
        with self.assertRaisesRegex(ValueError, "immutable forecast"):
            compare(forecast, actual)

    def test_missing_proxy_is_data_quality_error(self):
        forecast, actual = record(), record("actual")
        actual["usage_proxies"] = {"cpu_hours": {"value": 4, "unit": "vCPU-hour"}}
        result = compare(forecast, actual)
        self.assertEqual("data_quality", result["usage_proxies"]["stored_gb"]["category"])


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