Add some unit tests for extract_sub_info

1 files changed, 44 insertions, 0 deletions

diff --git a/tests/extract_sub_info.py b/tests/extract_sub_info.py
new file mode 100644
index 0000000..7ade470
--- /dev/null
+++ b/tests/extract_sub_info.py
@@ -0,0 +1,44 @@
+from unittest import TestCase
+from unittest.mock import MagicMock, patch
+
+from components.extractor.extract_sub_info import get_feed_details, get_channel_feed, get_playlist_feed
+
+class TestExtractSubInfo(TestCase):
+    def test_get_feed_details(self) -> None:
+        self.assertEqual(get_feed_details("tests/data/feed@gentoo@mentaloutlaw@001.xml"), {
+            "id": "yt:playlist:PL3cu45aM3C2CADmCYeVhS4KTVut9MoMc9",
+            "link": "http://www.youtube.com/feeds/videos.xml?playlist_id=PL3cu45aM3C2CADmCYeVhS4KTVut9MoMc9",
+            "title": "Gentoo",
+        })
+        self.assertEqual(get_feed_details("tests/data/feed@mentaloutlaw@001.xml"), {
+            "id": "yt:channel:7YOGHUfC1Tb6E4pudI9STA",
+            "link": "http://www.youtube.com/feeds/videos.xml?channel_id=UC7YOGHUfC1Tb6E4pudI9STA",
+            "title": "Mental Outlaw",
+        })
+        self.assertEqual(get_feed_details("tests/data/feed@ytnnews24@001.xml"), {
+            "id": "yt:channel:hlgI3UHCOnwUGzWzbJ3H5w",
+            "link": "http://www.youtube.com/feeds/videos.xml?channel_id=UChlgI3UHCOnwUGzWzbJ3H5w",
+            "title": "YTN",
+        })
+        self.assertEqual(get_feed_details("tests/data/feed@ytnnews24@002.xml"), {
+            "id": "yt:channel:hlgI3UHCOnwUGzWzbJ3H5w",
+            "link": "http://www.youtube.com/feeds/videos.xml?channel_id=UChlgI3UHCOnwUGzWzbJ3H5w",
+            "title": "YTN",
+        })
+
+    def test_get_channel_feed(self) -> None:
+        with open("tests/data/channel@aljazeera.html", 'r') as file:
+            channel_html = file.read()
+        self.assertEqual("https://www.youtube.com/feeds/videos.xml?channel_id=UCfiwzLy-8yKzIbsmZTzxDgw",
+                         get_channel_feed(url='https://www.youtube.com/@aljazeera', html=channel_html))
+        for tab in ('videos', 'shorts', 'streams', 'playlists', 'posts'):
+            with open("tests/data/channel@rossmanngroup@%s.html" % tab, 'r') as file:
+                channel_html = file.read()
+            self.assertEqual("https://www.youtube.com/feeds/videos.xml?channel_id=UCl2mFZoRqjw_ELax4Yisf6w",
+                             get_channel_feed('https://www.youtube.com/@rossmanngroup/'+tab, html=channel_html))
+
+    def test_get_playlist_feed(self) -> None:
+        self.assertEqual(get_playlist_feed("https://www.youtube.com/watch?v=kYB8IZa5AuE&list=PLZHQObOWTQDPD3MizzM2xVFitgF8hE_ab&index=3"),
+                         "https://www.youtube.com/feeds/videos.xml?playlist_id=PLZHQObOWTQDPD3MizzM2xVFitgF8hE_ab")
+        self.assertEqual(get_playlist_feed("https://www.youtube.com/playlist?list=PLZHQObOWTQDPD3MizzM2xVFitgF8hE_ab"),
+                         "https://www.youtube.com/feeds/videos.xml?playlist_id=PLZHQObOWTQDPD3MizzM2xVFitgF8hE_ab")