def test_check_marathon_check_fail(mock_stdout, mock_is_file_in_dir):
# marathon.yaml exists and is valid
mock_is_file_in_dir.return_value = False
expected_output = "%s\n" % PaastaCheckMessages.MARATHON_YAML_MISSING
marathon_check('path')
output = mock_stdout.getvalue()
assert output == expected_output
def test_check_marathon_check_fail(mock_stdout, mock_is_file_in_dir):
# marathon.yaml exists and is valid
mock_is_file_in_dir.return_value = False
expected_output = "%s\n" % PaastaCheckMessages.MARATHON_YAML_MISSING
marathon_check('path')
output = mock_stdout.getvalue()
assert output == expected_output
def test_check_marathon_check_pass(mock_stdout, mock_is_file_in_dir):
# marathon.yaml exists and is valid
mock_is_file_in_dir.return_value = "/fake/path"
expected_output = "%s\n" % PaastaCheckMessages.MARATHON_YAML_FOUND
marathon_check('path')
output = mock_stdout.getvalue()
assert output == expected_output
def test_check_marathon_check_pass(mock_stdout, mock_is_file_in_dir):
# marathon.yaml exists and is valid
mock_is_file_in_dir.return_value = "/fake/path"
expected_output = "%s\n" % PaastaCheckMessages.MARATHON_YAML_FOUND
marathon_check('path')
output = mock_stdout.getvalue()
assert output == expected_output