from __future__ import annotations

import unittest

from core.software_contracts import (
    DEFAULT_CAPABILITIES,
    get_contract,
    node_supports_request,
    version_at_least,
)


def _request(plot_type: str) -> dict:
    roles = {"input": "sample", "x": "x", "y": "y"}
    if plot_type in {"contour", "surface_3d", "ternary", "heatmap"}:
        roles["z"] = "z"
    return {
        "schema_version": 2,
        "inputs": [{
            "key": "sample",
            "artifact_id": "f4186347-65cc-4f07-9c26-bf11992beef8",
        }],
        "operation": {"plot": {"type": plot_type, "series": [roles]}},
        "outputs": [{"key": "figure_png", "type": "figure", "format": "png"}],
    }


class SoftwareContractTests(unittest.TestCase):
    def test_origin_contract_drives_schema_outputs_and_summary(self) -> None:
        contract = get_contract("origin.plot@v2")
        request = _request("line")
        normalized, digest = contract.normalize_request(request)
        self.assertEqual(normalized, request)
        self.assertEqual(len(digest), 64)
        self.assertEqual(contract.output_namespace, "origin")
        self.assertEqual(
            set(contract.expected_outputs(request)),
            {"figure_png", "plot_spec", "provenance"},
        )
        self.assertEqual(DEFAULT_CAPABILITIES, ("origin.plot@v2",))

    def test_adapter_version_and_features_gate_additive_requests(self) -> None:
        contract = get_contract("origin.plot@v2")
        legacy_runtime = {
            "available_slots": 1,
            "origin": {"health": "ready", "adapter_version": "0.3.0"},
        }
        self.assertFalse(node_supports_request(contract, _request("line"), legacy_runtime))
        self.assertFalse(node_supports_request(contract, _request("heatmap"), legacy_runtime))
        current_runtime = {
            "capability_runtime": {
                "origin.plot@v2": {
                    "health": "ready",
                    "available_slots": 1,
                    "adapter_version": "0.5.0",
                    "features": ["line", "heatmap"],
                }
            }
        }
        self.assertTrue(node_supports_request(contract, _request("heatmap"), current_runtime))
        self.assertFalse(node_supports_request(contract, _request("bar"), current_runtime))

    def test_semantic_version_comparison_is_numeric(self) -> None:
        self.assertTrue(version_at_least("0.10.0", "0.4.0"))
        self.assertFalse(version_at_least("0.3.9", "0.4.0"))


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