def test_wait_on_exit(): exit_status = 3 mock_container = mock.Mock( spec=Container, name='cname', wait=mock.Mock(return_value=exit_status)) expected = '{} exited with code {}\n'.format(mock_container.name, exit_status) assert expected == wait_on_exit(mock_container)
def test_wait_on_exit(): exit_status = 3 mock_container = mock.Mock(spec=Container, name='cname', wait=mock.Mock(return_value=exit_status)) expected = '{} exited with code {}\n'.format(mock_container.name, exit_status) assert expected == wait_on_exit(mock_container)
def test_wait_on_exit_raises(): status_code = 500 def mock_wait(): resp = requests.Response() resp.status_code = status_code raise APIError("Bad server", resp) mock_container = mock.Mock(spec=Container, name="cname", wait=mock_wait) expected = "Unexpected API error for {} (HTTP code {})\n".format(mock_container.name, status_code) assert expected in wait_on_exit(mock_container)
def test_wait_on_exit_raises(): status_code = 500 def mock_wait(): resp = requests.Response() resp.status_code = status_code raise APIError('Bad server', resp) mock_container = mock.Mock(spec=Container, name='cname', wait=mock_wait) expected = 'Unexpected API error for {} (HTTP code {})\n'.format( mock_container.name, status_code, ) assert expected in wait_on_exit(mock_container)