[yt-dlp.git] / test / test_subtitles.py

#!/usr/bin/env python
from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from test.helper import FakeYDL, md5


from youtube_dl.extractor import (
    BlipTVIE,
    YoutubeIE,
    DailymotionIE,
    TEDIE,
    VimeoIE,
    WallaIE,
)


class BaseTestSubtitles(unittest.TestCase):
    url = None
    IE = None
    def setUp(self):
        self.DL = FakeYDL()
        self.ie = self.IE(self.DL)

    def getInfoDict(self):
        info_dict = self.ie.extract(self.url)
        return info_dict

    def getSubtitles(self):
        info_dict = self.getInfoDict()
        return info_dict['subtitles']


class TestYoutubeSubtitles(BaseTestSubtitles):
    url = 'QRS8MkLhQmM'
    IE = YoutubeIE

    def test_youtube_no_writesubtitles(self):
        self.DL.params['writesubtitles'] = False
        subtitles = self.getSubtitles()
        self.assertEqual(subtitles, None)

    def test_youtube_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '4cd9278a35ba2305f47354ee13472260')

    def test_youtube_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['it']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['it']), '164a51f16f260476a05b50fe4c2f161d')

    def test_youtube_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles.keys()), 13)

    def test_youtube_subtitles_sbv_format(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitlesformat'] = 'sbv'
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '13aeaa0c245a8bed9a451cb643e3ad8b')

    def test_youtube_subtitles_vtt_format(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitlesformat'] = 'vtt'
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '3cb210999d3e021bd6c7f0ea751eab06')

    def test_youtube_list_subtitles(self):
        self.DL.expect_warning('Video doesn\'t have automatic captions')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_youtube_automatic_captions(self):
        self.url = '8YoUxe5ncPo'
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslangs'] = ['it']
        subtitles = self.getSubtitles()
        self.assertTrue(subtitles['it'] is not None)

    def test_youtube_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'n5BB19UTcdA'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_youtube_multiple_langs(self):
        self.url = 'QRS8MkLhQmM'
        self.DL.params['writesubtitles'] = True
        langs = ['it', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestDailymotionSubtitles(BaseTestSubtitles):
    url = 'http://www.dailymotion.com/video/xczg00'
    IE = DailymotionIE

    def test_no_writesubtitles(self):
        subtitles = self.getSubtitles()
        self.assertEqual(subtitles, None)

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '976553874490cba125086bbfea3ff76f')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), '594564ec7d588942e384e920e5341792')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles.keys()), 5)

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_automatic_captions(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslang'] = ['en']
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) == 0)

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://www.dailymotion.com/video/x12u166_le-zapping-tele-star-du-08-aout-2013_tv'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestTedSubtitles(BaseTestSubtitles):
    url = 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
    IE = TEDIE

    def test_no_writesubtitles(self):
        subtitles = self.getSubtitles()
        self.assertEqual(subtitles, None)

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '4262c1665ff928a2dada178f62cb8d14')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), '66a63f7f42c97a50f8c0e90bc7797bb5')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) >= 28)

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_automatic_captions(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslang'] = ['en']
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) == 0)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestBlipTVSubtitles(BaseTestSubtitles):
    url = 'http://blip.tv/a/a-6603250'
    IE = BlipTVIE

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_allsubtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['en']))
        self.assertEqual(md5(subtitles['en']), '5b75c300af65fe4476dff79478bb93e4')


class TestVimeoSubtitles(BaseTestSubtitles):
    url = 'http://vimeo.com/76979871'
    IE = VimeoIE

    def test_no_writesubtitles(self):
        subtitles = self.getSubtitles()
        self.assertEqual(subtitles, None)

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '8062383cf4dec168fc40a088aa6d5888')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), 'b6191146a6c5d3a452244d853fde6dc8')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['de', 'en', 'es', 'fr']))

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_automatic_captions(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslang'] = ['en']
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) == 0)

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://vimeo.com/56015672'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestWallaSubtitles(BaseTestSubtitles):
    url = 'http://vod.walla.co.il/movie/2705958/the-yes-men'
    IE = WallaIE

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_allsubtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['heb']))
        self.assertEqual(md5(subtitles['heb']), 'e758c5d7cb982f6bef14f377ec7a3920')

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://vod.walla.co.il/movie/2642630/one-direction-all-for-one'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)


