add with_tags params to ForumFactory

gip-inclusion · Aug 8, 2024 · 54bcae7 · 54bcae7
1 parent 287d2f6
commit 54bcae7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lacommunaute/forum/factories.py b/lacommunaute/forum/factories.py
@@ -38,6 +38,15 @@ def upvoted_by(self, create, extracted, **kwargs):
         for user in extracted:
             UpVote.objects.create(voter=user, content_object=self)
 
+    @factory.post_generation
+    def with_tags(self, create, extracted, **kwargs):
+        if not create or not extracted:
+            return
+
+        if isinstance(extracted, list):
+            for tag in extracted:
+                self.tags.add(tag)
+
 
 class CategoryForumFactory(ForumFactory):
     type = Forum.FORUM_CAT