]> Raphaƫl G. Git Repositories - youtubedl/blob - test/test_swfinterp.py
Prepare for upload.
[youtubedl] / test / test_swfinterp.py
1 #!/usr/bin/env python
2 from __future__ import unicode_literals
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
10
11 import errno
12 import io
13 import json
14 import re
15 import subprocess
16
17 from youtube_dl.swfinterp import SWFInterpreter
18
19
20 TEST_DIR = os.path.join(
21 os.path.dirname(os.path.abspath(__file__)), 'swftests')
22
23
24 class TestSWFInterpreter(unittest.TestCase):
25 pass
26
27
28 def _make_testfunc(testfile):
29 m = re.match(r'^(.*)\.(as)$', testfile)
30 if not m:
31 return
32 test_id = m.group(1)
33
34 def test_func(self):
35 as_file = os.path.join(TEST_DIR, testfile)
36 swf_file = os.path.join(TEST_DIR, test_id + '.swf')
37 if ((not os.path.exists(swf_file))
38 or os.path.getmtime(swf_file) < os.path.getmtime(as_file)):
39 # Recompile
40 try:
41 subprocess.check_call([
42 'mxmlc', '-output', swf_file,
43 '-static-link-runtime-shared-libraries', as_file])
44 except OSError as ose:
45 if ose.errno == errno.ENOENT:
46 print('mxmlc not found! Skipping test.')
47 return
48 raise
49
50 with open(swf_file, 'rb') as swf_f:
51 swf_content = swf_f.read()
52 swfi = SWFInterpreter(swf_content)
53
54 with io.open(as_file, 'r', encoding='utf-8') as as_f:
55 as_content = as_f.read()
56
57 def _find_spec(key):
58 m = re.search(
59 r'(?m)^//\s*%s:\s*(.*?)\n' % re.escape(key), as_content)
60 if not m:
61 raise ValueError('Cannot find %s in %s' % (key, testfile))
62 return json.loads(m.group(1))
63
64 input_args = _find_spec('input')
65 output = _find_spec('output')
66
67 swf_class = swfi.extract_class(test_id)
68 func = swfi.extract_function(swf_class, 'main')
69 res = func(input_args)
70 self.assertEqual(res, output)
71
72 test_func.__name__ = str('test_swf_' + test_id)
73 setattr(TestSWFInterpreter, test_func.__name__, test_func)
74
75
76 for testfile in os.listdir(TEST_DIR):
77 _make_testfunc(testfile)
78
79 if __name__ == '__main__':
80 unittest.main()