if __name__ == '__main__':
    unittest.main()
Commit	Line	Data
ae84f879	1	#!/usr/bin/env python
a0f59cdc	2	from __future__ import unicode_literals
ae84f879 JMF	3
	4	# Allow direct execution
	5	import os
	6	import sys
	7	import unittest
	8	sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
	9
d0efb9ec	10	from test.helper import FakeYDL, md5
ae84f879 JMF	11
	12
	13	from youtube_dl.extractor import (
b4bcffef	14	BlipTVIE,
ae84f879 JMF	15	YoutubeIE,
	16	DailymotionIE,
	17	TEDIE,
4edff4cf	18	VimeoIE,
7bc8780c	19	WallaIE,
ae84f879 JMF	20	)
	21
	22
	23	class BaseTestSubtitles(unittest.TestCase):
	24	url = None
	25	IE = None
	26	def setUp(self):
	27	self.DL = FakeYDL()
	28	self.ie = self.IE(self.DL)
	29
	30	def getInfoDict(self):
	31	info_dict = self.ie.extract(self.url)
	32	return info_dict
	33
	34	def getSubtitles(self):
	35	info_dict = self.getInfoDict()
	36	return info_dict['subtitles']
	37
	38
	39	class TestYoutubeSubtitles(BaseTestSubtitles):
	40	url = 'QRS8MkLhQmM'
	41	IE = YoutubeIE
	42
ae84f879 JMF	43	def test_youtube_no_writesubtitles(self):
	44	self.DL.params['writesubtitles'] = False
	45	subtitles = self.getSubtitles()
	46	self.assertEqual(subtitles, None)
	47
	48	def test_youtube_subtitles(self):
	49	self.DL.params['writesubtitles'] = True
	50	subtitles = self.getSubtitles()
	51	self.assertEqual(md5(subtitles['en']), '4cd9278a35ba2305f47354ee13472260')
	52
	53	def test_youtube_subtitles_lang(self):
	54	self.DL.params['writesubtitles'] = True
	55	self.DL.params['subtitleslangs'] = ['it']
	56	subtitles = self.getSubtitles()
	57	self.assertEqual(md5(subtitles['it']), '164a51f16f260476a05b50fe4c2f161d')
	58
	59	def test_youtube_allsubtitles(self):
	60	self.DL.params['writesubtitles'] = True
	61	self.DL.params['allsubtitles'] = True
	62	subtitles = self.getSubtitles()
	63	self.assertEqual(len(subtitles.keys()), 13)
	64
	65	def test_youtube_subtitles_sbv_format(self):
	66	self.DL.params['writesubtitles'] = True
	67	self.DL.params['subtitlesformat'] = 'sbv'
	68	subtitles = self.getSubtitles()
	69	self.assertEqual(md5(subtitles['en']), '13aeaa0c245a8bed9a451cb643e3ad8b')
	70
	71	def test_youtube_subtitles_vtt_format(self):
	72	self.DL.params['writesubtitles'] = True
	73	self.DL.params['subtitlesformat'] = 'vtt'
	74	subtitles = self.getSubtitles()
