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 ConfigParser | |
6 import json | |
7 import os | |
8 import shutil | |
9 import xml.etree.ElementTree as ET | 5 import xml.etree.ElementTree as ET |
10 import zipfile | |
11 | 6 |
12 import pytest | 7 import pytest |
13 | 8 |
14 from buildtools import packager, packagerEdge | 9 from buildtools import packager, packagerEdge |
15 | 10 |
16 TEST_DIR = os.path.dirname(__file__) | |
17 TEST_METADATA = os.path.join(TEST_DIR, 'metadata.edge') | |
18 CHARS = b''.join(chr(i % 200 + 30) for i in range(500)) | |
19 MESSAGES_EN_US = json.dumps({ | |
20 'name': {'message': 'Adblock Plus'}, | |
21 'name_devbuild': {'message': 'devbuild-marker'}, | |
22 'description': { | |
23 'message': 'Adblock Plus is the most popular ad blocker ever, ' | |
24 'and also supports websites by not blocking ' | |
25 'unobstrusive ads by default (configurable).' | |
26 }, | |
27 }) | |
28 | |
29 | |
30 @pytest.fixture | |
31 def metadata(): | |
32 """Loaded metadata config.""" | |
33 conf_parser = ConfigParser.ConfigParser() | |
34 conf_parser.read(TEST_METADATA) | |
35 return conf_parser | |
36 | |
37 | 11 |
38 @pytest.fixture | 12 @pytest.fixture |
39 def files(): | 13 def files(): |
40 """Minimal Files() for testing manifest and blockmap.""" | 14 """Minimal Files() for testing blockmap.""" |
| 15 str500 = b''.join(chr(i % 200 + 30) for i in range(500)) |
41 files = packager.Files(set(), set()) | 16 files = packager.Files(set(), set()) |
42 for size in ['44', '50', '150']: | 17 files['Extension/foo.xml'] = str500 |
43 files['Assets/logo_{}.png'.format(size)] = CHARS | 18 files['Extension/bar.png'] = str500 * 200 |
44 files['Extension/_locales/en_US/messages.json'] = MESSAGES_EN_US | |
45 files['Extension/foo.xml'] = CHARS | |
46 files['Extension/bar.png'] = CHARS * 200 | |
47 return files | 19 return files |
48 | 20 |
49 | 21 |
50 @pytest.fixture | |
51 def srcdir(tmpdir): | |
52 """Source directory for building the package.""" | |
53 srcdir = tmpdir.mkdir('src') | |
54 shutil.copy(TEST_METADATA, str(srcdir.join('metadata.edge'))) | |
55 for size in ['44', '50', '150']: | |
56 path = srcdir.join('chrome', 'icons', 'abp-{}.png'.format(size)) | |
57 path.write(size, ensure=True) | |
58 localedir = srcdir.mkdir('_locales') | |
59 en_us_dir = localedir.mkdir('en_US') | |
60 en_us_dir.join('messages.json').write(MESSAGES_EN_US) | |
61 return srcdir | |
62 | |
63 | |
64 def blockmap2dict(xml_data): | 22 def blockmap2dict(xml_data): |
65 """Convert AppxBlockMap.xml to a dict of dicts easier to inspect.""" | 23 """Convert AppxBlockMap.xml to a dict of dicts easier to inspect.""" |
66 return { | 24 return { |
67 file.get('Name'): { | 25 file.get('Name'): { |
68 'size': file.get('Size'), | 26 'size': file.get('Size'), |
69 'lfhsize': file.get('LfhSize'), | 27 'lfhsize': file.get('LfhSize'), |
70 'blocks': [b.get('Hash') for b in file] | 28 'blocks': [b.get('Hash') for b in file] |
71 } | 29 } |
72 for file in ET.fromstring(xml_data) | 30 for file in ET.fromstring(xml_data) |
73 } | 31 } |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
120 'js': 'application/javascript', | 78 'js': 'application/javascript', |
121 'json': 'application/json', | 79 'json': 'application/json', |
122 'otf': 'application/octet-stream', | 80 'otf': 'application/octet-stream', |
123 'png': 'image/png', | 81 'png': 'image/png', |
124 'xml': 'application/xml' | 82 'xml': 'application/xml' |
125 } | 83 } |
126 assert ctm_dict['overrides'] == { | 84 assert ctm_dict['overrides'] == { |
127 '/AppxBlockMap.xml': 'application/vnd.ms-appx.blockmap+xml', | 85 '/AppxBlockMap.xml': 'application/vnd.ms-appx.blockmap+xml', |
128 '/AppxManifest.xml': 'application/vnd.ms-appx.manifest+xml' | 86 '/AppxManifest.xml': 'application/vnd.ms-appx.manifest+xml' |
129 } | 87 } |
130 | |
131 | |
132 def test_create_appx_manifest(metadata, files): | |
133 namespaces = { | |
134 'ns': 'http://schemas.microsoft.com/' | |
135 'appx/manifest/foundation/windows10', | |
136 'uap': 'http://schemas.microsoft.com/appx/manifest/uap/windows10', | |
137 'uap3': 'http://schemas.microsoft.com/appx/manifest/uap/windows10/3', | |
138 } | |
139 | |
140 def first(elem): | |
141 return elem[0] | |
142 | |
143 def text(elem): | |
144 return elem.text | |
145 | |
146 def attr(attr): | |
147 def wrapper(elem): | |
148 return elem.attrib[attr] | |
149 return wrapper | |
150 | |
151 base = [ | |
152 ('.//*', [len], 21.0), | |
153 ('./ns:Identity', [first, attr('Publisher')], | |
154 'CN=4F066043-8AFE-41C9-B762-6C15E77E3F88'), | |
155 ('./ns:Properties/ns:PublisherDisplayName', [first, text], | |
156 'Eyeo GmbH'), | |
157 ('./ns:Properties/ns:Logo', [first, text], 'Assets\\logo_50.png'), | |
158 ('./ns:Dependencies/ns:TargetDeviceFamily', | |
159 [first, attr('MinVersion')], | |
160 '10.0.14332.0'), | |
161 ('./ns:Dependencies/ns:TargetDeviceFamily', | |
162 [first, attr('MaxVersionTested')], | |
163 '12.0.0.0'), | |
164 ('./ns:Applications/ns:Application/uap:VisualElements', | |
165 [first, attr('Square150x150Logo')], | |
166 'Assets\\logo_150.png'), | |
167 ('./ns:Applications/ns:Application/uap:VisualElements', | |
168 [first, attr('Square44x44Logo')], | |
169 'Assets\\logo_44.png'), | |
170 ('./ns:Applications/ns:Application/uap:VisualElements', | |
171 [first, attr('Description')], | |
172 'Adblock Plus is the most popular ad blocker ever, and also ' | |
173 'supports websites by not blocking unobstrusive ads by ' | |
174 'default (configurable).'), | |
175 ('./ns:Applications/ns:Application/uap:VisualElements', | |
176 [first, attr('BackgroundColor')], | |
177 'red'), | |
178 ] | |
179 | |
180 devbuild = base + [ | |
181 ('./ns:Identity', [first, attr('Name')], | |
182 'EyeoGmbH.AdblockPlusdevelopmentbuild'), | |
183 ('./ns:Identity', [first, attr('Version')], '1.2.1000.0'), | |
184 ('./ns:Properties/ns:DisplayName', [first, text], 'devbuild-marker'), | |
185 ('./ns:Applications/ns:Application/uap:VisualElements', | |
186 [first, attr('DisplayName')], | |
187 'devbuild-marker'), | |
188 ('./ns:Applications/ns:Application/ns:Extensions/uap3:Extension/' | |
189 'uap3:AppExtension', | |
190 [first, attr('Id')], | |
191 'EdgeExtension'), | |
192 ('./ns:Applications/ns:Application/ns:Extensions/uap3:Extension/' | |
193 'uap3:AppExtension', | |
194 [first, attr('DisplayName')], | |
195 'devbuild-marker'), | |
196 ] | |
197 | |
198 release = base + [ | |
199 ('./ns:Identity', [first, attr('Name')], 'EyeoGmbH.AdblockPlus'), | |
200 ('./ns:Identity', [first, attr('Version')], '1.2.3.0'), | |
201 ('./ns:Properties/ns:DisplayName', [first, text], 'Adblock Plus'), | |
202 ('./ns:Applications/ns:Application/uap:VisualElements', | |
203 [first, attr('DisplayName')], | |
204 'Adblock Plus'), | |
205 ('./ns:Applications/ns:Application/ns:Extensions/uap3:Extension/' | |
206 'uap3:AppExtension', | |
207 [first, attr('Id')], | |
208 '1.0'), | |
209 ('./ns:Applications/ns:Application/ns:Extensions/uap3:Extension/' | |
210 'uap3:AppExtension', | |
211 [first, attr('DisplayName')], | |
212 'Adblock Plus'), | |
213 ] | |
214 | |
215 for args, pairs in [(('1000', False), devbuild), ((None, True), release)]: | |
216 manifest = ET.fromstring(packagerEdge.create_appx_manifest( | |
217 {'metadata': metadata}, files, *args | |
218 )) | |
219 for expression, modifiers, value in pairs: | |
220 res = reduce( | |
221 lambda val, func: func(val), | |
222 modifiers, | |
223 manifest.findall(expression, namespaces=namespaces)) | |
224 assert res == value | |
225 | |
226 | |
227 def test_move_files_to_extension(): | |
228 files = packager.Files(set(), set()) | |
229 files['foo.xml'] = CHARS | |
230 files['foo/bar.xml'] = CHARS | |
231 files['Extension/foo.xml'] = CHARS | |
232 packagerEdge.move_files_to_extension(files) | |
233 assert set(files.keys()) == { | |
234 'Extension/foo.xml', | |
235 'Extension/foo/bar.xml', | |
236 'Extension/Extension/foo.xml' | |
237 } | |
238 | |
239 | |
240 def test_create_build(tmpdir, srcdir): | |
241 out_file = str(tmpdir.join('abp.appx')) | |
242 packagerEdge.createBuild(str(srcdir), outFile=out_file, releaseBuild=True) | |
243 appx = zipfile.ZipFile(out_file) | |
244 | |
245 names = set(appx.namelist()) | |
246 assert 'AppxManifest.xml' in names | |
247 assert 'AppxBlockMap.xml' in names | |
248 assert '[Content_Types].xml' in names | |
249 | |
250 assert 'devbuild-marker' not in appx.read('AppxManifest.xml') | |
251 assert appx.read('Assets/logo_44.png') == '44' | |
252 assert appx.read('Extension/icons/abp-44.png') == '44' | |
253 | |
254 | |
255 def test_create_devbuild(tmpdir, srcdir): | |
256 out_file = str(tmpdir.join('abp.appx')) | |
257 packagerEdge.createBuild(str(srcdir), outFile=out_file, releaseBuild=False) | |
258 appx = zipfile.ZipFile(out_file) | |
259 assert 'devbuild-marker' in appx.read('AppxManifest.xml') | |
OLD | NEW |