forked from mirrors/gecko-dev
d024a31639
This patch was generated via: $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f) $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f) The same set of commands are being applied to standalone taskgraph as well so they remain in sync. Differential Revision: https://phabricator.services.mozilla.com/D123234
131 lines
4.1 KiB
Python
131 lines
4.1 KiB
Python
# 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/.
|
|
|
|
|
|
import unittest
|
|
|
|
from taskgraph.parameters import (
|
|
Parameters,
|
|
load_parameters_file,
|
|
)
|
|
from mozunit import main, MockedOpen
|
|
|
|
|
|
class TestParameters(unittest.TestCase):
|
|
|
|
vals = {
|
|
"app_version": "app_version",
|
|
"backstop": False,
|
|
"base_repository": "base_repository",
|
|
"build_date": 0,
|
|
"build_number": 0,
|
|
"do_not_optimize": [],
|
|
"existing_tasks": {},
|
|
"filters": [],
|
|
"head_ref": "head_ref",
|
|
"head_repository": "head_repository",
|
|
"head_rev": "head_rev",
|
|
"hg_branch": "hg_branch",
|
|
"level": "level",
|
|
"message": "message",
|
|
"moz_build_date": "moz_build_date",
|
|
"next_version": "next_version",
|
|
"optimize_strategies": None,
|
|
"optimize_target_tasks": False,
|
|
"owner": "owner",
|
|
"phabricator_diff": "phabricator_diff",
|
|
"project": "project",
|
|
"pushdate": 0,
|
|
"pushlog_id": "pushlog_id",
|
|
"release_enable_emefree": False,
|
|
"release_enable_partner_repack": False,
|
|
"release_enable_partner_attribution": False,
|
|
"release_eta": None,
|
|
"release_history": {},
|
|
"release_partners": [],
|
|
"release_partner_config": None,
|
|
"release_partner_build_number": 1,
|
|
"release_type": "release_type",
|
|
"release_product": None,
|
|
"required_signoffs": [],
|
|
"signoff_urls": {},
|
|
"target_tasks_method": "target_tasks_method",
|
|
"test_manifest_loader": "default",
|
|
"tasks_for": "tasks_for",
|
|
"try_mode": "try_mode",
|
|
"try_options": None,
|
|
"try_task_config": {},
|
|
"version": "version",
|
|
}
|
|
|
|
def test_Parameters_immutable(self):
|
|
p = Parameters(**self.vals)
|
|
|
|
def assign():
|
|
p["head_ref"] = 20
|
|
|
|
self.assertRaises(Exception, assign)
|
|
|
|
def test_Parameters_missing_KeyError(self):
|
|
p = Parameters(**self.vals)
|
|
self.assertRaises(KeyError, lambda: p["z"])
|
|
|
|
def test_Parameters_invalid_KeyError(self):
|
|
"""even if the value is present, if it's not a valid property, raise KeyError"""
|
|
p = Parameters(xyz=10, strict=True, **self.vals)
|
|
self.assertRaises(Exception, lambda: p.check())
|
|
|
|
def test_Parameters_get(self):
|
|
p = Parameters(head_ref=10, level=20)
|
|
self.assertEqual(p["head_ref"], 10)
|
|
|
|
def test_Parameters_check(self):
|
|
p = Parameters(**self.vals)
|
|
p.check() # should not raise
|
|
|
|
def test_Parameters_check_missing(self):
|
|
p = Parameters()
|
|
self.assertRaises(Exception, lambda: p.check())
|
|
|
|
p = Parameters(strict=False)
|
|
p.check() # should not raise
|
|
|
|
def test_Parameters_check_extra(self):
|
|
p = Parameters(xyz=10, **self.vals)
|
|
self.assertRaises(Exception, lambda: p.check())
|
|
|
|
p = Parameters(strict=False, xyz=10, **self.vals)
|
|
p.check() # should not raise
|
|
|
|
def test_load_parameters_file_yaml(self):
|
|
with MockedOpen({"params.yml": "some: data\n"}):
|
|
self.assertEqual(load_parameters_file("params.yml"), {"some": "data"})
|
|
|
|
def test_load_parameters_file_json(self):
|
|
with MockedOpen({"params.json": '{"some": "data"}'}):
|
|
self.assertEqual(load_parameters_file("params.json"), {"some": "data"})
|
|
|
|
def test_load_parameters_override(self):
|
|
"""
|
|
When ``load_parameters_file`` is passed overrides, they are included in
|
|
the generated parameters.
|
|
"""
|
|
self.assertEqual(
|
|
load_parameters_file("", overrides={"some": "data"}), {"some": "data"}
|
|
)
|
|
|
|
def test_load_parameters_override_file(self):
|
|
"""
|
|
When ``load_parameters_file`` is passed overrides, they overwrite data
|
|
loaded from a file.
|
|
"""
|
|
with MockedOpen({"params.json": '{"some": "data"}'}):
|
|
self.assertEqual(
|
|
load_parameters_file("params.json", overrides={"some": "other"}),
|
|
{"some": "other"},
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|