4e0084d9	75	self.assertEqual(md5(subtitles['en']), '3cb210999d3e021bd6c7f0ea751eab06')
ae84f879 JMF	76
ae84f879 JMF	77	def test_youtube_list_subtitles(self):
a0f59cdc	78	self.DL.expect_warning('Video doesn\'t have automatic captions')
ae84f879 JMF	79	self.DL.params['listsubtitles'] = True
	80	info_dict = self.getInfoDict()
	81	self.assertEqual(info_dict, None)
	82
	83	def test_youtube_automatic_captions(self):
	84	self.url = '8YoUxe5ncPo'
	85	self.DL.params['writeautomaticsub'] = True
	86	self.DL.params['subtitleslangs'] = ['it']
	87	subtitles = self.getSubtitles()
	88	self.assertTrue(subtitles['it'] is not None)
	89
	90	def test_youtube_nosubtitles(self):
a0f59cdc	91	self.DL.expect_warning('video doesn\'t have subtitles')
345e3783	92	self.url = 'n5BB19UTcdA'
ae84f879 JMF	93	self.DL.params['writesubtitles'] = True
	94	self.DL.params['allsubtitles'] = True
	95	subtitles = self.getSubtitles()
	96	self.assertEqual(len(subtitles), 0)
	97
	98	def test_youtube_multiple_langs(self):
	99	self.url = 'QRS8MkLhQmM'
	100	self.DL.params['writesubtitles'] = True
	101	langs = ['it', 'fr', 'de']
	102	self.DL.params['subtitleslangs'] = langs
	103	subtitles = self.getSubtitles()
	104	for lang in langs:
a0f59cdc	105	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879 JMF	106
	107
	108	class TestDailymotionSubtitles(BaseTestSubtitles):
	109	url = 'http://www.dailymotion.com/video/xczg00'
	110	IE = DailymotionIE
	111
	112	def test_no_writesubtitles(self):
	113	subtitles = self.getSubtitles()
	114	self.assertEqual(subtitles, None)
	115
	116	def test_subtitles(self):
	117	self.DL.params['writesubtitles'] = True
	118	subtitles = self.getSubtitles()
	119	self.assertEqual(md5(subtitles['en']), '976553874490cba125086bbfea3ff76f')
	120
	121	def test_subtitles_lang(self):
	122	self.DL.params['writesubtitles'] = True
	123	self.DL.params['subtitleslangs'] = ['fr']
	124	subtitles = self.getSubtitles()
	125	self.assertEqual(md5(subtitles['fr']), '594564ec7d588942e384e920e5341792')
	126
	127	def test_allsubtitles(self):
	128	self.DL.params['writesubtitles'] = True
	129	self.DL.params['allsubtitles'] = True
	130	subtitles = self.getSubtitles()
	131	self.assertEqual(len(subtitles.keys()), 5)
	132
	133	def test_list_subtitles(self):
a0f59cdc	134	self.DL.expect_warning('Automatic Captions not supported by this server')
ae84f879 JMF	135	self.DL.params['listsubtitles'] = True
	136	info_dict = self.getInfoDict()
	137	self.assertEqual(info_dict, None)
	138
	139	def test_automatic_captions(self):
a0f59cdc	140	self.DL.expect_warning('Automatic Captions not supported by this server')
ae84f879 JMF	141	self.DL.params['writeautomaticsub'] = True
	142	self.DL.params['subtitleslang'] = ['en']
	143	subtitles = self.getSubtitles()
	144	self.assertTrue(len(subtitles.keys()) == 0)
	145
	146	def test_nosubtitles(self):
a0f59cdc	147	self.DL.expect_warning('video doesn\'t have subtitles')
ae84f879 JMF	148	self.url = 'http://www.dailymotion.com/video/x12u166_le-zapping-tele-star-du-08-aout-2013_tv'
	149	self.DL.params['writesubtitles'] = True
	150	self.DL.params['allsubtitles'] = True
	151	subtitles = self.getSubtitles()
	152	self.assertEqual(len(subtitles), 0)
	153
	154	def test_multiple_langs(self):
	155	self.DL.params['writesubtitles'] = True
	156	langs = ['es', 'fr', 'de']
	157	self.DL.params['subtitleslangs'] = langs
	158	subtitles = self.getSubtitles()
	159	for lang in langs:
