| Index: sitescripts/extensions/test/test_updateManifests.py | 
| =================================================================== | 
| new file mode 100644 | 
| --- /dev/null | 
| +++ b/sitescripts/extensions/test/test_updateManifests.py | 
| @@ -0,0 +1,131 @@ | 
| +# This file is part of the Adblock Plus web scripts, | 
| +# Copyright (C) 2006-2016 Eyeo GmbH | 
| +# | 
| +# Adblock Plus is free software: you can redistribute it and/or modify | 
| +# it under the terms of the GNU General Public License version 3 as | 
| +# published by the Free Software Foundation. | 
| +# | 
| +# Adblock Plus is distributed in the hope that it will be useful, | 
| +# but WITHOUT ANY WARRANTY; without even the implied warranty of | 
| +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
| +# GNU General Public License for more details. | 
| +# | 
| +# You should have received a copy of the GNU General Public License | 
| +# along with Adblock Plus.  If not, see <http://www.gnu.org/licenses/>. | 
| + | 
| +"""Tests for update manifest generation script.""" | 
| + | 
| +import json | 
| +import os | 
| +import subprocess | 
| +import xml.etree.ElementTree as ET | 
| + | 
| +import pytest | 
| +import py | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def tests_dir(): | 
| +    """Directory that contains this tests and the data files it uses.""" | 
| +    return py.path.local(__file__).dirpath() | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def oracle(tests_dir): | 
| +    """Function that returns expected content of generated files.""" | 
| +    def expected_value_of(what): | 
| +        return tests_dir.join('oracle').join(what).read().strip() | 
| +    return expected_value_of | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def data_dir(tests_dir): | 
| +    return tests_dir.join('data') | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def keys_dir(tmpdir, tests_dir): | 
| +    keys_dir = tmpdir.mkdir('keys') | 
| +    key_filename = 'adblockplussafari.pem' | 
| +    tests_dir.join(key_filename).copy(keys_dir.join(key_filename)) | 
| +    return keys_dir | 
| + | 
| + | 
| +def call_hg(cwd, *params): | 
| +    return subprocess.check_call(['hg'] + list(params), cwd=str(cwd)) | 
| + | 
| + | 
| +REPOS = { | 
| +    'adblockplus': ('metadata.gecko', '2.7.3'), | 
| +    'adblockplusie': ('README.txt', '1.33.7'), | 
| +    'adblockpluschrome': ('metadata.safari', '1.12.3'), | 
| +    'adblockplusandroid': ('AndroidManifest.xml', '1.3') | 
| +} | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def hg_dir(tmpdir, data_dir): | 
| +    """Directory that contains the repository mocks.""" | 
| +    hg_dir = tmpdir.mkdir('hg') | 
| + | 
| +    # Mock plugin repositories. | 
| +    for repo, config in REPOS.items(): | 
| +        filename, tag = config | 
| +        repo_dir = hg_dir.mkdir(repo) | 
| +        call_hg(repo_dir, 'init') | 
| +        data_dir.join(filename).copy(repo_dir.join(filename)) | 
| +        call_hg(repo_dir, 'add', filename) | 
| +        call_hg(repo_dir, 'commit', '-m', '1') | 
| +        call_hg(repo_dir, 'tag', tag) | 
| + | 
| +    # Mock the downloads repository. | 
| +    downloads_list = data_dir.join('downloads.list').read().splitlines() | 
| +    downloads_dir = hg_dir.mkdir('downloads') | 
| +    call_hg(downloads_dir, 'init') | 
| +    for item in downloads_list: | 
| +        downloads_dir.join(item).write('') | 
| +    call_hg(downloads_dir, 'add', *downloads_list) | 
| +    call_hg(downloads_dir, 'commit', '-m', 'ok') | 
| + | 
| +    return hg_dir | 
| + | 
| + | 
| +@pytest.fixture() | 
| +def config_ini(tests_dir, tmpdir, hg_dir, keys_dir): | 
| +    """Sitescripts configuration.""" | 
| +    template = tests_dir.join('sitescripts.ini.template').read() | 
| +    conf = template.format(hg_dir=hg_dir, out_dir=tmpdir, keys_dir=keys_dir) | 
| +    config_ini = tmpdir.join('sitescripts.ini') | 
| +    config_ini.write(conf) | 
| +    return config_ini | 
| + | 
| + | 
| +def rdf2data(rdf): | 
| +    """Convert RDF to a more comparable data strcuture.""" | 
| +    # We need this to address the RDF item ordering discrepancies. | 
| +    def et2data(node): | 
| +        return { | 
| +            'tag': node.tag, | 
| +            'text': node.text, | 
| +            'attrib': node.attrib, | 
| +            'subs': sorted(et2data(sub) for sub in node) | 
| +        } | 
| +    return et2data(ET.fromstring(rdf)) | 
| + | 
| + | 
| +def test_update_manifests(config_ini, hg_dir, tmpdir, oracle): | 
| +    env = dict(os.environ) | 
| +    env['SITESCRIPTS_CONFIG'] = str(config_ini) | 
| +    cmd = ['python', '-m', 'sitescripts.extensions.bin.updateUpdateManifests'] | 
| +    subprocess.check_call(cmd, env=env) | 
| +    for filename in ['androidupdates.json', 'androidupdates.xml', | 
| +                     'ieupdate.json', 'update.rdf', 'updates.plist']: | 
| +        got = tmpdir.join(filename).read().strip() | 
| +        expect = oracle(filename) | 
| +        if filename.endswith('.json'): | 
| +            got = json.loads(got) | 
| +            expect = json.loads(expect) | 
| +        elif filename.endswith('.rdf'): | 
| +            got = rdf2data(got) | 
| +            expect = rdf2data(expect) | 
| +        assert got == expect | 
|  |