fix test for count_videos

EsupPortail · Nov 23, 2021 · 8f13475 · 8f13475
1 parent 77502db
commit 8f13475
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pod/video/tests/test_views.py b/pod/video/tests/test_views.py
@@ -82,6 +82,7 @@ def test_regroup_videos_by_theme(self):
             "has_more_themes": False,
             "has_more_videos": False,
             "videos": [self.v],
+            "count_videos": 1,
             "count_themes": 1,
             "theme": None,
             "channel": self.c,
@@ -126,6 +127,7 @@ def test_regroup_videos_by_theme(self):
         )
         expected.pop("next_videos", None)
         expected.pop("has_more_videos", None)
+        expected.pop("count_videos", None)
         self.assertEqual(response.status_code, HTTPStatus.OK)
         self.assertCountEqual(expected, response.json())
 
@@ -138,6 +140,7 @@ def test_regroup_videos_by_theme(self):
         )
         expected["next_videos"] = None
         expected["has_more_videos"] = False
+        expected["count_videos"] = 1
 
         expected.pop("next", None)
         expected.pop("previous", None)