def test_check_pipeline_check_fail_httperr(mock_stdout, mock_urllib2, mock_error): mock_urllib2.urlopen.side_effect = mock_error expected_output = "%s\n" % PaastaCheckMessages.PIPELINE_MISSING pipeline_check("fake_service") output = mock_stdout.getvalue() assert output == expected_output
def test_check_pipeline_check_fail_404(mock_stdout, mock_urlopen): mock_result = MagicMock() mock_result.getcode.return_value = 404 mock_urlopen.return_value = mock_result expected_output = "%s\n" % PaastaCheckMessages.PIPELINE_MISSING pipeline_check("fake_service") output = mock_stdout.getvalue() assert output == expected_output