OLD | NEW |
1 # This Source Code Form is subject to the terms of the Mozilla Public | 1 # This Source Code Form is subject to the terms of the Mozilla Public |
2 # License, v. 2.0. If a copy of the MPL was not distributed with this | 2 # License, v. 2.0. If a copy of the MPL was not distributed with this |
3 # file, You can obtain one at http://mozilla.org/MPL/2.0/. | 3 # file, You can obtain one at http://mozilla.org/MPL/2.0/. |
4 | 4 |
5 import xml.etree.ElementTree as ET | 5 import xml.etree.ElementTree as ET |
6 | 6 |
7 import pytest | 7 import pytest |
8 | 8 |
9 from buildtools import packager, packagerEdge | 9 from buildtools import packager, packagerEdge |
10 | 10 |
11 | 11 |
12 @pytest.fixture | 12 @pytest.fixture |
13 def files(): | 13 def files(): |
14 """Minimal Files() for testing blockmap.""" | 14 """Minimal Files() for testing blockmap.""" |
15 str500 = b''.join(chr(i % 200 + 30) for i in range(500)) | 15 str500 = b''.join(chr(i % 200 + 30) for i in range(500)) |
16 files = packager.Files(set(), set()) | 16 files = packager.Files(set(), set()) |
17 files['Extension/foo.xml'] = str500 | 17 files['Extension/foo.xml'] = str500 |
18 files['Extension/bar.png'] = str500 * 200 | 18 files['Extension/bar.png'] = str500 * 200 |
19 return files | 19 return files |
20 | 20 |
21 | 21 |
22 def blockmap2dict(xml_data): | 22 def blockmap2dict(xml_data): |
23 """Convert AppxBlockMap.xml to a dict of dicts easier to inspect.""" | 23 """Convert AppxBlockMap.xml to a dict of dicts easier to inspect.""" |
24 return { | 24 return { |
25 file.get('Name'): { | 25 file.get('Name'): { |
26 'size': file.get('Size'), | 26 'size': file.get('Size'), |
27 'lfhsize': file.get('LfhSize'), | 27 'lfhsize': file.get('LfhSize'), |
28 'blocks': [b.get('Hash') for b in file] | 28 'blocks': [b.get('Hash') for b in file], |
29 } | 29 } |
30 for file in ET.fromstring(xml_data) | 30 for file in ET.fromstring(xml_data) |
31 } | 31 } |
32 | 32 |
33 | 33 |
34 def test_create_appx_blockmap(files): | 34 def test_create_appx_blockmap(files): |
35 blockmap = blockmap2dict(packagerEdge.create_appx_blockmap(files)) | 35 blockmap = blockmap2dict(packagerEdge.create_appx_blockmap(files)) |
36 assert blockmap['Extension\\foo.xml'] == { | 36 assert blockmap['Extension\\foo.xml'] == { |
37 'size': '500', | 37 'size': '500', |
38 'lfhsize': '47', | 38 'lfhsize': '47', |
39 'blocks': ['Vhwfmzss1Ney+j/ssR2QVISvFyMNBQeS2P+UjeE/di0='] | 39 'blocks': ['Vhwfmzss1Ney+j/ssR2QVISvFyMNBQeS2P+UjeE/di0='], |
40 } | 40 } |
41 assert blockmap['Extension\\bar.png'] == { | 41 assert blockmap['Extension\\bar.png'] == { |
42 'size': '100000', | 42 'size': '100000', |
43 'lfhsize': '47', | 43 'lfhsize': '47', |
44 'blocks': [ | 44 'blocks': [ |
45 'KPW2SxeEikUEGhoKmKxruUSexKun0bGXMppOqUFrX5E=', | 45 'KPW2SxeEikUEGhoKmKxruUSexKun0bGXMppOqUFrX5E=', |
46 'KQHnov1SZ1z34ttdDUjX2leYtpIIGndUVoUteieS2cw=', | 46 'KQHnov1SZ1z34ttdDUjX2leYtpIIGndUVoUteieS2cw=', |
47 ] | 47 ], |
48 } | 48 } |
49 | 49 |
50 | 50 |
51 def ctm2dict(content_types_map): | 51 def ctm2dict(content_types_map): |
52 """Convert content type map to a dict.""" | 52 """Convert content type map to a dict.""" |
53 ret = {'defaults': {}, 'overrides': {}} | 53 ret = {'defaults': {}, 'overrides': {}} |
54 for node in ET.fromstring(content_types_map): | 54 for node in ET.fromstring(content_types_map): |
55 ct = node.get('ContentType') | 55 ct = node.get('ContentType') |
56 if node.tag.endswith('Default'): | 56 if node.tag.endswith('Default'): |
57 ret['defaults'][node.get('Extension')] = ct | 57 ret['defaults'][node.get('Extension')] = ct |
(...skipping 14 matching lines...) Expand all Loading... |
72 filenames = ['no-extension', packagerEdge.MANIFEST, packagerEdge.BLOCKMAP] | 72 filenames = ['no-extension', packagerEdge.MANIFEST, packagerEdge.BLOCKMAP] |
73 filenames += ['file.' + x for x in 'json html js png css git otf'.split()] | 73 filenames += ['file.' + x for x in 'json html js png css git otf'.split()] |
74 ctm_dict = ctm2dict(packagerEdge.create_content_types_map(filenames)) | 74 ctm_dict = ctm2dict(packagerEdge.create_content_types_map(filenames)) |
75 assert ctm_dict['defaults'] == { | 75 assert ctm_dict['defaults'] == { |
76 'css': 'text/css', | 76 'css': 'text/css', |
77 'html': 'text/html', | 77 'html': 'text/html', |
78 'js': 'application/javascript', | 78 'js': 'application/javascript', |
79 'json': 'application/json', | 79 'json': 'application/json', |
80 'otf': 'application/octet-stream', | 80 'otf': 'application/octet-stream', |
81 'png': 'image/png', | 81 'png': 'image/png', |
82 'xml': 'application/xml' | 82 'xml': 'application/xml', |
83 } | 83 } |
84 assert ctm_dict['overrides'] == { | 84 assert ctm_dict['overrides'] == { |
85 '/AppxBlockMap.xml': 'application/vnd.ms-appx.blockmap+xml', | 85 '/AppxBlockMap.xml': 'application/vnd.ms-appx.blockmap+xml', |
86 '/AppxManifest.xml': 'application/vnd.ms-appx.manifest+xml' | 86 '/AppxManifest.xml': 'application/vnd.ms-appx.manifest+xml', |
87 } | 87 } |
OLD | NEW |