93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
import factory
|
|
|
|
from ietf.doc.models import Document, DocEvent, NewRevisionDocEvent, DocAlias, State, DocumentAuthor
|
|
|
|
def draft_name_generator(type_id,group,n):
|
|
return '%s-%s-%s-%s%d'%(
|
|
type_id,
|
|
'bogusperson',
|
|
group.acronym if group else 'netherwhere',
|
|
'musings',
|
|
n,
|
|
)
|
|
|
|
class DocumentFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Document
|
|
|
|
type_id = 'draft'
|
|
title = factory.Faker('sentence',nb_words=6)
|
|
rev = '00'
|
|
group = factory.SubFactory('ietf.group.factories.GroupFactory',type_id='individ')
|
|
std_level_id = None
|
|
intended_std_level_id = None
|
|
|
|
@factory.lazy_attribute_sequence
|
|
def name(self, n):
|
|
return draft_name_generator(self.type_id,self.group,n)
|
|
|
|
newrevisiondocevent = factory.RelatedFactory('ietf.doc.factories.NewRevisionDocEventFactory','doc')
|
|
|
|
alias = factory.RelatedFactory('ietf.doc.factories.DocAliasFactory','document')
|
|
|
|
@factory.post_generation
|
|
def other_aliases(obj, create, extracted, **kwargs): # pylint: disable=no-self-argument
|
|
if create and extracted:
|
|
for alias in extracted:
|
|
obj.docalias_set.create(name=alias)
|
|
|
|
@factory.post_generation
|
|
def states(obj, create, extracted, **kwargs): # pylint: disable=no-self-argument
|
|
if create and extracted:
|
|
for (state_type_id,state_slug) in extracted:
|
|
obj.set_state(State.objects.get(type_id=state_type_id,slug=state_slug))
|
|
|
|
@factory.post_generation
|
|
def authors(obj, create, extracted, **kwargs): # pylint: disable=no-self-argument
|
|
if create and extracted:
|
|
order = 0
|
|
for email in extracted:
|
|
DocumentAuthor.objects.create(document=obj, author=email, order=order)
|
|
order += 1
|
|
|
|
@classmethod
|
|
def _after_postgeneration(cls, obj, create, results=None):
|
|
"""Save again the instance if creating and at least one hook ran."""
|
|
if create and results:
|
|
# Some post-generation hooks ran, and may have modified us.
|
|
obj._has_an_event_so_saving_is_allowed = True
|
|
obj.save()
|
|
|
|
|
|
class DocAliasFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = DocAlias
|
|
|
|
document = factory.SubFactory('ietf.doc.factories.DocumentFactory')
|
|
|
|
@factory.lazy_attribute
|
|
def name(self):
|
|
return self.document.name
|
|
|
|
|
|
class DocEventFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = DocEvent
|
|
|
|
type = 'added_comment'
|
|
by = factory.SubFactory('ietf.person.factories.PersonFactory')
|
|
doc = factory.SubFactory(DocumentFactory)
|
|
desc = factory.Faker('sentence',nb_words=6)
|
|
|
|
class NewRevisionDocEventFactory(DocEventFactory):
|
|
class Meta:
|
|
model = NewRevisionDocEvent
|
|
|
|
type = 'new_revision'
|
|
rev = '00'
|
|
|
|
@factory.lazy_attribute
|
|
def desc(self):
|
|
return 'New version available %s-%s'%(self.doc.name,self.rev)
|
|
|