From 24cc32a698f828dabf1e45882974926be22a4aa0 Mon Sep 17 00:00:00 2001
From: Geert Kapteijns <ghkapteijns@gmail.com>
Date: Thu, 15 Dec 2016 16:37:16 +0100
Subject: [PATCH] Update ContributorFactory

You can now just call ContributorFactory() to get a freshly instantiated
Contributor to be used for testing purposes.
---
 scipost/factories.py | 2 ++
 theses/factories.py  | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/scipost/factories.py b/scipost/factories.py
index 05f9239b6..80a4c0458 100644
--- a/scipost/factories.py
+++ b/scipost/factories.py
@@ -12,6 +12,8 @@ class ContributorFactory(factory.django.DjangoModelFactory):
 
     title = "MR"
     user = factory.SubFactory(UserFactory, contributor=None)
+    status = 1
+    vetted_by = factory.SubFactory(ContributorFactory)
 
 
 class UserFactory(factory.django.DjangoModelFactory):
diff --git a/theses/factories.py b/theses/factories.py
index 60e67d1ae..522f63b96 100644
--- a/theses/factories.py
+++ b/theses/factories.py
@@ -1,12 +1,13 @@
 import factory
 from .models import ThesisLink
+from scipost.factories import UserFactory, ContributorFactory
 
 
 class ThesisLinkFactory(factory.django.DjangoModelFactory):
     class Meta:
         model = ThesisLink
 
-    # requested_by = factory.SubFactory(ContributorFactory)
+    requested_by = factory.SubFactory(ContributorFactory)
     type = ThesisLink.MASTER_THESIS
     title = factory.Sequence(lambda n: "thesis {0}".format(n))
     pub_link = factory.Faker('uri')
-- 
GitLab