Skip to content

Commit 35a6235

Browse files
CoolCat467A5rocks
andcommitted
Avoid a few type errors with fake socket listener
Co-authored-by: A5rocks <[email protected]>
1 parent 96e437f commit 35a6235

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tests/test_connection.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -296,11 +296,15 @@ async def test_serve_with_zero_listeners() -> None:
296296
WebSocketServer(echo_request_handler, [])
297297

298298

299+
def memory_listener() -> trio.SocketListener:
300+
return MemoryListener() # type: ignore[return-value]
301+
302+
299303
async def test_serve_non_tcp_listener(nursery: trio.Nursery) -> None:
300-
listeners = [MemoryListener()]
304+
listeners = [memory_listener()]
301305
server = WebSocketServer(
302306
echo_request_handler,
303-
listeners, # type: ignore[arg-type]
307+
listeners,
304308
)
305309
await nursery.start(server.run)
306310
assert len(server.listeners) == 1
@@ -313,11 +317,11 @@ async def test_serve_non_tcp_listener(nursery: trio.Nursery) -> None:
313317

314318
async def test_serve_multiple_listeners(nursery: trio.Nursery) -> None:
315319
listener1 = (await trio.open_tcp_listeners(0, host=HOST))[0]
316-
listener2 = MemoryListener()
320+
listener2 = memory_listener()
317321
server = WebSocketServer(
318322
echo_request_handler, [
319323
listener1,
320-
listener2, # type: ignore[list-item]
324+
listener2,
321325
]
322326
)
323327
await nursery.start(server.run)

0 commit comments

Comments
 (0)