gecko-dev/python/mozrelease/test/test_balrog.py

44 строки
1.3 KiB
Python

# -*- coding: utf-8 -*-
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import, print_function, unicode_literals
import mozunit
import pytest
import json
from mozrelease.util import load as yaml_load
from pathlib2 import Path
from mozrelease.balrog import generate_update_properties
from mozilla_version.gecko import GeckoVersion
DATA_PATH = Path(__file__).parent.joinpath('data')
@pytest.mark.parametrize('context,config_file,output_file', [
({
'release-type': 'release',
'product': 'firefox',
'version': GeckoVersion.parse('62.0.3'),
}, 'whatsnew-62.0.3.yml', 'Firefox-62.0.3.update.json'),
({
'release-type': 'beta',
'product': 'firefox',
'version': GeckoVersion.parse('64.0'),
}, 'whatsnew-62.0.3.yml', 'Firefox-64.0b13.update.json'),
])
def test_update_properties(context, config_file, output_file):
with DATA_PATH.joinpath(config_file).open('r', encoding='utf-8') as f:
config = yaml_load(f)
update_line = generate_update_properties(context, config)
assert update_line == json.load(DATA_PATH.joinpath(output_file).open('r', encoding='utf-8'))
if __name__ == '__main__':
mozunit.main()