From 0c1bdba85fd80160641bb2995c5ac51b6a5e3e1c Mon Sep 17 00:00:00 2001
From: Colin Goutte <cgte@bk.ru>
Date: Sun, 27 Aug 2023 18:22:29 +0200
Subject: [PATCH] Add a test for count

---
 utests/test_api.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/utests/test_api.py b/utests/test_api.py
index 462c74d..cef7ddd 100644
--- a/utests/test_api.py
+++ b/utests/test_api.py
@@ -112,6 +112,7 @@ class BaseCrud(unittest.TestCase):
     def test_list_movies(self):
         response = client.get("/movies/")
         assert response.status_code == 200
+        primary_count = response.json()["count"]
         # assert response.json() == []
 
         N = 10
@@ -124,16 +125,21 @@ class BaseCrud(unittest.TestCase):
             response = client.post("/movies/", json=self.create_payload)
             assert response.status_code == 200
 
-        movies = client.get("/movies/").json()["movies"]
+        response = client.get("/movies/").json()
 
+        movies = response["movies"]
+        count = response["count"]
         movies_by_title = {m["title"]: m for m in movies}
         found = list(movies_by_title[title] for title in names)
         assert all(movies_by_title[title] for title in names)
 
+        assert count == primary_count + N
+
     def test_list_movies_payload_format(self):
         response = client.get("/movies/")
         assert response.status_code == 200
         # assert response.json() == []
+        primary_count = response.json()["count"]
 
         N = 10
         names = []
@@ -149,6 +155,7 @@ class BaseCrud(unittest.TestCase):
 
         assert isinstance(movies["count"], int)
         assert isinstance(movies["movies"], list)
+        assert movies["count"] == primary_count + N
 
 
 class ApiTestCase(unittest.TestCase):