import factory
from faker import Faker

from atlas_um.pgdb import ClaimName
from atlas_um.tokens.claim_values_serializers import RegisteredSerializers


class ClaimNameFactory(factory.Factory):
    class Meta:
        model = ClaimName

    resource_group = factory.SubFactory(
        "tests.atlas_um.factories.ResourceGroupFactory"
    )
    friendly = factory.Sequence(lambda n: f"{Faker().pystr()}_{n}")
    path = factory.Sequence(lambda n: f"{Faker().uri_path()}_{n}")
    external_id = factory.Sequence(lambda n: f"{Faker().pystr()}_{n}")
    claim_values_serializer = RegisteredSerializers.BooleanAnd.name