0b3f9f5dc5
* refactor: clarify handling what groups looked like at past session times * fix: replace missed instance of historic_parent reference * fix: reflect that group_at_the_time always returns something * chore: update copyright lines
121 lines
4.4 KiB
Python
121 lines
4.4 KiB
Python
# Copyright The IETF Trust 2015-2022, All Rights Reserved
|
|
import datetime
|
|
import debug # pyflakes:ignore
|
|
import factory
|
|
|
|
from typing import List # pyflakes:ignore
|
|
|
|
from django.utils import timezone
|
|
|
|
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 = timezone.now()+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
|
|
|
|
time = lambda: timezone.now()
|
|
group = factory.SubFactory(GroupFactory, state_id='active')
|
|
|
|
name = factory.LazyAttribute(lambda obj: obj.group.name)
|
|
state_id = factory.LazyAttribute(lambda obj: obj.group.state_id)
|
|
type_id = factory.LazyAttribute(lambda obj: obj.group.type_id)
|
|
parent = factory.LazyAttribute(lambda obj: obj.group.parent)
|
|
uses_milestone_dates = factory.LazyAttribute(lambda obj: obj.group.uses_milestone_dates)
|
|
used_roles = factory.LazyAttribute(lambda obj: obj.group.used_roles)
|
|
description = factory.LazyAttribute(lambda obj: obj.group.description)
|
|
list_email = factory.LazyAttribute(lambda obj: '%s@ietf.org'% obj.group.acronym) #TODO : move this to GroupFactory
|
|
list_subscribe = factory.LazyAttribute(lambda obj: obj.group.list_subscribe)
|
|
list_archive = factory.LazyAttribute(lambda obj: obj.group.list_archive)
|
|
comments = factory.LazyAttribute(lambda obj: obj.group.comments)
|
|
meeting_seen_as_area = factory.LazyAttribute(lambda obj: obj.group.meeting_seen_as_area)
|
|
acronym = factory.LazyAttribute(lambda obj: obj.group.acronym)
|
|
|
|
@factory.post_generation
|
|
def unused_states(obj, create, extracted, **kwargs):
|
|
if create:
|
|
if extracted:
|
|
obj.unused_states.set(extracted)
|
|
else:
|
|
obj.unused_states.set(obj.group.unused_states.all())
|
|
@factory.post_generation
|
|
def unused_tags(obj, create, extracted, **kwargs):
|
|
if create:
|
|
if extracted:
|
|
obj.unused_tags.set(extracted)
|
|
else:
|
|
obj.unused_tags.set(obj.group.unused_states.all())
|
|
|
|
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())
|
|
|