feat(forum): add upvotes_count methode on Forum model

gip-inclusion · Jul 25, 2023 · 18f999c · 18f999c
1 parent ddec93f
commit 18f999c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lacommunaute/forum/models.py b/lacommunaute/forum/models.py
@@ -48,3 +48,6 @@ def get_unanswered_topics(self):
     @cached_property
     def count_unanswered_topics(self):
         return self.get_unanswered_topics().count()
+
+    def upvotes_count(self):
+        return self.upvotes.count()
diff --git a/lacommunaute/forum/tests/tests_model.py b/lacommunaute/forum/tests/tests_model.py
@@ -5,6 +5,7 @@
 from lacommunaute.forum.factories import ForumFactory
 from lacommunaute.forum.models import Forum
 from lacommunaute.forum_conversation.factories import TopicFactory
+from lacommunaute.users.factories import UserFactory
 
 
 class ForumManagerTest(TestCase):
@@ -50,3 +51,9 @@ def test_get_absolute_url(self):
             forum.get_absolute_url(),
             f"/forum/{forum.slug}-{forum.pk}/",
         )
+
+    def test_upvotes_count(self):
+        forum = ForumFactory()
+        self.assertEqual(forum.upvotes_count(), 0)
+        forum.upvotes.create(voter=UserFactory())
+        self.assertEqual(forum.upvotes_count(), 1)