"""This module contains the factories for the ``saba`` discrete_app"""

import uuid

import factory

from mtp.apps.assessments.factories.certifications import CertificationProgramUserStatusFactory

from .models import CertificationProgramUserStatusDjangoTask


class CertificationProgramUserStatusDjangoTaskFactory(factory.django.DjangoModelFactory):
    """Factory for ``CertificationProgramUserStatusDjangoTask``"""

    class Meta:
        model = CertificationProgramUserStatusDjangoTask

    celery_task_id = factory.Sequence(lambda n: str(uuid.uuid4()))
    certification = factory.SubFactory(CertificationProgramUserStatusFactory)
