X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=test%2Ftest_write_info_json.py;h=8134dda371c0e31f57a9fc199452e33808a5702b;hb=45014296be276399f6d4b761e6ffec3a6c6b292b;hp=e59fee91c8e12b19b5a499c9a500f449667ad4dc;hpb=3cc687d486c5fc4710b6fdec4340b7635cec91c9;p=youtube-dl diff --git a/test/test_write_info_json.py b/test/test_write_info_json.py index e59fee91c..8134dda37 100644 --- a/test/test_write_info_json.py +++ b/test/test_write_info_json.py @@ -1,7 +1,8 @@ #!/usr/bin/env python +# coding: utf-8 +import json import os -import socket import sys import unittest @@ -20,7 +21,6 @@ cookie_processor = compat_urllib_request.HTTPCookieProcessor(jar) proxy_handler = compat_urllib_request.ProxyHandler() opener = compat_urllib_request.build_opener(proxy_handler, cookie_processor, YoutubeDLHandler()) compat_urllib_request.install_opener(opener) -socket.setdefaulttimeout(300) # 5 minutes should be enough (famous last words) class FileDownloader(youtube_dl.FileDownloader): def __init__(self, *args, **kwargs): @@ -31,9 +31,16 @@ with io.open(PARAMETERS_FILE, encoding='utf-8') as pf: params = json.load(pf) params['writeinfojson'] = True params['skip_download'] = True +params['writedescription'] = True TEST_ID = 'BaW_jenozKc' INFO_JSON_FILE = TEST_ID + '.mp4.info.json' +DESCRIPTION_FILE = TEST_ID + '.mp4.description' +EXPECTED_DESCRIPTION = u'''test chars: "'/\ä↭𝕐 + +This is a test video for youtube-dl. + +For more information, contact phihag@phihag.de .''' class TestInfoJSON(unittest.TestCase): def setUp(self): @@ -46,10 +53,25 @@ class TestInfoJSON(unittest.TestCase): fd.add_info_extractor(ie) fd.download([TEST_ID]) self.assertTrue(os.path.exists(INFO_JSON_FILE)) + with io.open(INFO_JSON_FILE, 'r', encoding='utf-8') as jsonf: + jd = json.load(jsonf) + self.assertEqual(jd['upload_date'], u'20121002') + self.assertEqual(jd['description'], EXPECTED_DESCRIPTION) + self.assertEqual(jd['id'], TEST_ID) + self.assertEqual(jd['extractor'], 'youtube') + self.assertEqual(jd['title'], u'''youtube-dl test video "'/\ä↭𝕐''') + self.assertEqual(jd['uploader'], 'Philipp Hagemeister') + + self.assertTrue(os.path.exists(DESCRIPTION_FILE)) + with io.open(DESCRIPTION_FILE, 'r', encoding='utf-8') as descf: + descr = descf.read() + self.assertEqual(descr, EXPECTED_DESCRIPTION) def tearDown(self): if os.path.exists(INFO_JSON_FILE): os.remove(INFO_JSON_FILE) + if os.path.exists(DESCRIPTION_FILE): + os.remove(DESCRIPTION_FILE) if __name__ == '__main__': unittest.main()