a0f59cdc	160	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879 JMF	161
	162
	163	class TestTedSubtitles(BaseTestSubtitles):
	164	url = 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
	165	IE = TEDIE
	166
	167	def test_no_writesubtitles(self):
	168	subtitles = self.getSubtitles()
	169	self.assertEqual(subtitles, None)
	170
	171	def test_subtitles(self):
	172	self.DL.params['writesubtitles'] = True
	173	subtitles = self.getSubtitles()
26dca166	174	self.assertEqual(md5(subtitles['en']), '4262c1665ff928a2dada178f62cb8d14')
ae84f879 JMF	175
	176	def test_subtitles_lang(self):
	177	self.DL.params['writesubtitles'] = True
	178	self.DL.params['subtitleslangs'] = ['fr']
	179	subtitles = self.getSubtitles()
26dca166	180	self.assertEqual(md5(subtitles['fr']), '66a63f7f42c97a50f8c0e90bc7797bb5')
ae84f879 JMF	181
	182	def test_allsubtitles(self):
	183	self.DL.params['writesubtitles'] = True
	184	self.DL.params['allsubtitles'] = True
	185	subtitles = self.getSubtitles()
0321213c	186	self.assertTrue(len(subtitles.keys()) >= 28)
ae84f879 JMF	187
ae84f879 JMF	188	def test_list_subtitles(self):
a0f59cdc	189	self.DL.expect_warning('Automatic Captions not supported by this server')
ae84f879 JMF	190	self.DL.params['listsubtitles'] = True
	191	info_dict = self.getInfoDict()
	192	self.assertEqual(info_dict, None)
	193
	194	def test_automatic_captions(self):
a0f59cdc	195	self.DL.expect_warning('Automatic Captions not supported by this server')
ae84f879 JMF	196	self.DL.params['writeautomaticsub'] = True
	197	self.DL.params['subtitleslang'] = ['en']
	198	subtitles = self.getSubtitles()
	199	self.assertTrue(len(subtitles.keys()) == 0)
	200
	201	def test_multiple_langs(self):
	202	self.DL.params['writesubtitles'] = True
	203	langs = ['es', 'fr', 'de']
	204	self.DL.params['subtitleslangs'] = langs
	205	subtitles = self.getSubtitles()
	206	for lang in langs:
a0f59cdc	207	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879	208
b4bcffef PH	209
	210	class TestBlipTVSubtitles(BaseTestSubtitles):
	211	url = 'http://blip.tv/a/a-6603250'
	212	IE = BlipTVIE
	213
	214	def test_list_subtitles(self):
a0f59cdc	215	self.DL.expect_warning('Automatic Captions not supported by this server')
b4bcffef PH	216	self.DL.params['listsubtitles'] = True
	217	info_dict = self.getInfoDict()
	218	self.assertEqual(info_dict, None)
	219
	220	def test_allsubtitles(self):
a0f59cdc	221	self.DL.expect_warning('Automatic Captions not supported by this server')
b4bcffef PH	222	self.DL.params['writesubtitles'] = True
	223	self.DL.params['allsubtitles'] = True
	224	subtitles = self.getSubtitles()
	225	self.assertEqual(set(subtitles.keys()), set(['en']))
	226	self.assertEqual(md5(subtitles['en']), '5b75c300af65fe4476dff79478bb93e4')
	227
	228
4edff4cf S	229	class TestVimeoSubtitles(BaseTestSubtitles):
	230	url = 'http://vimeo.com/76979871'
	231	IE = VimeoIE
