46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import debug # pyflakes:ignore
|
|
import factory
|
|
|
|
from ietf.group.models import Group, Role, GroupEvent
|
|
from ietf.review.factories import ReviewTeamSettingsFactory
|
|
|
|
class GroupFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Group
|
|
django_get_or_create = ('acronym',)
|
|
|
|
name = factory.Faker('sentence',nb_words=6)
|
|
acronym = factory.Sequence(lambda n: 'acronym%d' %n)
|
|
state_id = 'active'
|
|
type_id = 'wg'
|
|
|
|
class ReviewTeamFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Group
|
|
|
|
type_id = 'dir'
|
|
name = factory.Faker('sentence',nb_words=6)
|
|
acronym = factory.Sequence(lambda n: 'acronym%d' %n)
|
|
state_id = 'active'
|
|
|
|
@factory.post_generation
|
|
def settings(obj, create, extracted, **kwargs):
|
|
ReviewTeamSettingsFactory.create(group=obj,**kwargs)
|
|
|
|
class RoleFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Role
|
|
|
|
group = factory.SubFactory(GroupFactory)
|
|
person = factory.SubFactory('ietf.person.factories.PersonFactory')
|
|
email = factory.LazyAttribute(lambda obj: obj.person.email())
|
|
|
|
class GroupEventFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = GroupEvent
|
|
|
|
group = factory.SubFactory(GroupFactory)
|
|
by = factory.SubFactory('ietf.person.factories.PersonFactory')
|
|
type = 'comment'
|
|
desc = factory.Faker('paragraph')
|