46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import factory
|
|
|
|
from ietf.doc.models import Document, DocEvent, NewRevisionDocEvent
|
|
from ietf.person.factories import PersonFactory
|
|
|
|
class DocumentFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Document
|
|
|
|
type_id = 'draft'
|
|
title = factory.Faker('sentence',nb_words=6)
|
|
rev = '00'
|
|
group = None
|
|
|
|
@factory.lazy_attribute_sequence
|
|
def name(self, n):
|
|
return '%s-%s-%s-%s%d'%(
|
|
self.type_id,
|
|
'bogusperson',
|
|
self.group.acronym if self.group else 'netherwhere',
|
|
'musings',
|
|
n,
|
|
)
|
|
|
|
newrevisiondocevent = factory.RelatedFactory('ietf.doc.factories.NewRevisionDocEventFactory','doc')
|
|
|
|
class DocEventFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = DocEvent
|
|
|
|
type = 'added_comment'
|
|
by = factory.SubFactory(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)
|