Fixed query removal bug in make_pre_authed_request
Change-Id: I1b8238fb2ffe07b1474f7d8f040fdc620b6897d7
This commit is contained in:
@@ -330,7 +330,7 @@ def make_pre_authed_env(env, method=None, path=None, agent='Swift',
|
|||||||
newenv['REQUEST_METHOD'] = method
|
newenv['REQUEST_METHOD'] = method
|
||||||
if path:
|
if path:
|
||||||
newenv['PATH_INFO'] = path
|
newenv['PATH_INFO'] = path
|
||||||
if query_string:
|
if query_string is not None:
|
||||||
newenv['QUERY_STRING'] = query_string
|
newenv['QUERY_STRING'] = query_string
|
||||||
if agent:
|
if agent:
|
||||||
newenv['HTTP_USER_AGENT'] = (
|
newenv['HTTP_USER_AGENT'] = (
|
||||||
|
|||||||
@@ -195,5 +195,17 @@ class TestWSGI(unittest.TestCase):
|
|||||||
body='tester', headers={})
|
body='tester', headers={})
|
||||||
self.assertEquals(r.path, quote('/a space'))
|
self.assertEquals(r.path, quote('/a space'))
|
||||||
|
|
||||||
|
def test_pre_auth_req_drops_query(self):
|
||||||
|
r = wsgi.make_pre_authed_request(
|
||||||
|
{'QUERY_STRING': 'original'}, 'GET', 'path')
|
||||||
|
self.assertEquals(r.query_string, 'original')
|
||||||
|
r = wsgi.make_pre_authed_request(
|
||||||
|
{'QUERY_STRING': 'original'}, 'GET', 'path?replacement')
|
||||||
|
self.assertEquals(r.query_string, 'replacement')
|
||||||
|
r = wsgi.make_pre_authed_request(
|
||||||
|
{'QUERY_STRING': 'original'}, 'GET', 'path?')
|
||||||
|
self.assertEquals(r.query_string, '')
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|||||||
Reference in New Issue
Block a user