From 7acc292badb96ee003105419ccf33711a72f2a25 Mon Sep 17 00:00:00 2001 From: faraphel Date: Mon, 4 Dec 2023 20:00:55 +0100 Subject: [PATCH] fixed Absence factory not working --- Palto/Palto/factories.py | 20 +++----------------- 1 file changed, 3 insertions(+), 17 deletions(-) diff --git a/Palto/Palto/factories.py b/Palto/Palto/factories.py index 54def7a..21a7baa 100644 --- a/Palto/Palto/factories.py +++ b/Palto/Palto/factories.py @@ -172,24 +172,10 @@ class FakeAbsenceFactory(factory.django.DjangoModelFactory): message: str = factory.Faker("paragraph") + department: models.Department = factory.SubFactory(FakeDepartmentFactory) student: models.User = factory.SubFactory(FakeUserFactory) - - @factory.post_generation - def sessions(self, create, extracted, **kwargs): - if not create: - return - - if extracted is not None: - self.sessions.add(*extracted) - else: - # all the sessions should be in the same department - department = FakeDepartmentFactory() - - # create a group of between 1 and 8 sessions from the department - self.sessions.add(*[ - FakeTeachingSessionFactory.create(unit__department=department) - for _ in range(random.randint(1, 8)) - ]) + start: datetime = factory.Faker("date_time", tzinfo=timezone.get_current_timezone()) + end: datetime = factory.LazyAttribute(lambda obj: obj.start + timedelta(days=random.randint(1, 8))) class FakeAbsenceAttachmentFactory(factory.django.DjangoModelFactory):