"""Unit tests for Tenant.""" import uuid import pytest from permissions.types import Tenant, TenantType @pytest.mark.parametrize( 'tenant_type', [ (TenantType.ACCOUNT), (TenantType.SUBACCOUNT), (TenantType.LABEL_PARTICIPANT), (TenantType.COLLABORATOR), ], ) @pytest.mark.parametrize( 'tenant_uuid, expected', [ pytest.param(str(uuid.uuid4()), False), pytest.param( '053a1a75-acc5-4cd8-9206-a194335d2afa', True, ), ], ) def test_tenant_is_vendor_star(tenant_type: TenantType, tenant_uuid: str, expected: bool) -> None: """Test Tenant.is_vendor_star().""" tenant = Tenant(tenant_type=tenant_type, tenant_uuid=tenant_uuid) actual = tenant.is_vendor_star() assert actual == expected