async def test_body_stream_single_chunk() -> None:
body = Body(None, None)
body.append(b"data")
body.set_complete()
async def _check_data() -> None:
async for data in body:
assert data == b"data"
await asyncio.wait_for(_check_data(), 1)
def test_body_exceeds_max_content_length() -> None:
max_content_length = 5
body = Body(max_content_length)
with pytest.raises(RequestEntityTooLarge):
body.append(b' ' * (max_content_length + 1))
async def _fill_body(body: Body, semaphore: asyncio.Semaphore,
limit: int) -> None:
for number in range(limit):
body.append(b"%d" % number)
await semaphore.acquire()
body.set_complete()
async def test_body_exceeds_max_content_length() -> None:
max_content_length = 5
body = Body(None, max_content_length)
body.append(b" " * (max_content_length + 1))
with pytest.raises(RequestEntityTooLarge):
await body