# Copyright The IETF Trust 2015-2020, All Rights Reserved import datetime import debug # pyflakes:ignore import factory from typing import List # pyflakes:ignore from ietf.group.models import Group, Role, GroupEvent, GroupMilestone, \ GroupHistory, RoleHistory from ietf.review.factories import ReviewTeamSettingsFactory class GroupFactory(factory.django.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' list_email = factory.LazyAttribute(lambda a: '%s@ietf.org'% a.acronym) uses_milestone_dates = True used_roles = [] # type: List[str] @factory.lazy_attribute def parent(self): if self.type_id in ['wg','ag']: return GroupFactory(type_id='area') elif self.type_id in ['rg','rag']: return GroupFactory(acronym='irtf', type_id='irtf') else: return None class ReviewTeamFactory(GroupFactory): type_id = 'review' @factory.post_generation def settings(obj, create, extracted, **kwargs): ReviewTeamSettingsFactory.create(group=obj,**kwargs) class RoleFactory(factory.django.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.django.DjangoModelFactory): class Meta: model = GroupEvent group = factory.SubFactory(GroupFactory) by = factory.SubFactory('ietf.person.factories.PersonFactory') type = 'comment' desc = factory.Faker('paragraph') class BaseGroupMilestoneFactory(factory.django.DjangoModelFactory): class Meta: model = GroupMilestone group = factory.SubFactory(GroupFactory) state_id = 'active' desc = factory.Faker('sentence') class DatedGroupMilestoneFactory(BaseGroupMilestoneFactory): group = factory.SubFactory(GroupFactory, uses_milestone_dates=True) due = datetime.datetime.today()+datetime.timedelta(days=180) class DatelessGroupMilestoneFactory(BaseGroupMilestoneFactory): group = factory.SubFactory(GroupFactory, uses_milestone_dates=False) order = factory.Sequence(lambda n: n) class GroupHistoryFactory(factory.django.DjangoModelFactory): class Meta: model=GroupHistory name = factory.LazyAttribute(lambda obj: obj.group.name) state_id = 'active' type_id = factory.LazyAttribute(lambda obj: obj.group.type_id) list_email = factory.LazyAttribute(lambda obj: '%s@ietf.org'% obj.group.acronym) uses_milestone_dates = True used_roles = [] # type: List[str] group = factory.SubFactory(GroupFactory) acronym = factory.LazyAttribute(lambda obj: obj.group.acronym) class RoleHistoryFactory(factory.django.DjangoModelFactory): class Meta: model=RoleHistory group = factory.SubFactory(GroupHistoryFactory) person = factory.SubFactory('ietf.person.factories.PersonFactory') email = factory.LazyAttribute(lambda obj: obj.person.email())