Merge pull request #1924 from python-trio/token-weakref

Author: altendky

newsfragments/1924.bugfix.rst

@@ -0,0 +1 @@
+The :class:`~.lowlevel.TrioToken` class can now be used as a target of a weak reference.

trio/_core/_entry_queue.py

@@ -145,7 +145,7 @@ class TrioToken(metaclass=NoPublicConstructor):
 
     """
 
-    __slots__ = ("_reentry_queue",)
+    __slots__ = ("_reentry_queue", "__weakref__")
 
     def __init__(self, reentry_queue):
         self._reentry_queue = reentry_queue

trio/tests/test_threads.py

@@ -1,8 +1,10 @@
 import threading
 import queue as stdlib_queue
 import time
+import weakref
 
 import pytest
+from trio._core import TrioToken, current_trio_token
 
 from .. import _core
 from .. import Event, CapacityLimiter, sleep
@@ -576,3 +578,10 @@ async def main():
 
     _core.run(main)
     assert record == ["ok"]
+
+
+async def test_trio_token_weak_referenceable():
+    token = current_trio_token()
+    assert isinstance(token, TrioToken)
+    weak_reference = weakref.ref(token)
+    assert token is weak_reference()