Add tests for APIs which fetch info

1 files changed, 45 insertions, 0 deletions

diff --git a/tests/integration/wsgi.py b/tests/integration/wsgi.py
index 4fbcd34..1f9ae70 100644
--- a/tests/integration/wsgi.py
+++ b/tests/integration/wsgi.py
@@ -65,5 +65,50 @@ class TestFlask(TestCase):
         })
         self.assertEqual(response.status_code, 409)
 
+    def test_subs_info(self) -> None:
+        # Should be empty at start.
+        response = self.client.get("/subs-info")
+        self.assertEqual(response.status_code, 200)
+        response_data = response.get_json()
+        self.assertListEqual(response_data, [])
+        # Fetching any individual subscription now should fail.
+        response = self.client.get("/sub-info/yt:channel:Ba659QWEk1AI4Tg--mrJ2A")
+        self.assertEqual(response.status_code, 404)
+        # Fetching the videos of an individual subscription should also fail.
+        response = self.client.get("/vid-from-link/yt:channel:Ba659QWEk1AI4Tg--mrJ2A")
+        self.assertEqual(response.status_code, 404)
+        # Now add some data to test on...
+        self.client.post("/add-sub/", data={
+            'url': "https://www.youtube.com/channel/UCBa659QWEk1AI4Tg--mrJ2A",
+            'time_between_fetches': 1,
+        })
+        self.client.post("/add-sub/", data={
+            'url': "https://www.youtube.com/playlist?list=PLZHQObOWTQDMsr9K-rj53DwVRMYO3t5Yr",
+            'time_between_fetches': 1,
+        })
+        # Now should contain 2 subscriptions.
+        response = self.client.get("/subs-info")
+        self.assertEqual(response.status_code, 200)
+        response_data = response.get_json()
+        self.assertEqual(len(response_data), 2)
+        for sub_info in response_data:
+            self.assertEqual(sub_info["videos"], 0)
+        # Now fetching an added subscriptions should work.
+        response = self.client.get("/sub-info/yt:channel:Ba659QWEk1AI4Tg--mrJ2A")
+        self.assertEqual(response.status_code, 200)
+        response_data = response.get_json()
+        self.assertEqual(response_data["_id"], "yt:channel:Ba659QWEk1AI4Tg--mrJ2A")
+        # Same for its videos - though they should still be empty.
+        response = self.client.get("/vid-from-link/yt:channel:Ba659QWEk1AI4Tg--mrJ2A")
+        self.assertEqual(response.status_code, 200)
+        response_data = response.get_json()
+        self.assertListEqual(response_data, [])
+        # Subscriptions not added should still not work.
+        response = self.client.get("/sub-info/yt:channel:7YOGHUfC1Tb6E4pudI9STA")
+        self.assertEqual(response.status_code, 404)
+        # Same for for their videos.
+        response = self.client.get("/vid-from-link/yt:channel:7YOGHUfC1Tb6E4pudI9STA")
+        self.assertEqual(response.status_code, 404)
+
     def tearDown(self) -> None:
         subscriptions.delete_many({})