More testing of the response object.

Author: Lukasa

hyper/http11/response.py

@@ -42,7 +42,7 @@ def __init__(self, code, reason, headers, sock):
         # bother checking for content-length, we just keep reading until
         # we no longer can.
         self._expect_close = False
-        if b'close' in self.headers.get('connection', []):
+        if b'close' in self.headers.get(b'connection', []):
             self._expect_close = True
 
         # The expected length of the body.

test/test_http11.py

@@ -15,6 +15,7 @@
 import hyper
 from hyper.http11.connection import HTTP11Connection
 from hyper.http11.response import HTTP11Response
+from hyper.http20.exceptions import ConnectionResetError
 from hyper.common.headers import HTTPHeaderMap
 from hyper.compat import bytes
 
@@ -320,6 +321,28 @@ def test_short_circuit_read(self):
 
         assert r.read() == b''
 
+    def test_aborted_reads(self):
+        d = DummySocket()
+        r = HTTP11Response(200, 'OK', {b'content-length': [b'15']}, d)
+
+        with pytest.raises(ConnectionResetError):
+            r.read()
+
+    def test_read_expect_close(self):
+        d = DummySocket()
+        r = HTTP11Response(200, 'OK', {b'connection': [b'close']}, d)
+
+        assert r.read() == b''
+
+    def test_response_as_context_manager(self):
+        r = HTTP11Response(
+            200, 'OK', {b'content-length': [b'0']}, DummySocket()
+        )
+
+        with r:
+            assert r.read() == b''
+
+        assert r._sock == None
 
 class DummySocket(object):
     def __init__(self):