Future-proof asyncio.wait usage.

Fix #762.

Author: DavidBord

.circleci/config.yml

@@ -38,6 +38,15 @@ jobs:
       - checkout
       - run: sudo pip install tox
       - run: tox -e py38
+  py39:
+    docker:
+      - image: circleci/python:3.9.0b1
+    steps:
+      # Remove IPv6 entry for localhost in Circle CI containers because it doesn't work anyway.
+      - run: sudo cp /etc/hosts /tmp; sudo sed -i '/::1/d' /tmp/hosts; sudo cp /tmp/hosts /etc
+      - checkout
+      - run: sudo pip install tox
+      - run: tox -e py39
 
 workflows:
   version: 2
@@ -53,3 +62,6 @@ workflows:
       - py38:
           requires:
             - main
+      - py39:
+          requires:
+            - main

.gitignore

@@ -1,6 +1,7 @@
 *.pyc
 *.so
 .coverage
+.idea/
 .mypy_cache
 .tox
 build/

src/websockets/server.py

@@ -714,7 +714,10 @@ async def _close(self) -> None:
         # asyncio.wait doesn't accept an empty first argument
         if self.websockets:
             await asyncio.wait(
-                [websocket.close(1001) for websocket in self.websockets],
+                [
+                    asyncio.ensure_future(websocket.close(1001))
+                    for websocket in self.websockets
+                ],
                 loop=self.loop if sys.version_info[:2] < (3, 8) else None,
             )
 

tox.ini

@@ -1,5 +1,5 @@
 [tox]
-envlist = py36,py37,py38,coverage,black,flake8,isort,mypy
+envlist = py36,py37,py38,py39,coverage,black,flake8,isort,mypy
 
 [testenv]
 commands = python -W default -m unittest {posargs}