ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-26 07:41:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Feature: 细化内置驱动器请求参数中的超时控制颗粒度 (#3571)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ju4tCode <42488585+yanyongyu@users.noreply.github.com>
This commit is contained in:
Ailitonia
2025-07-15 10:29:18 +08:00
committed by GitHub
parent bc0682af8f
commit 4ec9bfb7d7
8 changed files with 115 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_driver.py
View File

@ -16,6 +16,7 @@ from nonebot.drivers import (
HTTPServerSetup,
Request,
Response,
Timeout,
WebSocket,
WebSocketClientMixin,
WebSocketServerSetup,
@ -235,6 +236,7 @@ async def test_http_client(driver: Driver, server_url: URL):
headers={"X-Test": "test"},
cookies={"session": "test"},
content="test",
timeout=Timeout(total=4, connect=2, read=2),
)
response = await driver.request(request)
assert server_url.host is not None
@ -250,6 +252,7 @@ async def test_http_client(driver: Driver, server_url: URL):
headers={"X-Test": "test"},
cookies={"session": "test"},
content="test",
timeout=Timeout(total=4, connect=2, read=2),
)
assert request.url == request_raw_url.url, (
"request.url should be equal to request_raw_url.url"
@ -312,6 +315,7 @@ async def test_http_client(driver: Driver, server_url: URL):
headers={"X-Test": "stream"},
cookies={"session": "stream"},
content="stream_test" * 1024,
timeout=Timeout(total=4, connect=2, read=2),
)
chunks = []
async for resp in driver.stream_request(request, chunk_size=4):
@ -414,6 +418,7 @@ async def test_http_client_session(driver: Driver, server_url: URL):
headers={"X-Test": "test"},
cookies={"cookie": "test"},
content="test",
timeout=Timeout(total=4, connect=2, read=2),
)
response = await session.request(request)
assert response.status_code == 200
@ -499,6 +504,7 @@ async def test_http_client_session(driver: Driver, server_url: URL):
headers={"X-Test": "stream"},
cookies={"cookie": "stream"},
content="stream_test" * 1024,
timeout=Timeout(total=4, connect=2, read=2),
)
chunks = []
async for resp in session.stream_request(request, chunk_size=4):