test(body_cap): assert content_length on the streamed-with-declared-length path

Author: lesnik512

tests/test_body_cap.py

@@ -125,6 +125,7 @@ def handler(request: httpx2.Request) -> httpx2.Response:  # noqa: ARG001
         with pytest.raises(ResponseTooLargeError) as caught:
             _read_capped(resp, 1000, resp.request)
         assert caught.value.reason == "streamed"  # compressed CL (small) passed; decoded tripped
+        assert caught.value.content_length == len(raw)  # the declared (compressed) length, threaded through
     finally:
         resp.close()
         client.close()
@@ -192,6 +193,23 @@ def handler(request: httpx2.Request) -> httpx2.Response:  # noqa: ARG001
         await client.aclose()
 
 
+async def test_read_capped_async_gzip_bomb_trips_on_decoded_bytes() -> None:
+    raw = gzip.compress(b"A" * 100_000)
+
+    def handler(request: httpx2.Request) -> httpx2.Response:  # noqa: ARG001
+        return httpx2.Response(200, headers={"content-encoding": "gzip"}, content=raw)
+
+    client, resp = await _async_stream(handler)
+    try:
+        with pytest.raises(ResponseTooLargeError) as caught:
+            await _read_capped_async(resp, 1000, resp.request)
+        assert caught.value.reason == "streamed"  # compressed CL (small) passed; decoded tripped
+        assert caught.value.content_length == len(raw)  # the declared (compressed) length, threaded through
+    finally:
+        await resp.aclose()
+        await client.aclose()
+
+
 async def test_read_capped_async_head_with_large_declared_length_not_rejected() -> None:
     def handler(request: httpx2.Request) -> httpx2.Response:  # noqa: ARG001
         return httpx2.Response(200, headers={"content-length": "50000000"})