test-wsgirequest.py
422 lines
| 17.7 KiB
| text/x-python
|
PythonLexer
/ tests / test-wsgirequest.py
|
r36912 | from __future__ import absolute_import, print_function | ||
import unittest | ||||
from mercurial.hgweb import ( | ||||
request as requestmod, | ||||
) | ||||
|
r36913 | from mercurial import ( | ||
error, | ||||
) | ||||
|
r36912 | |||
DEFAULT_ENV = { | ||||
r'REQUEST_METHOD': r'GET', | ||||
r'SERVER_NAME': r'testserver', | ||||
r'SERVER_PORT': r'80', | ||||
r'SERVER_PROTOCOL': r'http', | ||||
r'wsgi.version': (1, 0), | ||||
r'wsgi.url_scheme': r'http', | ||||
r'wsgi.input': None, | ||||
r'wsgi.errors': None, | ||||
r'wsgi.multithread': False, | ||||
r'wsgi.multiprocess': True, | ||||
r'wsgi.run_once': False, | ||||
} | ||||
|
r36928 | def parse(env, reponame=None, altbaseurl=None, extra=None): | ||
|
r36912 | env = dict(env) | ||
env.update(extra or {}) | ||||
|
r36928 | return requestmod.parserequestfromenv(env, reponame=reponame, | ||
|
r36916 | altbaseurl=altbaseurl) | ||
|
r36912 | |||
class ParseRequestTests(unittest.TestCase): | ||||
def testdefault(self): | ||||
r = parse(DEFAULT_ENV) | ||||
self.assertEqual(r.url, b'http://testserver') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.method, b'GET') | ||||
self.assertIsNone(r.remoteuser) | ||||
self.assertIsNone(r.remotehost) | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
|
r36914 | self.assertIsNone(r.dispatchpath) | ||
|
r36912 | self.assertIsNone(r.reponame) | ||
self.assertEqual(r.querystring, b'') | ||||
self.assertEqual(len(r.qsparams), 0) | ||||
self.assertEqual(len(r.headers), 0) | ||||
def testcustomport(self): | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SERVER_PORT': r'8000', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver:8000') | ||||
self.assertEqual(r.baseurl, b'http://testserver:8000') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver:8000') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver:8000') | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SERVER_PORT': r'4000', | ||||
r'wsgi.url_scheme': r'https', | ||||
}) | ||||
self.assertEqual(r.url, b'https://testserver:4000') | ||||
self.assertEqual(r.baseurl, b'https://testserver:4000') | ||||
self.assertEqual(r.advertisedurl, b'https://testserver:4000') | ||||
self.assertEqual(r.advertisedbaseurl, b'https://testserver:4000') | ||||
def testhttphost(self): | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'HTTP_HOST': r'altserver', | ||||
}) | ||||
self.assertEqual(r.url, b'http://altserver') | ||||
self.assertEqual(r.baseurl, b'http://altserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
def testscriptname(self): | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SCRIPT_NAME': r'', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
|
r36914 | self.assertIsNone(r.dispatchpath) | ||
|
r36912 | |||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SCRIPT_NAME': r'/script', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/script') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/script') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/script') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
|
r36914 | self.assertIsNone(r.dispatchpath) | ||
|
r36912 | |||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SCRIPT_NAME': r'/multiple words', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/multiple%20words') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/multiple%20words') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/multiple words') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
|
r36914 | self.assertIsNone(r.dispatchpath) | ||
|
r36912 | |||
def testpathinfo(self): | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'PATH_INFO': r'', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertEqual(r.dispatchpath, b'') | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'PATH_INFO': r'/pathinfo', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/pathinfo') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/pathinfo') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, [b'pathinfo']) | ||||
self.assertEqual(r.dispatchpath, b'pathinfo') | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'PATH_INFO': r'/one/two/', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/one/two/') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/one/two/') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, [b'one', b'two']) | ||||
self.assertEqual(r.dispatchpath, b'one/two') | ||||
def testscriptandpathinfo(self): | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SCRIPT_NAME': r'/script', | ||||
r'PATH_INFO': r'/pathinfo', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/script/pathinfo') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/script/pathinfo') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/script') | ||||
self.assertEqual(r.dispatchparts, [b'pathinfo']) | ||||
self.assertEqual(r.dispatchpath, b'pathinfo') | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'SCRIPT_NAME': r'/script1/script2', | ||||
r'PATH_INFO': r'/path1/path2', | ||||
}) | ||||
self.assertEqual(r.url, | ||||
b'http://testserver/script1/script2/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://testserver/script1/script2/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/script1/script2') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
r = parse(DEFAULT_ENV, extra={ | ||||
r'HTTP_HOST': r'hostserver', | ||||
r'SCRIPT_NAME': r'/script', | ||||
r'PATH_INFO': r'/pathinfo', | ||||
}) | ||||
self.assertEqual(r.url, b'http://hostserver/script/pathinfo') | ||||
self.assertEqual(r.baseurl, b'http://hostserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/script/pathinfo') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/script') | ||||
self.assertEqual(r.dispatchparts, [b'pathinfo']) | ||||
self.assertEqual(r.dispatchpath, b'pathinfo') | ||||
|
r37733 | if not getattr(unittest.TestCase, 'assertRaisesRegex', False): | ||
# Python 3.7 deprecates the regex*p* version, but 2.7 lacks | ||||
# the regex version. | ||||
assertRaisesRegex = (# camelcase-required | ||||
unittest.TestCase.assertRaisesRegexp) | ||||
|
r36912 | def testreponame(self): | ||
|
r36913 | """repository path components get stripped from URL.""" | ||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, | ||
|
r37912 | 'reponame requires PATH_INFO'): | ||
|
r36913 | parse(DEFAULT_ENV, reponame=b'repo') | ||
|
r36912 | |||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, | ||
|
r37912 | 'PATH_INFO does not begin with repo ' | ||
'name'): | ||||
|
r36913 | parse(DEFAULT_ENV, reponame=b'repo', extra={ | ||
r'PATH_INFO': r'/pathinfo', | ||||
}) | ||||
|
r36912 | |||
|
r37733 | with self.assertRaisesRegex(error.ProgrammingError, | ||
|
r37912 | 'reponame prefix of PATH_INFO'): | ||
|
r36913 | parse(DEFAULT_ENV, reponame=b'repo', extra={ | ||
r'PATH_INFO': r'/repoextra/path', | ||||
}) | ||||
r = parse(DEFAULT_ENV, reponame=b'repo', extra={ | ||||
|
r36912 | r'PATH_INFO': r'/repo/path1/path2', | ||
}) | ||||
self.assertEqual(r.url, b'http://testserver/repo/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://testserver/repo/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/repo') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertEqual(r.reponame, b'repo') | ||||
|
r36913 | r = parse(DEFAULT_ENV, reponame=b'prefix/repo', extra={ | ||
|
r36912 | r'PATH_INFO': r'/prefix/repo/path1/path2', | ||
}) | ||||
self.assertEqual(r.url, b'http://testserver/prefix/repo/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://testserver/prefix/repo/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://testserver') | ||||
self.assertEqual(r.apppath, b'/prefix/repo') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertEqual(r.reponame, b'prefix/repo') | ||||
|
r36916 | def testaltbaseurl(self): | ||
# Simple hostname remap. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver') | ||
|
r36916 | |||
self.assertEqual(r.url, b'http://testserver') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# With a custom port. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver:8000') | ||
|
r36916 | self.assertEqual(r.url, b'http://testserver') | ||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver:8000') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver:8000') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# With a changed protocol. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'https://altserver') | ||
|
r36916 | self.assertEqual(r.url, b'http://testserver') | ||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'https://altserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'https://altserver') | ||||
# URL scheme is defined as the actual scheme, not advertised. | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# Need to specify explicit port number for proper https:// alt URLs. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'https://altserver:443') | ||
|
r36916 | self.assertEqual(r.url, b'http://testserver') | ||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'https://altserver') | ||||
self.assertEqual(r.advertisedbaseurl, b'https://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# With only PATH_INFO defined. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver', extra={ | ||
|
r36916 | r'PATH_INFO': r'/path1/path2', | ||
}) | ||||
self.assertEqual(r.url, b'http://testserver/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertIsNone(r.reponame) | ||||
# Path on alt URL. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver/altpath') | ||
|
r36916 | self.assertEqual(r.url, b'http://testserver') | ||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver/altpath') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'/altpath') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# With a trailing slash. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver/altpath/') | ||
|
r36916 | self.assertEqual(r.url, b'http://testserver') | ||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver/altpath/') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'/altpath/') | ||||
self.assertEqual(r.dispatchparts, []) | ||||
self.assertIsNone(r.dispatchpath) | ||||
self.assertIsNone(r.reponame) | ||||
# PATH_INFO + path on alt URL. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver/altpath', extra={ | ||
|
r36916 | r'PATH_INFO': r'/path1/path2', | ||
}) | ||||
self.assertEqual(r.url, b'http://testserver/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://altserver/altpath/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'/altpath') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertIsNone(r.reponame) | ||||
# PATH_INFO + path on alt URL with trailing slash. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver/altpath/', extra={ | ||
|
r36916 | r'PATH_INFO': r'/path1/path2', | ||
}) | ||||
self.assertEqual(r.url, b'http://testserver/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://altserver/altpath//path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'/altpath/') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertIsNone(r.reponame) | ||||
# Local SCRIPT_NAME is ignored. | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver', extra={ | ||
|
r36916 | r'SCRIPT_NAME': r'/script', | ||
r'PATH_INFO': r'/path1/path2', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/script/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, b'http://altserver/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertIsNone(r.reponame) | ||||
# Use remote's path for script name, app path | ||||
|
r37912 | r = parse(DEFAULT_ENV, altbaseurl=b'http://altserver/altroot', extra={ | ||
|
r36916 | r'SCRIPT_NAME': r'/script', | ||
r'PATH_INFO': r'/path1/path2', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/script/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://altserver/altroot/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.urlscheme, b'http') | ||||
self.assertEqual(r.apppath, b'/altroot') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertIsNone(r.reponame) | ||||
# reponame is factored in properly. | ||||
r = parse(DEFAULT_ENV, reponame=b'repo', | ||||
|
r37912 | altbaseurl=b'http://altserver/altroot', | ||
|
r36916 | extra={ | ||
r'SCRIPT_NAME': r'/script', | ||||
r'PATH_INFO': r'/repo/path1/path2', | ||||
}) | ||||
self.assertEqual(r.url, b'http://testserver/script/repo/path1/path2') | ||||
self.assertEqual(r.baseurl, b'http://testserver') | ||||
self.assertEqual(r.advertisedurl, | ||||
b'http://altserver/altroot/repo/path1/path2') | ||||
self.assertEqual(r.advertisedbaseurl, b'http://altserver') | ||||
self.assertEqual(r.apppath, b'/altroot/repo') | ||||
self.assertEqual(r.dispatchparts, [b'path1', b'path2']) | ||||
self.assertEqual(r.dispatchpath, b'path1/path2') | ||||
self.assertEqual(r.reponame, b'repo') | ||||
|
r36912 | if __name__ == '__main__': | ||
import silenttestrunner | ||||
silenttestrunner.main(__name__) | ||||