Source code for mails.factories

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


import factory

# import pytz
# import random

from .models import MailLog, MAIL_NOT_RENDERED, MAIL_RENDERED

# from faker import Faker


class MailLogFactory(factory.django.DjangoModelFactory):
    processed = False
    status = MAIL_NOT_RENDERED
    body = ""
    body_html = ""

    from_email = factory.Faker("ascii_safe_email")
    mail_code = factory.Faker("slug")
    subject = factory.Faker("word")
    to_recipients = factory.List([factory.Faker("ascii_safe_email") for _ in range(2)])
    bcc_recipients = factory.List([factory.Faker("ascii_safe_email") for _ in range(2)])

    class Meta:
        model = MailLog


class RenderedMailLogFactory(MailLogFactory):
    processed = True
    status = MAIL_RENDERED
    body = factory.Faker("text")
    body_html = factory.Faker("text")