From e884caf369619c5b117f4fee08f75fb8d286fc64 Mon Sep 17 00:00:00 2001 From: Bert JW Regeer Date: Fri, 31 Aug 2018 22:26:32 -0600 Subject: Fix tests that verify WSGI environment --- waitress/tests/test_task.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'waitress/tests') diff --git a/waitress/tests/test_task.py b/waitress/tests/test_task.py index 15885eb..d620a90 100644 --- a/waitress/tests/test_task.py +++ b/waitress/tests/test_task.py @@ -704,8 +704,8 @@ class TestWSGITask(unittest.TestCase): 'PATH_INFO', 'QUERY_STRING', 'REMOTE_ADDR', 'REQUEST_METHOD', 'SCRIPT_NAME', 'SERVER_NAME', 'SERVER_PORT', 'SERVER_PROTOCOL', 'SERVER_SOFTWARE', 'wsgi.errors', 'wsgi.file_wrapper', 'wsgi.input', - 'wsgi.multiprocess', 'wsgi.multithread', 'wsgi.run_once', - 'wsgi.url_scheme', 'wsgi.version']) + 'wsgi.input_terminated', 'wsgi.multiprocess', 'wsgi.multithread', + 'wsgi.run_once', 'wsgi.url_scheme', 'wsgi.version']) self.assertEqual(environ['REQUEST_METHOD'], 'GET') self.assertEqual(environ['SERVER_PORT'], '80') @@ -727,6 +727,7 @@ class TestWSGITask(unittest.TestCase): self.assertEqual(environ['wsgi.multiprocess'], False) self.assertEqual(environ['wsgi.run_once'], False) self.assertEqual(environ['wsgi.input'], 'stream') + self.assertEqual(environ['wsgi.input_terminated'], True) self.assertEqual(inst.environ, environ) def test_get_environment_values_w_scheme_override_untrusted(self): @@ -767,8 +768,8 @@ class TestWSGITask(unittest.TestCase): 'PATH_INFO', 'QUERY_STRING', 'REMOTE_ADDR', 'REQUEST_METHOD', 'SCRIPT_NAME', 'SERVER_NAME', 'SERVER_PORT', 'SERVER_PROTOCOL', 'SERVER_SOFTWARE', 'wsgi.errors', 'wsgi.file_wrapper', 'wsgi.input', - 'wsgi.multiprocess', 'wsgi.multithread', 'wsgi.run_once', - 'wsgi.url_scheme', 'wsgi.version']) + 'wsgi.input_terminated', 'wsgi.multiprocess', 'wsgi.multithread', + 'wsgi.run_once', 'wsgi.url_scheme', 'wsgi.version']) self.assertEqual(environ['REQUEST_METHOD'], 'GET') self.assertEqual(environ['SERVER_PORT'], '80') @@ -790,6 +791,7 @@ class TestWSGITask(unittest.TestCase): self.assertEqual(environ['wsgi.multiprocess'], False) self.assertEqual(environ['wsgi.run_once'], False) self.assertEqual(environ['wsgi.input'], 'stream') + self.assertEqual(environ['wsgi.input_terminated'], True) self.assertEqual(inst.environ, environ) def test_get_environment_values_w_bogus_scheme_override(self): -- cgit v1.2.1