Throw upgrade exceptions from HTTP/1.1.

Author: Lukasa

hyper/http11/connection.py

@@ -12,6 +12,7 @@
 from .response import HTTP11Response
 from ..tls import wrap_socket
 from ..common.bufsocket import BufferedSocket
+from ..common.exceptions import TLSUpgrade
 from ..common.headers import HTTPHeaderMap
 from ..common.util import to_bytestring
 from ..compat import bytes
@@ -84,12 +85,17 @@ def connect(self):
         """
         if self._sock is None:
             sock = socket.create_connection((self.host, self.port), 5)
+            proto = None
 
             if self.secure:
                 sock, proto = wrap_socket(sock, self.host)
-                assert proto in ('http/1.1', None)
 
-            self._sock = BufferedSocket(sock, self.network_buffer_size)
+            sock = BufferedSocket(sock, self.network_buffer_size)
+
+            if proto not in ('http/1.1', None):
+                raise TLSUpgrade(proto, sock)
+
+            self._sock = sock
 
         return