add extra assertion for deadlock prevention

richardsheridan · Jul 30, 2023 · 569af75 · 569af75
1 parent 9b38070
commit 569af75
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trio_parallel/_impl.py b/trio_parallel/_impl.py
@@ -185,6 +185,7 @@ async def _aclose(self):
         self._lifetime.waiting_task = trio.lowlevel.current_task()
         with trio.CancelScope(shield=True):
             if self._lifetime.calc_running() != 0:
+                assert self is not get_default_context()
                 await trio.sleep_forever()  # woken by self._lifetime.__aexit__
             await trio.to_thread.run_sync(
                 self._worker_cache.shutdown, self.grace_period