You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.6 KiB
77 lines
2.6 KiB
#!/usr/bin/env python |
|
# coding: utf-8 |
|
|
|
import json |
|
import os |
|
import sys |
|
import unittest |
|
|
|
# Allow direct execution |
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
|
import youtube_dl.FileDownloader |
|
import youtube_dl.InfoExtractors |
|
from youtube_dl.utils import * |
|
|
|
PARAMETERS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "parameters.json") |
|
|
|
# General configuration (from __init__, not very elegant...) |
|
jar = compat_cookiejar.CookieJar() |
|
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) |
|
|
|
class FileDownloader(youtube_dl.FileDownloader): |
|
def __init__(self, *args, **kwargs): |
|
youtube_dl.FileDownloader.__init__(self, *args, **kwargs) |
|
self.to_stderr = self.to_screen |
|
|
|
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): |
|
# Clear old files |
|
self.tearDown() |
|
|
|
def test_info_json(self): |
|
ie = youtube_dl.InfoExtractors.YoutubeIE() |
|
fd = FileDownloader(params) |
|
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()
|
|
|