Use factories for ipr tests. Commit ready for merge.

 - Legacy-Id: 15492

Author: rjsparks

ietf/doc/factories.py

@@ -55,6 +55,14 @@ def authors(obj, create, extracted, **kwargs): # pylint: disable=no-self-argumen
                 DocumentAuthor.objects.create(document=obj, person=person, email=person.email(), order=order)
                 order += 1
 
+    @factory.post_generation
+    def relations(obj, create, extracted, **kwargs): # pylint: disable=no-self-argument
+        if create and extracted:
+            for (rel_id,docalias) in extracted:
+                if isinstance(docalias,Document):
+                    docalias = docalias.docalias_set.first()
+                obj.relateddocument_set.create(relationship_id=rel_id,target=docalias)
+
     @classmethod
     def _after_postgeneration(cls, obj, create, results=None):
         """Save again the instance if creating and at least one hook ran."""

ietf/ipr/factories.py

@@ -4,7 +4,7 @@
 
 from ietf.ipr.models import (
     IprDisclosureBase, HolderIprDisclosure, ThirdPartyIprDisclosure, NonDocSpecificIprDisclosure,
-    GenericIprDisclosure, IprDocRel
+    GenericIprDisclosure, IprDocRel, RelatedIpr
 )
 
 def _fake_patent_info():
@@ -36,6 +36,14 @@ def docs(self, create, extracted, **kwargs):
             for doc in extracted:
                 IprDocRel.objects.create(disclosure=self,document=doc.docalias_set.first())
 
+    @factory.post_generation
+    def updates(self, create, extracted, **kwargs):
+        if not create:
+            return
+        if extracted:
+            for ipr in extracted:
+                RelatedIpr.objects.create(source=self,target=ipr,relationship_id='updates')
+
 
 class HolderIprDisclosureFactory(IprDisclosureBaseFactory):
     class Meta: