def test_preserve_chunked_on_broken_connection(self, monkeypatch):
self.chunked_requests = 0
def socket_handler(listener):
for i in range(2):
sock = listener.accept()[0]
request = ConnectionMarker.consume_request(sock)
if b"Transfer-Encoding: chunked" in request.split(b"\r\n"):
self.chunked_requests += 1
if i == 0:
# Bad HTTP version will trigger a connection close
sock.sendall(b"HTTP/0.5 200 OK\r\n\r\n")
else:
sock.sendall(b"HTTP/1.1 200 OK\r\n\r\n")
sock.close()
self._start_server(socket_handler)
with ConnectionMarker.mark(monkeypatch):
with HTTPConnectionPool(self.host, self.port) as pool:
retries = Retry(read=1)
pool.urlopen("GET",
"/",
chunked=True,
preload_content=False,
retries=retries)
assert self.chunked_requests == 2
def test_preserve_chunked_on_redirect(
self, monkeypatch: pytest.MonkeyPatch) -> None:
self.chunked_requests = 0
def socket_handler(listener: socket.socket) -> None:
for i in range(2):
sock = listener.accept()[0]
request = ConnectionMarker.consume_request(sock)
if b"Transfer-Encoding: chunked" in request.split(b"\r\n"):
self.chunked_requests += 1
if i == 0:
sock.sendall(b"HTTP/1.1 301 Moved Permanently\r\n"
b"Location: /redirect\r\n\r\n")
else:
sock.sendall(b"HTTP/1.1 200 OK\r\n\r\n")
sock.close()
self._start_server(socket_handler)
with ConnectionMarker.mark(monkeypatch):
with HTTPConnectionPool(self.host, self.port) as pool:
retries = Retry(redirect=1)
pool.urlopen("GET",
"/",
chunked=True,
preload_content=False,
retries=retries)
assert self.chunked_requests == 2