Fix aiohttp test server teardown (#3726)

strawberry-graphql · Dec 14, 2024 · a0573da · a0573da
1 parent 7a42b03
commit a0573da
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/http/clients/aiohttp.py b/tests/http/clients/aiohttp.py
@@ -172,11 +172,9 @@ async def ws_connect(
         *,
         protocols: List[str],
     ) -> AsyncGenerator[WebSocketClient, None]:
-        server = TestServer(self.app)
-        await server.start_server()
-        client = TestClient(server)
-        async with client.ws_connect(url, protocols=protocols) as ws:
-            yield AioWebSocketClient(ws)
+        async with TestClient(TestServer(self.app)) as client:
+            async with client.ws_connect(url, protocols=protocols) as ws:
+                yield AioWebSocketClient(ws)
 
 
 class AioWebSocketClient(WebSocketClient):