]>
Commit | Line | Data |
---|---|---|
0cb20563 | 1 | #!/usr/bin/env python |
a0f59cdc | 2 | from __future__ import unicode_literals |
0cb20563 PH |
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 | ||
54330a1c | 11 | import errno |
0cb20563 PH |
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 | ||
e75c24e8 | 28 | def _make_testfunc(testfile): |
0cb20563 PH |
29 | m = re.match(r'^(.*)\.(as)$', testfile) |
30 | if not m: | |
e75c24e8 | 31 | return |
0cb20563 PH |
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') | |
8fb3ac36 PH |
37 | if ((not os.path.exists(swf_file)) or |
38 | os.path.getmtime(swf_file) < os.path.getmtime(as_file)): | |
0cb20563 PH |
39 | # Recompile |
40 | try: | |
39f0a2a6 PH |
41 | subprocess.check_call([ |
42 | 'mxmlc', '-output', swf_file, | |
43 | '-static-link-runtime-shared-libraries', as_file]) | |
0cb20563 PH |
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 | ||
e75c24e8 PH |
76 | for testfile in os.listdir(TEST_DIR): |
77 | _make_testfunc(testfile) | |
78 | ||
0cb20563 PH |
79 | if __name__ == '__main__': |
80 | unittest.main() |