acdl/tests/test_contract_resolver.py

import json
import os
import sys
from pathlib import Path

import jsonschema
import pytest
import yaml

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

ROOT = Path(__file__).resolve().parent.parent


class TestContractSchema:
    def test_schema_is_valid_json_schema(self):
        schema = json.load(open(ROOT / "schemas/contract.schema.json"))
        jsonschema.Draft202012Validator.check_schema(schema)

    def test_schema_requires_uses_module_environment_inputs(self):
        schema = json.load(open(ROOT / "schemas/contract.schema.json"))
        for field in ["uses", "module", "environment", "inputs"]:
            assert field in schema["required"]


class TestResolveStaticAsset:
    def test_resolve_static_asset_contract(self, tmp_path):
        from core.contract_resolver import resolve
        stack = resolve(str(ROOT / "contracts/static-asset.yaml"), str(ROOT))
        assert stack["stack"]["name"] == "static-asset"
        assert stack["stack"]["kind"] == "l2"
        assert stack["stack"]["depth"] == 1
        assert len(stack["resources"]) >= 1

    def test_resolve_static_asset_has_s3_resource(self):
        from core.contract_resolver import resolve
        stack = resolve(str(ROOT / "contracts/static-asset.yaml"), str(ROOT))
        s3_res = [r for r in stack["resources"] if r["type"] == "aws:s3:bucket"]
        assert len(s3_res) == 1
        assert s3_res[0]["inputs"]["bucket_name"] == "acdl-spike-bucket"
        assert s3_res[0]["inputs"]["region"] == "us-east-1"

    def test_resolve_static_asset_validates_against_stack_schema(self):
        from core.contract_resolver import resolve
        stack = resolve(str(ROOT / "contracts/static-asset.yaml"), str(ROOT))
        schema = json.load(open(ROOT / "schemas/stack.schema.json"))
        jsonschema.validate(stack, schema)


class TestResolveMicroservice:
    def test_resolve_microservice_contract(self):
        contract = {
            "uses": "acdl/pipelines/deploy.yaml@v1",
            "module": "microservice",
            "environment": "dev",
            "inputs": {
                "image": "581513795199.dkr.ecr.us-east-1.amazonaws.com/acdl-microservice:latest",
                "port": 8080,
                "region": "us-east-1",
            },
        }
        contract_path = ROOT / "contracts" / "test-microservice.yaml"
        with open(contract_path, "w") as fh:
            yaml.dump(contract, fh)
        try:
            from core.contract_resolver import resolve
            stack = resolve(str(contract_path), str(ROOT))
            assert stack["stack"]["name"] == "microservice"
            assert stack["stack"]["kind"] == "l2"
            assert len(stack["resources"]) >= 6
        finally:
            os.remove(contract_path)


class TestResolveL1Direct:
    def test_resolve_s3_direct(self, tmp_path):
        contract = {
            "uses": "acdl/pipelines/deploy.yaml@v1",
            "module": "s3",
            "environment": "dev",
            "inputs": {
                "bucket_name": "my-test-bucket",
                "region": "us-east-1",
            },
        }
        contract_path = tmp_path / "test-s3.yaml"
        with open(contract_path, "w") as fh:
            yaml.dump(contract, fh)
        from core.contract_resolver import resolve
        stack = resolve(str(contract_path), str(ROOT))
        assert stack["stack"]["name"] == "s3"
        assert stack["stack"]["kind"] == "l1"
        assert len(stack["resources"]) == 1
        assert stack["resources"][0]["type"] == "aws:s3:bucket"
        assert stack["resources"][0]["inputs"]["bucket_name"] == "my-test-bucket"

    def test_resolve_s3_validates_against_stack_schema(self, tmp_path):
        contract = {
            "uses": "acdl/pipelines/deploy.yaml@v1",
            "module": "s3",
            "environment": "dev",
            "inputs": {"bucket_name": "test", "region": "us-east-1"},
        }
        contract_path = tmp_path / "test-s3-schema.yaml"
        with open(contract_path, "w") as fh:
            yaml.dump(contract, fh)
        from core.contract_resolver import resolve
        stack = resolve(str(contract_path), str(ROOT))
        schema = json.load(open(ROOT / "schemas/stack.schema.json"))
        jsonschema.validate(stack, schema)


class TestResolveErrors:
    def test_unknown_module_raises(self, tmp_path):
        contract = {
            "uses": "acdl/pipelines/deploy.yaml@v1",
            "module": "nonexistent",
            "environment": "dev",
            "inputs": {},
        }
        contract_path = tmp_path / "bad.yaml"
        with open(contract_path, "w") as fh:
            yaml.dump(contract, fh)
        from core.contract_resolver import resolve
        with pytest.raises(ValueError, match="not found in registry"):
            resolve(str(contract_path), str(ROOT))

    def test_missing_required_field_fails_validation(self, tmp_path):
        contract = {"uses": "acdl/pipelines/deploy.yaml@v1", "module": "s3"}
        contract_path = tmp_path / "incomplete.yaml"
        with open(contract_path, "w") as fh:
            yaml.dump(contract, fh)
        from core.contract_resolver import resolve
        with pytest.raises(jsonschema.ValidationError):
            resolve(str(contract_path), str(ROOT))


class TestDeployPipelineContract:
    def test_deploy_pipeline_validates_against_schema(self):
        schema = json.load(open(ROOT / "schemas/deploy-pipeline.schema.json"))
        with open(ROOT / "pipelines/deploy.yaml") as fh:
            contract = yaml.safe_load(fh)
        jsonschema.validate(contract, schema)

    def test_deploy_pipeline_has_six_stages(self):
        with open(ROOT / "pipelines/deploy.yaml") as fh:
            contract = yaml.safe_load(fh)
        stage_names = [s["name"] for s in contract["stages"]]
        assert "validate-contract" in stage_names
        assert "resolve-stack" in stage_names
        assert "terraform-plan" in stage_names
        assert "checkov" in stage_names
        assert "confidence" in stage_names
        assert "apply" in stage_names