feat: added testing of video id parsing
This commit is contained in:
33
tests/test_video_id.py
Normal file
33
tests/test_video_id.py
Normal file
@@ -0,0 +1,33 @@
|
||||
import unittest
|
||||
|
||||
from python_sponsorblock import _get_video_id, exceptions
|
||||
|
||||
|
||||
MAGIC_ID = "dQw4w9WgXcQ"
|
||||
|
||||
|
||||
class TestVideoIdParsing(unittest.TestCase):
|
||||
def test_already_parsed_id(self):
|
||||
self.assertEqual(_get_video_id(MAGIC_ID), MAGIC_ID)
|
||||
|
||||
def test_non_existing_id(self):
|
||||
with self.assertRaises(exceptions.SponsorBlockIdNotFoundError):
|
||||
_get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee")
|
||||
|
||||
self.assertEqual(_get_video_id("eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee", silent=True), None)
|
||||
|
||||
def test_youtu_dot_be(self):
|
||||
self.assertEqual(_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)
|
||||
|
||||
def test_invidious(self):
|
||||
self.assertEqual(_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)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user