Skip to content

Commit

Permalink
Add a test too
Browse files Browse the repository at this point in the history
  • Loading branch information
jfongatyelp committed Dec 4, 2024
1 parent 86758e2 commit 0d26945
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions tests/cli/test_cmds_validate.py
Original file line number Diff line number Diff line change
Expand Up @@ -606,6 +606,37 @@ def test_instance_validate_schema_iam_role(
assert expected_output in output


@pytest.mark.parametrize(
"iam_role, service_account_name, instance_type, expected",
[
("arn:aws:iam::12345678:role/some_role", None, "kubernetes", True),
("arn:aws:iam::12345678:role/some_role", None, "eks", True),
("arn:aws:iam::12345678:role/some_role", "some_svc_account", "eks", False),
(None, "some_svc_account", "eks", True),
],
)
def test_instance_validate_schema_sa_and_iam_role(
iam_role,
service_account_name,
instance_type,
expected,
capsys,
):
instance_content = f"""
test_instance:
{"iam_role: "+iam_role if iam_role else ""}
{"service_account_name: "+service_account_name if service_account_name else ""}
"""
with patch(
"paasta_tools.cli.cmds.validate.get_file_contents", autospec=True
) as mock_get_file_contents:
mock_get_file_contents.return_value = instance_content
assert validate_schema("unused_service_path.yaml", instance_type) == expected
expected_output = SCHEMA_VALID if expected else SCHEMA_INVALID
output, _ = capsys.readouterr()
assert expected_output in output


@patch("paasta_tools.cli.cmds.validate.get_file_contents", autospec=True)
def test_tron_validate_schema_understands_underscores(mock_get_file_contents, capsys):
tron_content = """
Expand Down

0 comments on commit 0d26945

Please sign in to comment.