ez-a-sync 0.24.12__tar.gz → 0.24.13__tar.gz

{ez_a_sync-0.24.12 → ez_a_sync-0.24.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ez-a-sync
-Version: 0.24.12
+Version: 0.24.13
 Summary: A library that makes it easy to define objects that can be used for both sync and async use cases.
 Home-page: https://github.com/BobTheBuidler/a-sync
 Author: BobTheBuidler

{ez_a_sync-0.24.12 → ez_a_sync-0.24.13}/a_sync/_smart.py

@@ -6,7 +6,6 @@ to protect tasks from cancellation.
 """
 
 import asyncio
-import contextvars
 import logging
 import warnings
 import weakref
@@ -198,9 +197,7 @@ class SmartFuture(_SmartFutureMixin[T], asyncio.Future):
         if key:
             self._key = key
         self._waiters = weakref.WeakSet()
-        self._callbacks.append(
-            (SmartFuture._self_done_cleanup_callback, contextvars.copy_context())
-        )
+        self.add_done_callback(SmartFuture._self_done_cleanup_callback)
 
     def __repr__(self):
         return f"<{type(self).__name__} key={self._key} waiters={self.num_waiters} {self._state}>"
@@ -301,9 +298,7 @@ class SmartTask(_SmartFutureMixin[T], asyncio.Task):
         """
         asyncio.Task.__init__(self, coro, loop=loop, name=name)
         self._waiters: Set["asyncio.Task[T]"] = set()
-        self._callbacks.append(
-            (SmartTask._self_done_cleanup_callback, contextvars.copy_context())
-        )
+        self.add_done_callback(SmartTask._self_done_cleanup_callback)
 
 
 def smart_task_factory(

{ez_a_sync-0.24.12 → ez_a_sync-0.24.13}/ez_a_sync.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ez-a-sync
-Version: 0.24.12
+Version: 0.24.13
 Summary: A library that makes it easy to define objects that can be used for both sync and async use cases.
 Home-page: https://github.com/BobTheBuidler/a-sync
 Author: BobTheBuidler

{ez_a_sync-0.24.12 → ez_a_sync-0.24.13}/ez_a_sync.egg-info/SOURCES.txt

@@ -147,6 +147,7 @@ tests/test_executor.py
 tests/test_future.py
 tests/test_helpers.py
 tests/test_iter.py
+tests/test_smart.py
 tests/a_sync/test_abstract.py
 tests/a_sync/test_base.py
 tests/a_sync/test_cache.py

ez_a_sync-0.24.13/tests/test_smart.py

@@ -0,0 +1,15 @@
+import asyncio
+import pytest
+
+from a_sync._smart import SmartTask
+
+
+@pytest.mark.asyncio_cooperative
+async def test_smart_task_await():
+    await SmartTask(asyncio.sleep(0.1), loop=None)
+
+
+@pytest.mark.asyncio_cooperative
+async def test_smart_task_name():
+    t = SmartTask(asyncio.sleep(0.1), loop=None, name="test")
+    assert t.get_name() == "test"