| OLD | NEW |
| 1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
| 2 # Copyright (C) 2006-2016 Eyeo GmbH | 2 # Copyright (C) 2006-2016 Eyeo GmbH |
| 3 # | 3 # |
| 4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
| 5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
| 6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
| 7 # | 7 # |
| 8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
| 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
| 12 # | 12 # |
| 13 # You should have received a copy of the GNU General Public License | 13 # You should have received a copy of the GNU General Public License |
| 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 14 # along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| 15 | 15 |
| 16 """Tests for update manifest generation script.""" | 16 """Tests for update manifest generation script.""" |
| 17 | 17 |
| 18 import os | 18 import os |
| 19 import json | 19 import json |
| 20 import pytest | 20 import pytest |
| 21 import subprocess | 21 import subprocess |
| 22 import xml.etree.ElementTree as ET | |
| 23 | |
| 24 | |
| 25 def rdf2data(rdf): | |
| 26 """Convert RDF to a more comparable data strcuture.""" | |
| 27 # We need this to address the RDF item ordering discrepancies. | |
| 28 def et2data(node): | |
| 29 return { | |
| 30 'tag': node.tag, | |
| 31 'text': node.text, | |
| 32 'attrib': node.attrib, | |
| 33 'subs': sorted(et2data(sub) for sub in node) | |
| 34 } | |
| 35 return et2data(ET.fromstring(rdf)) | |
| 36 | 22 |
| 37 | 23 |
| 38 @pytest.fixture(scope='session') | 24 @pytest.fixture(scope='session') |
| 39 def oracle(tests_dir): | 25 def oracle(tests_dir): |
| 40 """Function that returns expected content of generated files.""" | 26 """Function that returns expected content of generated files.""" |
| 41 def expected_value_of(what): | 27 def expected_value_of(what): |
| 42 return tests_dir.join('oracle').join(what).read().strip() | 28 return tests_dir.join('oracle').join(what).read().strip() |
| 43 return expected_value_of | 29 return expected_value_of |
| 44 | 30 |
| 45 | 31 |
| 46 def test_update_manifests(config_ini, hg_dir, tmpdir, oracle): | 32 def test_update_manifests(config_ini, hg_dir, tmpdir, oracle): |
| 47 env = dict(os.environ) | 33 env = dict(os.environ) |
| 48 env['SITESCRIPTS_CONFIG'] = str(config_ini) | 34 env['SITESCRIPTS_CONFIG'] = str(config_ini) |
| 49 cmd = ['python', '-m', 'sitescripts.extensions.bin.updateUpdateManifests'] | 35 cmd = ['python', '-m', 'sitescripts.extensions.bin.updateUpdateManifests'] |
| 50 subprocess.check_call(cmd, env=env) | 36 subprocess.check_call(cmd, env=env) |
| 51 for filename in ['androidupdates.json', 'androidupdates.xml', | 37 for filename in ['androidupdates.json', 'androidupdates.xml', |
| 52 'ieupdate.json', 'update.rdf', 'updates.plist']: | 38 'ieupdate.json', 'updates.json', 'updates.plist']: |
| 53 got = tmpdir.join(filename).read().strip() | 39 got = tmpdir.join(filename).read().strip() |
| 54 expect = oracle(filename) | 40 expect = oracle(filename) |
| 55 if filename.endswith('.json'): | 41 if filename.endswith('.json'): |
| 56 got = json.loads(got) | 42 got = json.loads(got) |
| 57 expect = json.loads(expect) | 43 expect = json.loads(expect) |
| 58 elif filename.endswith('.rdf'): | |
| 59 got = rdf2data(got) | |
| 60 expect = rdf2data(expect) | |
| 61 assert got == expect | 44 assert got == expect |
| OLD | NEW |