add with_tags params to ForumFactory

gip-inclusion · Aug 7, 2024 · 0192545 · 0192545
1 parent cb8f9ed
commit 0192545
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