Start of http11 upgrade test

Author: fredthomsen

test/server.py

@@ -39,24 +39,29 @@ def __init__(self,
                  socket_handler,
                  host='localhost',
                  ready_event=None,
-                 h2=True):
+                 h2=True,
+                 secure=True):
         threading.Thread.__init__(self)
 
         self.socket_handler = socket_handler
         self.host = host
+        self.secure = secure
         self.ready_event = ready_event
 
-        self.cxt = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
-        if ssl.HAS_NPN and h2:
-            self.cxt.set_npn_protocols([NPN_PROTOCOL])
-        self.cxt.load_cert_chain(certfile='test/certs/server.crt',
-                                 keyfile='test/certs/server.key')
+        if self.secure:
+            self.cxt = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
+            if ssl.HAS_NPN and h2:
+                self.cxt.set_npn_protocols([NPN_PROTOCOL])
+            self.cxt.load_cert_chain(certfile='test/certs/server.crt',
+                                     keyfile='test/certs/server.key')
 
     def _start_server(self):
         sock = socket.socket(socket.AF_INET6)
         if sys.platform != 'win32':
             sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        sock = self.cxt.wrap_socket(sock, server_side=True)
+
+        if self.secure:
+            sock = self.cxt.wrap_socket(sock, server_side=True)
         sock.bind((self.host, 0))
         self.port = sock.getsockname()[1]
 
@@ -84,6 +89,7 @@ class SocketLevelTest(object):
     def set_up(self):
         self.host = None
         self.port = None
+        self.secure = None
         self.server_thread = None
 
     def _start_server(self, socket_handler):
@@ -95,17 +101,19 @@ def _start_server(self, socket_handler):
             socket_handler=socket_handler,
             ready_event=ready_event,
             h2=self.h2,
+            secure=self.secure
         )
         self.server_thread.start()
         ready_event.wait()
         self.host = self.server_thread.host
         self.port = self.server_thread.port
+        self.secure = self.server_thread.secure
 
     def get_connection(self):
         if self.h2:
-            return HTTP20Connection(self.host, self.port, secure=True)
+            return HTTP20Connection(self.host, self.port, self.secure)
         else:
-            return HTTP11Connection(self.host, self.port, secure=True)
+            return HTTP11Connection(self.host, self.port, self.secure)
 
     def get_encoder(self):
         """

test/test_integration_http11.py

@@ -214,3 +214,39 @@ def socket_handler(listener):
         assert data == b'hellotherehello'
 
         assert c._sock is None
+
+    def test_upgrade_response(self):
+        self.set_up()
+
+        send_event = threading.Event()
+
+        def socket_handler(listener):
+            sock = listener.accept()[0]
+
+            # We should get the initial request.
+            data = b''
+            while not data.endswith(b'\r\n\r\n'):
+                data += sock.recv(65535)
+
+            send_event.wait()
+
+            # We need to send back a response.
+            resp = (
+                b'HTTP/1.1 101 Upgrade\r\n'
+                b'Server: socket-level-server\r\n'
+                b'Content-Length: 0\r\n'
+                b'Connection: upgrade\r\n'
+                b'Upgrade: h2c\r\n'
+                b'\r\n'
+            )
+            sock.send(resp)
+
+            sock.close()
+
+        self._start_server(socket_handler)
+        c = self.get_connection()
+        c.request('GET', '/')
+        send_event.set()
+
+        with pytest.raises(HTTPUpgrade):
+            r = c.get_response()