From a3cba0d0ae06ef33c002d1d424564c16aef8ab61 Mon Sep 17 00:00:00 2001 From: Marc Abramowitz Date: Tue, 28 Apr 2015 09:57:32 -0700 Subject: tests/test_httpserver.py: Add test_environ_with_multiple_values Tests when there are two `Host:` headers present. --- tests/test_httpserver.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/test_httpserver.py b/tests/test_httpserver.py index a719788..3d72c79 100644 --- a/tests/test_httpserver.py +++ b/tests/test_httpserver.py @@ -27,3 +27,19 @@ def test_environ(): wsgi_handler.wsgi_setup() assert wsgi_handler.wsgi_environ['HTTP_HOST'] == 'mywebsite' + + +def test_environ_with_multiple_values(): + mock_socket = MockSocket() + mock_client_address = '1.2.3.4' + mock_server = MockServer() + + wsgi_handler = WSGIHandler(mock_socket, mock_client_address, mock_server) + wsgi_handler.command = 'GET' + wsgi_handler.path = '/path' + wsgi_handler.request_version = 'HTTP/1.0' + wsgi_handler.headers = email.message_from_string('Host: host1\nHost: host2') + + wsgi_handler.wsgi_setup() + + assert wsgi_handler.wsgi_environ['HTTP_HOST'] == 'host1,host2' -- cgit v1.2.1