mirror of
https://github.com/httpie/cli.git
synced 2024-11-24 08:22:22 +02:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import mock
|
|
from pytest import raises
|
|
from requests import Request, Timeout
|
|
from requests.exceptions import ConnectionError
|
|
|
|
from httpie import ExitStatus
|
|
from httpie.core import main
|
|
from utils import http, HTTP_OK
|
|
|
|
|
|
error_msg = None
|
|
|
|
|
|
@mock.patch('httpie.core.get_response')
|
|
def test_error(get_response):
|
|
def error(msg, *args, **kwargs):
|
|
global error_msg
|
|
error_msg = msg % args
|
|
|
|
exc = ConnectionError('Connection aborted')
|
|
exc.request = Request(method='GET', url='http://www.google.com')
|
|
get_response.side_effect = exc
|
|
ret = main(['--ignore-stdin', 'www.google.com'], custom_log_error=error)
|
|
assert ret == ExitStatus.ERROR
|
|
assert error_msg == (
|
|
'ConnectionError: '
|
|
'Connection aborted while doing GET request to URL: '
|
|
'http://www.google.com')
|
|
|
|
|
|
@mock.patch('httpie.core.get_response')
|
|
def test_error_traceback(get_response):
|
|
exc = ConnectionError('Connection aborted')
|
|
exc.request = Request(method='GET', url='http://www.google.com')
|
|
get_response.side_effect = exc
|
|
with raises(ConnectionError):
|
|
main(['--ignore-stdin', '--traceback', 'www.google.com'])
|
|
|
|
|
|
@mock.patch('httpie.core.get_response')
|
|
def test_timeout(get_response):
|
|
def error(msg, *args, **kwargs):
|
|
global error_msg
|
|
error_msg = msg % args
|
|
|
|
exc = Timeout('Request timed out')
|
|
exc.request = Request(method='GET', url='http://www.google.com')
|
|
get_response.side_effect = exc
|
|
ret = main(['--ignore-stdin', 'www.google.com'], custom_log_error=error)
|
|
assert ret == ExitStatus.ERROR_TIMEOUT
|
|
assert error_msg == 'Request timed out (30s).'
|
|
|
|
|
|
def test_max_headers_limit(httpbin_both):
|
|
with raises(ConnectionError) as e:
|
|
http('--max-headers=1', httpbin_both + '/get')
|
|
assert 'got more than 1 headers' in str(e.value)
|
|
|
|
|
|
def test_max_headers_no_limit(httpbin_both):
|
|
assert HTTP_OK in http('--max-headers=0', httpbin_both + '/get')
|