gecko-dev/testing/raptor/test/test_utils.py

81 строка
2.2 KiB
Python

from __future__ import absolute_import, unicode_literals
import mozunit
import os
import pytest
import sys
import tempfile
import yaml
# need this so raptor imports work both from /raptor and via mach
here = os.path.abspath(os.path.dirname(__file__))
raptor_dir = os.path.join(os.path.dirname(here), "raptor")
sys.path.insert(0, raptor_dir)
from utils import bool_from_str, transform_platform, write_yml_file
@pytest.mark.parametrize("platform", ["win", "mac", "linux64"])
def test_transform_platform(platform):
transformed = transform_platform("mitmproxy-rel-bin-{platform}.manifest", platform)
assert "{platform}" not in transformed
if platform == "mac":
assert "osx" in transformed
else:
assert platform in transformed
def test_transform_platform_no_change():
starting_string = "nothing-in-here-to-transform"
assert transform_platform(starting_string, "mac") == starting_string
@pytest.mark.parametrize("processor", ["x86_64", "other"])
def test_transform_platform_processor(processor):
transformed = transform_platform(
"string-with-processor-{x64}.manifest", "win", processor
)
assert "{x64}" not in transformed
if processor == "x86_64":
assert "_x64" in transformed
def test_write_yml_file():
yml_file = os.path.join(tempfile.mkdtemp(), "utils.yaml")
yml_data = dict(args=["--a", "apple", "--b", "banana"], env=dict(LOG_VERBOSE=1))
assert not os.path.exists(yml_file)
write_yml_file(yml_file, yml_data)
assert os.path.exists(yml_file)
with open(yml_file, "r") as yml_in:
yml_loaded = yaml.unsafe_load(yml_in)
assert yml_loaded == yml_data
@pytest.mark.parametrize(
"value, expected_result",
[
("true", True),
("TRUE", True),
("True", True),
("false", False),
("FALSE", False),
("False", False),
],
)
def test_bool_from_str(value, expected_result):
assert expected_result == bool_from_str(value)
@pytest.mark.parametrize("invalid_value", ["invalid_str", ""])
def test_bool_from_str_with_invalid_values(invalid_value):
with pytest.raises(ValueError):
bool_from_str(invalid_value)
if __name__ == "__main__":
mozunit.main()