from datetime import datetime

import factory

from atlas_um.pgdb import DNAAccount


class DNAAccountFactory(factory.Factory):
    class Meta:
        model = DNAAccount

    sub = factory.Faker("pystr")
    expiration_date = factory.Faker("future_date")
    created_at = factory.LazyFunction(datetime.now)
    updated_at = factory.LazyFunction(datetime.now)
    business_unit = factory.SubFactory(
        "tests.atlas_um.factories.BusinessUnitFactory"
    )
    job_category = factory.SubFactory(
        "tests.atlas_um.factories.JobCategoryFactory"
    )
    personnel_type = factory.SubFactory(
        "tests.atlas_um.factories.PersonnelTypeFactory"
    )
    supervisor_email = factory.Faker("email")
    supervisor_name = factory.Faker("name")
    job_title = factory.Faker("pystr")
    location = factory.Faker("pystr")
    preferred_username = factory.Faker("pystr")
    given_name = factory.Faker("first_name")
    family_name = factory.Faker("last_name")
    email = factory.Faker("email")
    usm_account = factory.Faker("pystr")