Test HTTP11Connection is context manager.

Author: Lukasa

test/test_integration_http11.py

@@ -165,3 +165,52 @@ def socket_handler(listener):
         assert r.headers[b'content-length'] == [b'15']
 
         assert r.read() == b'hellotherehello'
+
+    def test_connection_context_manager(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 200 OK\r\n'
+                b'Server: socket-level-server\r\n'
+                b'Content-Length: 15\r\n'
+                b'\r\n'
+            )
+            sock.send(resp)
+
+            chunks = [
+                b'hello',
+                b'there',
+                b'hello',
+            ]
+
+            for chunk in chunks:
+                sock.send(chunk)
+
+            sock.close()
+
+        self._start_server(socket_handler)
+        with self.get_connection() as c:
+            c.request('GET', '/')
+            send_event.set()
+            r = c.get_response()
+            data = r.read()
+
+        assert r.status == 200
+        assert r.reason == b'OK'
+        assert len(r.headers) == 2
+        assert data == b'hellotherehello'
+
+        assert c._sock is None