From a13145d1d6f86fca0fa6d9817deb452cebe0edd2 Mon Sep 17 00:00:00 2001
From: cytsoi <154662279+tcyjohn@users.noreply.github.com>
Date: Thu, 22 Oct 2026 14:25:47 +0800
Subject: [PATCH] test: catch asyncio drain timeouts on Python 3.10

---
 tests/test_serving_protocol.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_serving_protocol.py b/tests/test_serving_protocol.py
index eaa805d..1363f91 100644
--- a/tests/test_serving_protocol.py
+++ b/tests/test_serving_protocol.py
@@ -85,7 +85,7 @@ def test_positive_drain_timeout_reopens_before_any_mutation():
     async def run():
         state = ServingState(1)
         await state.enter()
-        with pytest.raises(TimeoutError):
+        with pytest.raises(asyncio.TimeoutError):
             await state.begin_reload(.001)
         assert not state.closed and state.failed is None and state.server_generation == 0
         await state.leave()