tests: check how hgweb handles HEAD requests

This test file is loosely based on test-hgweb.t.

HEAD support originally implemented in fda5a4b853ab.

tests/test-hgweb-head.t

@@ -0,0 +1,102 b''
+#require serve
+
+Some tests for hgweb responding to HEAD requests
+
+  $ hg init test
+  $ cd test
+  $ mkdir da
+  $ echo foo > da/foo
+  $ echo foo > foo
+  $ hg ci -Ambase
+  adding da/foo
+  adding foo
+  $ hg bookmark -r0 '@'
+  $ hg bookmark -r0 'a b c'
+  $ hg bookmark -r0 'd/e/f'
+  $ hg serve -n test -p $HGPORT -d --pid-file=hg.pid -A access.log -E errors.log
+  $ cat hg.pid >> $DAEMON_PIDS
+
+manifest
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/?style=raw' - date etag server
+  200 Script output follows
+  content-type: text/plain; charset=ascii
+  
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/da?style=raw' - date etag server
+  200 Script output follows
+  content-type: text/plain; charset=ascii
+  
+
+plain file
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/foo?style=raw' - date etag server
+  200 Script output follows
+  content-disposition: inline; filename="foo"
+  content-length: 4
+  content-type: application/binary
+  
+
+should give a 404 - static file that does not exist
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'static/bogus' - date etag server
+  404 Not Found
+  content-type: text/html; charset=ascii
+  
+  [1]
+
+should give a 404 - bad revision
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/spam/foo?style=raw' - date etag server
+  404 Not Found
+  content-type: text/plain; charset=ascii
+  
+  [1]
+
+should give a 400 - bad command
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/foo?cmd=spam&style=raw' - date etag server
+  400* (glob)
+  content-type: text/plain; charset=ascii
+  
+  [1]
+
+should give a 404 - file does not exist
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/bork?style=raw' - date etag server
+  404 Not Found
+  content-type: text/plain; charset=ascii
+  
+  [1]
+
+try bad style
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'file/tip/?style=foobar' - date etag server
+  200 Script output follows
+  content-type: text/html; charset=ascii
+  
+
+log
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'log?style=raw' - date etag server
+  200 Script output follows
+  content-type: text/plain; charset=ascii
+  
+
+access bookmarks
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'rev/@?style=paper' - date etag server
+  200 Script output follows
+  content-type: text/html; charset=ascii
+  
+
+static file
+
+  $ get-with-headers.py localhost:$HGPORT --method=HEAD 'static/style-gitweb.css' - date etag server
+  200 Script output follows
+  content-length: 9074
+  content-type: text/css
+  
+
+  $ killdaemons.py
+
+  $ cd ..

tests/get-with-headers.py

@@ -1,130 +1,131 b''
 #!/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
 import json
 import os
 import sys
 
 from mercurial import (
     pycompat,
     util,
 )
 
 httplib = util.httplib
 
 try:
     import msvcrt
 
     msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
     msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
 except ImportError:
     pass
 
 stdout = getattr(sys.stdout, 'buffer', sys.stdout)
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--twice', action='store_true')
 parser.add_argument('--headeronly', action='store_true')
 parser.add_argument('--json', action='store_true')
 parser.add_argument('--hgproto')
 parser.add_argument(
     '--requestheader',
     nargs='*',
     default=[],
     help='Send an additional HTTP request header. Argument '
     '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='*')
 
 args = parser.parse_args()
 
 twice = args.twice
 headeronly = args.headeronly
 formatjson = args.json
 hgproto = args.hgproto
 requestheaders = args.requestheader
 
 tag = None
 
 
-def request(host, path, show):
+def request(method, host, path, show):
     assert not path.startswith('/'), path
     global tag
     headers = {}
     if tag:
         headers['If-None-Match'] = tag
     if hgproto:
         headers['X-HgProto-1'] = hgproto
 
     for header in requestheaders:
         key, value = header.split('=', 1)
         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'))
     )
     if show[:1] == ['-']:
         show = sorted(
             h for h, v in response.getheaders() if h.lower() not in show
         )
     for h in [h.lower() for h in show]:
         if response.getheader(h, None) is not None:
             stdout.write(
                 b"%s: %s\n"
                 % (h.encode('ascii'), response.getheader(h).encode('ascii'))
             )
     if headeronly:
         # still read the body to prevent windows to be unhappy about that
         # (this might some flakyness in test-hgweb-filelog.t on Windows)
         data = response.read()
     else:
         stdout.write(b'\n')
         data = response.read()
 
         if args.bodyfile:
             bodyfh = open(args.bodyfile, 'wb')
         else:
             bodyfh = stdout
 
         # Pretty print JSON. This also has the beneficial side-effect
         # of verifying emitted JSON is well-formed.
         if formatjson:
             # json.dumps() will print trailing newlines. Eliminate them
             # to make tests easier to write.
             data = pycompat.json_loads(data)
             lines = json.dumps(data, sort_keys=True, indent=2).splitlines()
             for line in lines:
                 bodyfh.write(pycompat.sysbytes(line.rstrip()))
                 bodyfh.write(b'\n')
         else:
             bodyfh.write(data)
 
         if args.bodyfile:
             bodyfh.close()
 
     if twice and response.getheader('ETag', None):
         tag = response.getheader('ETag')
 
     # further try to please the windows-flakyness deity
     conn.close()
 
     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)
 sys.exit(1)