From 081c2deb0d82905681b03efc7f177d601e863be0 Mon Sep 17 00:00:00 2001
From: Jorran de Wit <jorrandewit@outlook.com>
Date: Tue, 6 Jun 2017 15:03:47 +0200
Subject: [PATCH] Add Report and RefereeInvitation factories
---
submissions/factories.py | 25 ++++++++++++++++++++++++-
1 file changed, 24 insertions(+), 1 deletion(-)
diff --git a/submissions/factories.py b/submissions/factories.py
index e3d419e4b..d9c32c745 100644
--- a/submissions/factories.py
+++ b/submissions/factories.py
@@ -12,7 +12,7 @@ from .constants import STATUS_UNASSIGNED, STATUS_EIC_ASSIGNED, STATUS_RESUBMISSI
STATUS_PUBLISHED, SUBMISSION_TYPE, STATUS_RESUBMITTED, STATUS_VETTED,\
REFEREE_QUALIFICATION, RANKING_CHOICES, QUALITY_SPEC, REPORT_REC,\
REPORT_STATUSES, STATUS_UNVETTED, STATUS_DRAFT
-from .models import Submission, Report
+from .models import Submission, Report, RefereeInvitation
from faker import Faker
@@ -182,3 +182,26 @@ class UnVettedReportFactory(ReportFactory):
class VettedReportFactory(ReportFactory):
status = STATUS_VETTED
+
+
+class RefereeInvitationFactory(factory.django.DjangoModelFactory):
+ class Meta:
+ model = RefereeInvitation
+
+ submission = factory.SubFactory('submissions.factories.SubmissionFactory')
+ referee = factory.Iterator(Contributor.objects.all())
+
+ invitation_key = factory.Faker('md5')
+ invited_by = factory.Iterator(Contributor.objects.all())
+
+ @factory.post_generation
+ def contributor_fields(self, create, extracted, **kwargs):
+ self.title = self.referee.title
+ self.first_name = self.referee.user.first_name
+ self.last_name = self.referee.user.last_name
+ self.email_address = self.referee.user.email
+
+
+class AcceptedRefereeInvitationFactory(RefereeInvitationFactory):
+ accepted = True
+ date_responded = Faker().date_time_between(start_date="-1y", end_date="now", tzinfo=pytz.UTC)
--
GitLab