wehub-resource-sync
131 lines
3.5 KiB
Python
131 lines
3.5 KiB
Python
import base64
|
|
import io
|
|
import sys
|
|
from typing import List
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from ci.ray_ci.utils import (
|
|
chunk_into_n,
|
|
ecr_docker_login,
|
|
filter_tests,
|
|
get_flaky_test_names,
|
|
)
|
|
|
|
from ray_release.test import Test
|
|
|
|
|
|
def test_chunk_into_n() -> None:
|
|
assert chunk_into_n([1, 2, 3, 4, 5], 2) == [[1, 2, 3], [4, 5]]
|
|
assert chunk_into_n([1, 2], 3) == [[1], [2], []]
|
|
assert chunk_into_n([1, 2], 1) == [[1, 2]]
|
|
|
|
|
|
@mock.patch("boto3.client")
|
|
def test_ecr_docker_login(mock_client) -> None:
|
|
def _mock_subprocess_run(
|
|
cmd: List[str],
|
|
stdin=None,
|
|
stdout=None,
|
|
stderr=None,
|
|
check=True,
|
|
) -> None:
|
|
assert stdin.read() == b"password"
|
|
|
|
mock_client.return_value.get_authorization_token.return_value = {
|
|
"authorizationData": [
|
|
{"authorizationToken": base64.b64encode(b"AWS:password")},
|
|
],
|
|
}
|
|
|
|
with mock.patch("subprocess.run", side_effect=_mock_subprocess_run):
|
|
ecr_docker_login("docker_ecr")
|
|
|
|
|
|
def _make_test(name: str, state: str, team: str) -> Test:
|
|
return Test(
|
|
{
|
|
"name": name,
|
|
"state": state,
|
|
"team": team,
|
|
}
|
|
)
|
|
|
|
|
|
@mock.patch("ray_release.test.Test.gen_from_s3")
|
|
def test_get_flaky_test_names(mock_gen_from_s3):
|
|
mock_gen_from_s3.side_effect = (
|
|
[
|
|
_make_test("darwin://test_1", "flaky", "core"),
|
|
_make_test("darwin://test_2", "flaky", "ci"),
|
|
_make_test("darwin://test_3", "passing", "core"),
|
|
],
|
|
[
|
|
_make_test("linux://test_1", "flaky", "core"),
|
|
_make_test("linux://test_2", "passing", "ci"),
|
|
],
|
|
)
|
|
flaky_test_names = get_flaky_test_names(
|
|
prefix="darwin:",
|
|
)
|
|
assert flaky_test_names == ["//test_1", "//test_2"]
|
|
flaky_test_names = get_flaky_test_names(
|
|
prefix="linux:",
|
|
)
|
|
assert flaky_test_names == ["//test_1"]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"state_filter, expected_value",
|
|
[
|
|
(
|
|
"-flaky",
|
|
"//test_3\n//test_4\n",
|
|
),
|
|
(
|
|
"flaky",
|
|
"//test_1\n//test_2\n",
|
|
),
|
|
],
|
|
)
|
|
@mock.patch("ray_release.test.Test.gen_from_s3")
|
|
def test_filter_tests(mock_gen_from_s3, state_filter, expected_value):
|
|
# Setup test input/output
|
|
mock_gen_from_s3.side_effect = (
|
|
[
|
|
_make_test("darwin://test_1", "flaky", "core"),
|
|
_make_test("darwin://test_2", "flaky", "ci"),
|
|
_make_test("darwin://test_3", "passing", "core"),
|
|
_make_test("darwin://test_4", "passing", "ci"),
|
|
],
|
|
)
|
|
test_targets = ["//test_1", "//test_2", "//test_3", "//test_4"]
|
|
output = io.StringIO()
|
|
|
|
filter_tests(io.StringIO("\n".join(test_targets)), output, "darwin:", state_filter)
|
|
assert output.getvalue() == expected_value
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"state_filter, prefix, error_message",
|
|
[
|
|
(
|
|
"wrong-option", # invalid filter option
|
|
"darwin:",
|
|
"Filter option must be one of",
|
|
),
|
|
("-flaky", "wrong-prefix", "Prefix must be one of"), # invalid prefix
|
|
],
|
|
)
|
|
def test_filter_tests_fail(state_filter, prefix, error_message):
|
|
test_targets = ["//test_1", "//test_2", "//test_3", "//test_4"]
|
|
output = io.StringIO()
|
|
with pytest.raises(ValueError, match=error_message):
|
|
filter_tests(io.StringIO("\n".join(test_targets)), output, prefix, state_filter)
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-v", __file__]))
|