Merge pull request #2 from minrk/not-all-weakrefs

Make sure it works with uvloop

Author: minrk

asyncio_atexit.py

@@ -21,7 +21,13 @@
 
 class _RegistryEntry:
     def __init__(self, loop):
-        self._close_ref = weakref.WeakMethod(loop.close)
+        try:
+            self._close_ref = weakref.WeakMethod(loop.close)
+        except TypeError:
+            # not everything can be weakref'd (Extensions such as uvloop).
+            # Hold a regular reference _on the object_, in those cases
+            loop._atexit_orig_close = loop.close
+            self._close_ref = lambda: loop._atexit_orig_close
         self.callbacks = []
 
     def close(self):

setup.cfg

@@ -20,7 +20,9 @@ py_modules = asyncio_atexit
 python_requires = >=3.6
 
 [options.extras_require]
-test = pytest
+test =
+    pytest
+    uvloop
 
 [tool:pytest]
 # asyncio_mode = auto

test_asyncio_atexit.py

@@ -1,6 +1,13 @@
 import asyncio
 import sys
 
+try:
+    import uvloop
+except ImportError:
+    uvloop = None
+
+import pytest
+
 import asyncio_atexit
 
 if sys.version_info >= (3, 7):
@@ -15,7 +22,24 @@ def asyncio_run(coro):
             loop.close()
 
 
-def test_asyncio_atexit():
+policies = ["default"]
+if uvloop is not None:
+    policies.append("uvloop")
+
+
+@pytest.fixture(params=policies)
+def policy(request):
+    before_policy = asyncio.get_event_loop_policy()
+    if request.param == "default":
+        policy = asyncio.DefaultEventLoopPolicy()
+    elif request.param == "uvloop":
+        policy = uvloop.EventLoopPolicy()
+    asyncio.set_event_loop_policy(policy)
+    yield
+    asyncio.set_event_loop_policy(before_policy)
+
+
+def test_asyncio_atexit(policy):
     sync_called = False
     async_called = False
 
@@ -37,7 +61,7 @@ async def test():
     assert async_called
 
 
-def test_unregister():
+def test_unregister(policy):
     sync_called = False
 
     def sync_cb():