# HG changeset patch
# User Anton Shestakov <av6@dwimlabs.net>
# Date 2023-01-11 13:51:04
# Node ID cd125eef4388bc06eeef658662c80b802154edfd
# Parent  3cbd0e919165f7bdb68aec6feb2c93533b4fa0b7

tests: check how hgweb handles HEAD requests

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

HEAD support originally implemented in fda5a4b853ab.

diff --git a/tests/get-with-headers.py b/tests/get-with-headers.py
--- a/tests/get-with-headers.py
+++ b/tests/get-with-headers.py
@@ -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 @@ parser.add_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='*')
@@ -54,7 +55,7 @@ requestheaders = args.requestheader
 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 @@ def request(host, path, show):
         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 @@ def request(host, path, show):
     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)
diff --git a/tests/test-hgweb-head.t b/tests/test-hgweb-head.t
new file mode 100644
--- /dev/null
+++ b/tests/test-hgweb-head.t
@@ -0,0 +1,102 @@
+#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 ..