extern/httpie-cli
1
0
Fork 1
mirror of https://github.com/httpie/cli.git synced 2025-03-13 06:18:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge eaa3d15741 into 5b604c37c6

Browse Source
This commit is contained in:
Alexander 2024-12-22 22:13:40 -03:00 committed by GitHub
commit ac51408a58
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 39 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
httpie/core.py
View File

@ -208,6 +208,7 @@ def program(args: argparse.Namespace, env: Environment) -> ExitStatus:
request_body_read_callback=request_body_read_callback)
force_separator = False
prev_with_body = False
request_message = None
# Process messages as they’re generated
for message in messages:
@ -231,14 +232,37 @@ def program(args: argparse.Namespace, env: Environment) -> ExitStatus:
exit_status = http_status_to_exit_status(http_status=message.status_code, follow=args.follow)
if exit_status != ExitStatus.SUCCESS and (not env.stdout_isatty or args.quiet == 1):
env.log_error(f'HTTP {message.raw.status} {message.raw.reason}', level=LogLevel.WARNING)
write_message(
requests_message=message,
env=env,
output_options=output_options._replace(
body=do_write_body
),
processing_options=processing_options
)
if args.auth_type == 'digest':
if isinstance(message, requests.models.PreparedRequest):
request_message = message
request_output_options = output_options._replace(body=do_write_body)
elif isinstance(message, requests.models.Response):
request_message.headers = message.request.headers
write_message(
requests_message=request_message,
env=env,
output_options=request_output_options,
processing_options=processing_options
)
write_message(
requests_message=message,
env=env,
output_options=output_options._replace(
body=do_write_body
),
processing_options=processing_options
)
else:
continue
else:
write_message(
requests_message=message,
env=env,
output_options=output_options._replace(
body=do_write_body
),
processing_options=processing_options
)
prev_with_body = output_options.body
# Cleanup

7
tests/test_auth.py
View File

@ -153,3 +153,10 @@ def test_ignore_netrc_null_auth():
env=MockEnvironment(),
)
assert isinstance(args.auth, ExplicitNullAuth)
@pytest.mark.parametrize('argument_name', ['--auth-type', '-A'])
def test_digest_auth_auth_header(httpbin_both, argument_name):
r = http(argument_name + '=digest', '--auth=user:password',
'GET', httpbin_both.url + '/digest-auth/auth/user/password', '-v')
assert 'Authorization: Digest username="user", realm="me@kennethreitz.com"' in r