Mercurial > public > mercurial-scm > hg-stable
diff tests/get-with-headers.py @ 49978:cd125eef4388
tests: check how hgweb handles HEAD requests
This test file is loosely based on test-hgweb.t.
HEAD support originally implemented in fda5a4b853ab.
author | Anton Shestakov <av6@dwimlabs.net> |
---|---|
date | Wed, 11 Jan 2023 17:51:04 +0400 |
parents | 6000f5b25c9b |
children | f8f14e6d032b |
line wrap: on
line diff
--- a/tests/get-with-headers.py Sun Jan 08 16:19:10 2023 +0400 +++ b/tests/get-with-headers.py Wed Jan 11 17:51:04 2023 +0400 @@ -1,7 +1,7 @@ #!/usr/bin/env python -"""This does HTTP GET requests given a host:port and path and returns -a subset of the headers plus the body of the result.""" +"""This does HTTP requests (GET by default) given a host:port and path and +returns a subset of the headers plus the body of the result.""" import argparse @@ -39,6 +39,7 @@ 'value is <header>=<value>', ) parser.add_argument('--bodyfile', help='Write HTTP response body to a file') +parser.add_argument('--method', default='GET', help='HTTP method to use') parser.add_argument('host') parser.add_argument('path') parser.add_argument('show', nargs='*') @@ -54,7 +55,7 @@ tag = None -def request(host, path, show): +def request(method, host, path, show): assert not path.startswith('/'), path global tag headers = {} @@ -68,7 +69,7 @@ headers[key] = value conn = httplib.HTTPConnection(host) - conn.request("GET", '/' + path, None, headers) + conn.request(method, '/' + path, None, headers) response = conn.getresponse() stdout.write( b'%d %s\n' % (response.status, response.reason.encode('ascii')) @@ -121,9 +122,9 @@ return response.status -status = request(args.host, args.path, args.show) +status = request(args.method, args.host, args.path, args.show) if twice: - status = request(args.host, args.path, args.show) + status = request(args.method, args.host, args.path, args.show) if 200 <= status <= 305: sys.exit(0)