diff options
Diffstat (limited to 'tests/extract_sub_info.py')
| -rw-r--r-- | tests/extract_sub_info.py | 44 |
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") |