from uuid import UUID from pdp.fastapi.schemas.tenant import ( UuidToIdExchangeTenant, ) def test_uuid_to_id_exchange_tenant_to_key_vals_tuple() -> None: """Test to_key_vals_tuple.""" obj = UuidToIdExchangeTenant.model_validate( { "tenant_uuid": UUID("fff741c2-6def-4493-bfdf-c2bcb1128e02"), "tenant_type": "account", } ) assert obj.to_key_vals_tuple() == [ ("tenant_id", "*"), ("tenant_type", "account"), ("tenant_uuid", "fff741c2-6def-4493-bfdf-c2bcb1128e02"), ] def test_uuid_to_id_exchange_tenant_to_cache_key() -> None: """Test to_cache_key.""" obj = UuidToIdExchangeTenant.model_validate( { "tenant_uuid": UUID("fff741c2-6def-4493-bfdf-c2bcb1128e02"), "tenant_type": "account", } ) assert ( obj.to_cache_key() == "tenant_hierarchy@tenant_type#account|tenant_uuid#fff741c2-6def-4493-bfdf-c2bcb1128e02" # noqa: E501 )