"""Regression tests for #28156 — Bedrock picker must be region-scoped.

Geo-prefixed cross-region inference profiles (us.*, eu.*, apac.*, ...) only
route from endpoints in their own geography. Offering us.* profiles to an
eu-central-2 picker produces configs AWS rejects regardless of credentials.
"""

from hermes_cli.model_setup_flows import (
    BEDROCK_GEO_PREFIXES,
    bedrock_model_routable_from_region,
    bedrock_region_geo_prefix,
)


class TestRegionGeoPrefix:
    def test_known_geographies(self):
        assert bedrock_region_geo_prefix("us-east-1") == "us."
        assert bedrock_region_geo_prefix("eu-central-2") == "eu."
        assert bedrock_region_geo_prefix("ap-southeast-1") == "ap."
        assert bedrock_region_geo_prefix("ca-central-1") == "ca."
        assert bedrock_region_geo_prefix("sa-east-1") == "sa."
        assert bedrock_region_geo_prefix("me-south-1") == "me."
        assert bedrock_region_geo_prefix("af-south-1") == "af."

    def test_unknown_region_is_empty(self):
        assert bedrock_region_geo_prefix("") == ""
        assert bedrock_region_geo_prefix("moon-base-1") == ""


class TestRoutableFromRegion:
    def test_us_profile_not_offered_in_eu(self):
        assert not bedrock_model_routable_from_region(
            "us.anthropic.claude-sonnet-4-6", "eu-central-2"
        )




class TestGeoPrefixContract:
    def test_every_geo_prefix_maps_to_a_routable_region_or_is_alias(self):
        """Invariant: each geo prefix is either reachable from some region's
        geo mapping or an ap-family alias — no dead entries."""
        mapped = {
            bedrock_region_geo_prefix(r)
            for r in (
                "us-east-1", "eu-central-2", "ap-southeast-1",
                "ca-central-1", "sa-east-1", "me-south-1", "af-south-1",
            )
        }
        ap_aliases = {"apac.", "jp."}
        for prefix in BEDROCK_GEO_PREFIXES:
            assert prefix in mapped or prefix in ap_aliases