test(tasks/tests): fixing tests

unb-mds · Dec 5, 2023 · 5b10cbd · 5b10cbd
1 parent 4580ec0
commit 5b10cbd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/apps/tasks/tests.py b/apps/tasks/tests.py
@@ -355,8 +355,13 @@ def test_send_submission_successfully(self) -> None:
     def test_send_submission_with_short_code(self) -> None:
         self.client.force_login(self.user)
 
+        initial_submission_count = Submission._default_manager.count()
+
         self.client.post(self.url, data={"code": "c"})
-        self.assertEqual(Submission._default_manager.count(), 1)
+
+        final_submission_count = Submission._default_manager.count()
+
+        self.assertEqual(final_submission_count, initial_submission_count + 1)
 
     def test_detail_view_model_is_task(self) -> None:
         self.assertEqual(DetailView.model, Task)