tweaked task cleanup on stop to try to fix pytest error

Author: craigbarratt

custom_components/pyscript/__init__.py

@@ -175,12 +175,10 @@ async def start_triggers(event):
 
     async def stop_triggers(event):
         _LOGGER.debug("stopping triggers")
-        for global_ctx_name, global_ctx in GlobalContextMgr.items():
-            if not global_ctx_name.startswith("file."):
-                continue
+        for _, global_ctx in GlobalContextMgr.items():
             global_ctx.stop()
         # tell reaper task to exit (after other tasks are cancelled)
-        Function.task_cancel(None)
+        await Function.reaper_stop()
 
     hass.bus.async_listen(EVENT_HOMEASSISTANT_STARTED, start_triggers)
     hass.bus.async_listen(EVENT_HOMEASSISTANT_STOP, stop_triggers)

custom_components/pyscript/function.py

@@ -92,6 +92,12 @@ async def task_cancel_reaper(reaper_q):
         cls.task_reaper_q = asyncio.Queue(0)
         cls.task_cancel_repeaer = Function.create_task(task_cancel_reaper(cls.task_reaper_q))
 
+    @classmethod
+    async def reaper_stop(cls):
+        """Tell the reaper task to exit by sending a special task None."""
+        cls.task_cancel(None)
+        await cls.task_cancel_repeaer
+
     @classmethod
     async def async_sleep(cls, duration):
         """Implement task.sleep()."""