_ Git - youtube-dl/blob - test/test_write_info_json.py

   1 #!/usr/bin/env python
   2 # coding: utf-8
   3
   4 import json
   5 import os
   6 import socket
   7 import sys
   8 import unittest
   9
  10 # Allow direct execution
  11 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  12
  13 import youtube_dl.FileDownloader
  14 import youtube_dl.InfoExtractors
  15 from youtube_dl.utils import *
  16
  17 PARAMETERS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "parameters.json")
  18
  19 # General configuration (from __init__, not very elegant...)
  20 jar = compat_cookiejar.CookieJar()
  21 cookie_processor = compat_urllib_request.HTTPCookieProcessor(jar)
  22 proxy_handler = compat_urllib_request.ProxyHandler()
  23 opener = compat_urllib_request.build_opener(proxy_handler, cookie_processor, YoutubeDLHandler())
  24 compat_urllib_request.install_opener(opener)
  25 socket.setdefaulttimeout(300) # 5 minutes should be enough (famous last words)
  26
  27 class FileDownloader(youtube_dl.FileDownloader):
  28     def __init__(self, *args, **kwargs):
  29         youtube_dl.FileDownloader.__init__(self, *args, **kwargs)
  30         self.to_stderr = self.to_screen
  31
  32 with io.open(PARAMETERS_FILE, encoding='utf-8') as pf:
  33     params = json.load(pf)
  34 params['writeinfojson'] = True
  35 params['skip_download'] = True
  36 params['writedescription'] = True
  37
  38 TEST_ID = 'BaW_jenozKc'
  39 INFO_JSON_FILE = TEST_ID + '.mp4.info.json'
  40 DESCRIPTION_FILE = TEST_ID + '.mp4.description'
  41 EXPECTED_DESCRIPTION = u'''test chars:  "'/\ä↭𝕐
  42
  43 This is a test video for youtube-dl.
  44
  45 For more information, contact phihag@phihag.de .'''
  46
  47 class TestInfoJSON(unittest.TestCase):
  48     def setUp(self):
  49         # Clear old files
  50         self.tearDown()
  51
  52     def test_info_json(self):
  53         ie = youtube_dl.InfoExtractors.YoutubeIE()
  54         fd = FileDownloader(params)
  55         fd.add_info_extractor(ie)
  56         fd.download([TEST_ID])
  57         self.assertTrue(os.path.exists(INFO_JSON_FILE))
  58         with io.open(INFO_JSON_FILE, 'r', encoding='utf-8') as jsonf:
  59             jd = json.load(jsonf)
  60         self.assertEqual(jd['upload_date'], u'20121002')
  61         self.assertEqual(jd['description'], EXPECTED_DESCRIPTION)
  62         self.assertEqual(jd['id'], TEST_ID)
  63         self.assertEqual(jd['extractor'], 'youtube')
  64         self.assertEqual(jd['title'], u'''youtube-dl test video "'/\ä↭𝕐''')
  65         self.assertEqual(jd['uploader'], 'Philipp Hagemeister')
  66
  67         self.assertTrue(os.path.exists(DESCRIPTION_FILE))
  68         with io.open(DESCRIPTION_FILE, 'r', encoding='utf-8') as descf:
  69             descr = descf.read()
  70         self.assertEqual(descr, EXPECTED_DESCRIPTION)
  71
  72     def tearDown(self):
  73         if os.path.exists(INFO_JSON_FILE):
  74             os.remove(INFO_JSON_FILE)
  75         if os.path.exists(DESCRIPTION_FILE):
  76             os.remove(DESCRIPTION_FILE)
  77
  78 if __name__ == '__main__':
  79     unittest.main()