b0268cb6	232
4edff4cf S	233	def test_no_writesubtitles(self):
	234	subtitles = self.getSubtitles()
	235	self.assertEqual(subtitles, None)
	236
	237	def test_subtitles(self):
	238	self.DL.params['writesubtitles'] = True
	239	subtitles = self.getSubtitles()
	240	self.assertEqual(md5(subtitles['en']), '8062383cf4dec168fc40a088aa6d5888')
	241
	242	def test_subtitles_lang(self):
	243	self.DL.params['writesubtitles'] = True
	244	self.DL.params['subtitleslangs'] = ['fr']
	245	subtitles = self.getSubtitles()
	246	self.assertEqual(md5(subtitles['fr']), 'b6191146a6c5d3a452244d853fde6dc8')
	247
	248	def test_allsubtitles(self):
	249	self.DL.params['writesubtitles'] = True
	250	self.DL.params['allsubtitles'] = True
	251	subtitles = self.getSubtitles()
	252	self.assertEqual(set(subtitles.keys()), set(['de', 'en', 'es', 'fr']))
	253
	254	def test_list_subtitles(self):
a0f59cdc	255	self.DL.expect_warning('Automatic Captions not supported by this server')
4edff4cf S	256	self.DL.params['listsubtitles'] = True
	257	info_dict = self.getInfoDict()
	258	self.assertEqual(info_dict, None)
	259
	260	def test_automatic_captions(self):
a0f59cdc	261	self.DL.expect_warning('Automatic Captions not supported by this server')
4edff4cf S	262	self.DL.params['writeautomaticsub'] = True
	263	self.DL.params['subtitleslang'] = ['en']
	264	subtitles = self.getSubtitles()
	265	self.assertTrue(len(subtitles.keys()) == 0)
	266
	267	def test_nosubtitles(self):
a0f59cdc	268	self.DL.expect_warning('video doesn\'t have subtitles')
4edff4cf S	269	self.url = 'http://vimeo.com/56015672'
	270	self.DL.params['writesubtitles'] = True
	271	self.DL.params['allsubtitles'] = True
	272	subtitles = self.getSubtitles()
	273	self.assertEqual(len(subtitles), 0)
	274
	275	def test_multiple_langs(self):
	276	self.DL.params['writesubtitles'] = True
	277	langs = ['es', 'fr', 'de']
	278	self.DL.params['subtitleslangs'] = langs
	279	subtitles = self.getSubtitles()
	280	for lang in langs:
a0f59cdc	281	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
4edff4cf S	282
4edff4cf S	283
0ca41c3d	284	class TestWallaSubtitles(BaseTestSubtitles):
7bc8780c S	285	url = 'http://vod.walla.co.il/movie/2705958/the-yes-men'
	286	IE = WallaIE
	287
	288	def test_list_subtitles(self):
a0f59cdc	289	self.DL.expect_warning('Automatic Captions not supported by this server')
7bc8780c S	290	self.DL.params['listsubtitles'] = True
	291	info_dict = self.getInfoDict()
	292	self.assertEqual(info_dict, None)
	293
	294	def test_allsubtitles(self):
a0f59cdc	295	self.DL.expect_warning('Automatic Captions not supported by this server')
7bc8780c S	296	self.DL.params['writesubtitles'] = True
	297	self.DL.params['allsubtitles'] = True
	298	subtitles = self.getSubtitles()
	299	self.assertEqual(set(subtitles.keys()), set(['heb']))
	300	self.assertEqual(md5(subtitles['heb']), 'e758c5d7cb982f6bef14f377ec7a3920')
	301
	302	def test_nosubtitles(self):
a0f59cdc	303	self.DL.expect_warning('video doesn\'t have subtitles')
7bc8780c S	304	self.url = 'http://vod.walla.co.il/movie/2642630/one-direction-all-for-one'
	305	self.DL.params['writesubtitles'] = True
	306	self.DL.params['allsubtitles'] = True
	307	subtitles = self.getSubtitles()
	308	self.assertEqual(len(subtitles), 0)
	309
	310
ae84f879 JMF	311	if __name__ == '__main__':
ae84f879 JMF	312	unittest.main()