Add unit tests

1 files changed, 68 insertions, 0 deletions

diff --git a/tests/feed.py b/tests/feed.py
new file mode 100644
index 0000000..aaeb96a
--- /dev/null
+++ b/tests/feed.py
@@ -0,0 +1,68 @@
+from typing import Any
+from mongomock import MongoClient
+from pymongo.collection import Collection
+from schedule import Scheduler
+from unittest import TestCase
+from components.subscriptions.main import Subscription
+from components.subscriptions.typing import SubsDict
+from components.users.typing import UserDict
+
+
+class TestFeeds(TestCase):
+    def setUp(self) -> None:
+        self.client: MongoClient[Any] = MongoClient(tz_aware=True)
+        self.collection: Collection[SubsDict] = self.client.db.collection
+        self.scheduler: Scheduler = Scheduler()
+
+    def test_insert(self) -> None:
+        sub = Subscription(
+            _id="yt:channel:bla",
+            link="http://www.youtube.com/feeds/videos.xml?channel_id=UCbla",
+            time_between_fetches=5,
+        )
+        sub._collection = self.collection
+        sub._scheduler = self.scheduler
+        sub.insert()
+        sub_dict = self.collection.find_one({"_id": "yt:channel:bla"})
+        self.assertIsNotNone(sub_dict)
+        assert sub_dict # To appease mypy.
+        self.assertDictEqual(sub.asdict(), sub_dict)
+
+    def test_feed_fetch(self) -> None:
+        sub = Subscription(
+            _id="yt:channel:hlgI3UHCOnwUGzWzbJ3H5w",
+            link=r"tests/data/feed@ytnnews24@001.xml",
+            time_between_fetches=1,
+        )
+        sub._collection = self.collection
+        sub._scheduler = self.scheduler
+        sub.insert()
+        sub.initialise_job()
+        self.scheduler.run_all()
+        self.assertEqual(15, len(sub.videos))
+        sub_dict = self.collection.find_one({"_id": "yt:channel:hlgI3UHCOnwUGzWzbJ3H5w"})
+        self.assertIsNotNone(sub_dict)
+        assert sub_dict # To appease mypy.
+        self.assertEqual(15, len(sub_dict["videos"]))
+
+    def test_feed_update(self) -> None:
+        sub = Subscription(
+            _id="yt:channel:hlgI3UHCOnwUGzWzbJ3H5w",
+            link=r"tests/data/feed@ytnnews24@001.xml",
+            time_between_fetches=1,
+        )
+        sub._collection = self.collection
+        sub._scheduler = self.scheduler
+        sub.insert()
+        sub.initialise_job()
+        self.scheduler.run_all()
+        sub.link=r"tests/data/feed@ytnnews24@002.xml"
+        self.scheduler.run_all()
+        self.assertEqual(16, len(sub.videos))
+        sub_dict = self.collection.find_one({"_id": "yt:channel:hlgI3UHCOnwUGzWzbJ3H5w"})
+        self.assertIsNotNone(sub_dict)
+        assert sub_dict # To appease mypy.
+        self.assertEqual(16, len(sub_dict["videos"]))
+
+    def tearDown(self) -> None:
+        self.client.close()