from datetime import datetime

import factory

from atlas_um.pgdb import DNAAccountActivity


class DNAAccountActivityFactory(factory.Factory):
    class Meta:
        model = DNAAccountActivity

    dna_account = factory.SubFactory(
        "tests.atlas_um.factories.DNAAccountFactory"
    )
    resource_group = factory.SubFactory(
        "tests.atlas_um.factories.ResourceGroupFactory"
    )

    first_login = factory.LazyFunction(datetime.now)
    last_login = factory.LazyFunction(datetime.now)
    last_activity = factory.LazyFunction(datetime.now)