From 30aab3fb9d6c8ae780ae277b20b5967fffa0871c Mon Sep 17 00:00:00 2001 From: lebaudantoine Date: Mon, 16 Dec 2024 19:52:21 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A8(backend)=20fix=20Django=20deprecat?= =?UTF-8?q?ion=20warning=20in=20Factory?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit _after_postgeneration method will stop saving the instance after postgeneration hooks in the next major release. Solved using Claude, feel free to challenge my fix. --- src/backend/core/factories.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/backend/core/factories.py b/src/backend/core/factories.py index f1ce85909..d37a9265f 100644 --- a/src/backend/core/factories.py +++ b/src/backend/core/factories.py @@ -19,6 +19,7 @@ class UserFactory(factory.django.DjangoModelFactory): class Meta: model = models.User + skip_postgeneration_save = True sub = factory.Sequence(lambda n: f"user{n!s}") email = factory.Faker("email") @@ -36,6 +37,8 @@ class UserFactory(factory.django.DjangoModelFactory): if create and (extracted is True): UserDocumentAccessFactory(user=self, role="owner") + self.save() + @factory.post_generation def with_owned_template(self, create, extracted, **kwargs): """ @@ -45,6 +48,8 @@ class UserFactory(factory.django.DjangoModelFactory): if create and (extracted is True): UserTemplateAccessFactory(user=self, role="owner") + self.save() + class DocumentFactory(factory.django.DjangoModelFactory): """A factory to create documents"""