import factory
from faker import Faker

from atlas_um.pgdb import ClaimValue


class ClaimValueFactory(factory.Factory):
    class Meta:
        model = ClaimValue

    class Params:
        boolean = factory.Trait(
            # Can’t use factory.Faker:
            # https://github.com/FactoryBoy/factory_boy/pull/785
            components=factory.LazyFunction(lambda: Faker().pybool()),
        )

    claim_name = factory.SubFactory(
        "tests.atlas_um.factories.ClaimNameFactory"
    )
    friendly = factory.Sequence(lambda n: f"{Faker().pystr()}_{n}")
    components = factory.Faker(
        "pydict",
        variable_nb_elements=True,
        value_types=("str", "float", "int", "uri", "email"),
    )
    external_id = factory.Sequence(lambda n: f"{Faker().pystr()}_{n}")