WIP Remove pytest asyncio pinning

conftest.py

@@ -32,3 +32,30 @@ def pytest_collection_modifyitems(config, items):
 
 
 pytest_plugins = ["distributed.pytest_resourceleaks"]
+
+# The following fixtures are required for pytest-asyncio compatibility
+
+
+@pytest.fixture()
+def event_loop():
+    from distributed.utils_test import clean
+
+    with clean() as loop:
+        yield loop.asyncio_loop
+
+
+@pytest.hookimpl(trylast=True)
+def pytest_fixture_post_finalizer(fixturedef, request):
+    """Pytest-asyncio resets the eventloop policy after the above event loop is closed."""
+    import asyncio
+
+    if fixturedef.argname == "event_loop":
+        try:
+            # Tornado > 6.0.3
+            from tornado.platform.asyncio import AnyThreadEventLoopPolicy
+        except ImportError:
+            try:
+                from distributed.utils import AnyThreadEventLoopPolicy
+            except ImportError:
+                return
+        asyncio.set_event_loop_policy(AnyThreadEventLoopPolicy())

continuous_integration/environment-3.7.yaml

@@ -24,7 +24,7 @@ dependencies:
   - prometheus_client
   - psutil
   - pytest
-  - pytest-asyncio<0.14.0
+  - pytest-asyncio
   - pytest-faulthandler
   - pytest-repeat
   - pytest-rerunfailures

continuous_integration/environment-3.8.yaml

@@ -22,7 +22,7 @@ dependencies:
   - prometheus_client
   - psutil
   - pytest
-  - pytest-asyncio<0.14.0
+  - pytest-asyncio
   - pytest-faulthandler
   - pytest-repeat
   - pytest-rerunfailures

continuous_integration/environment-3.9.yaml

@@ -24,7 +24,7 @@ dependencies:
   - prometheus_client
   - psutil
   - pytest
-  - pytest-asyncio<0.14.0
+  - pytest-asyncio
   - pytest-faulthandler
   - pytest-repeat
   - pytest-rerunfailures