Add test for write timeout error (#67)
* Add test for write timeout error * Bump up write data 10x
This commit is contained in:
parent
3ed08857ee
commit
b8b3714d8e
@ -10,6 +10,7 @@ from .exceptions import (
|
||||
PoolTimeout,
|
||||
ProtocolError,
|
||||
ReadTimeout,
|
||||
WriteTimeout,
|
||||
RedirectBodyUnavailable,
|
||||
RedirectLoop,
|
||||
ResponseClosed,
|
||||
|
||||
@ -6,6 +6,7 @@ from httpcore import (
|
||||
PoolLimits,
|
||||
PoolTimeout,
|
||||
ReadTimeout,
|
||||
WriteTimeout,
|
||||
TimeoutConfig,
|
||||
)
|
||||
|
||||
@ -19,6 +20,16 @@ async def test_read_timeout(server):
|
||||
await client.get("http://127.0.0.1:8000/slow_response")
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_write_timeout(server):
|
||||
timeout = TimeoutConfig(write_timeout=0.0001)
|
||||
|
||||
async with AsyncClient(timeout=timeout) as client:
|
||||
with pytest.raises(WriteTimeout):
|
||||
data = b"*" * 1024 * 1024 * 10
|
||||
await client.put("http://127.0.0.1:8000/slow_response", data=data)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_connect_timeout(server):
|
||||
timeout = TimeoutConfig(connect_timeout=0.0001)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user