music-kraken/python-sponsorblock
3
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: moved id parsing in class

Browse Source
This commit is contained in:
Hazel
2024-04-24 12:35:22 +02:00
parent 0f01238365
commit e28ae87bbe
2 changed files with 41 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/test_video_id.py
View File

@@ -1,6 +1,6 @@
import unittest
from python_sponsorblock import _get_video_id, exceptions
from python_sponsorblock import SponsorBlock, exceptions
MAGIC_ID = "dQw4w9WgXcQ"
@@ -8,25 +8,32 @@ MAGIC_ID = "dQw4w9WgXcQ"
class TestVideoIdParsing(unittest.TestCase):
def test_already_parsed_id(self):
self.assertEqual(_get_video_id(MAGIC_ID), MAGIC_ID)
s = SponsorBlock()
self.assertEqual(s._get_video_id(MAGIC_ID), MAGIC_ID)
def test_non_existing_id(self):
s = SponsorBlock()
with self.assertRaises(exceptions.SponsorBlockIdNotFoundError):
_get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee")
s._get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee")
self.assertEqual(_get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee", silent=True), None)
s = SponsorBlock(silent=True)
self.assertEqual(s._get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"), None)
def test_youtu_dot_be(self):
self.assertEqual(_get_video_id("https://youtu.be/" + MAGIC_ID), MAGIC_ID)
s = SponsorBlock()
self.assertEqual(s._get_video_id("https://youtu.be/" + MAGIC_ID), MAGIC_ID)
def test_youtube_dot_com(self):
self.assertEqual(_get_video_id("https://www.youtube.com/watch?v=" + MAGIC_ID), MAGIC_ID)
s = SponsorBlock()
self.assertEqual(s._get_video_id("https://www.youtube.com/watch?v=" + MAGIC_ID), MAGIC_ID)
def test_invidious(self):
self.assertEqual(_get_video_id("https://invidio.us/watch?v=" + MAGIC_ID), MAGIC_ID)
s = SponsorBlock()
self.assertEqual(s._get_video_id("https://invidio.us/watch?v=" + MAGIC_ID), MAGIC_ID)
def test_piped(self):
self.assertEqual(_get_video_id("https://piped.video/watch?v=" + MAGIC_ID), MAGIC_ID)
s = SponsorBlock()
self.assertEqual(s._get_video_id("https://piped.video/watch?v=" + MAGIC_ID), MAGIC_ID)
if __name__ == "__main__":