Add tests for justin.tv and twitch.tv
[youtube-dl] / test / test_justin_tv.py
1 #!/usr/bin/env python
2
3 import sys
4 import unittest
5
6 # Allow direct execution
7 import os
8 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
9
10 from youtube_dl.InfoExtractors import JustinTVIE
11
12 class TestJustinTVMatching(unittest.TestCase):
13     def test_justin_tv_channelid_matching(self):
14         self.assertTrue(JustinTVIE.suitable(u"justin.tv/vanillatv"))
15         self.assertTrue(JustinTVIE.suitable(u"twitch.tv/vanillatv"))
16         self.assertTrue(JustinTVIE.suitable(u"www.justin.tv/vanillatv"))
17         self.assertTrue(JustinTVIE.suitable(u"www.twitch.tv/vanillatv"))
18         self.assertTrue(JustinTVIE.suitable(u"http://www.justin.tv/vanillatv"))
19         self.assertTrue(JustinTVIE.suitable(u"http://www.twitch.tv/vanillatv"))
20         self.assertTrue(JustinTVIE.suitable(u"http://www.justin.tv/vanillatv/"))
21         self.assertTrue(JustinTVIE.suitable(u"http://www.twitch.tv/vanillatv/"))
22
23     def test_justintv_videoid_matching(self):
24         self.assertTrue(JustinTVIE.suitable(u"http://www.twitch.tv/vanillatv/b/328087483"))
25
26     def test_justin_tv_chapterid_matching(self):
27         self.assertTrue(JustinTVIE.suitable(u"http://www.twitch.tv/tsm_theoddone/c/2349361"))
28
29 if __name__ == '__main__':
30     unittest.main()