[CHIA-1359] Fix action scope callbacks across multiple .use calls (#…

…18572) Fix action scope callbacks across multiple `.use` calls
Chia-Network · Sep 12, 2024 · ab21776 · ab21776
1 parent 810f0bf
commit ab21776
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/chia/_tests/util/test_action_scope.py b/chia/_tests/util/test_action_scope.py
@@ -90,6 +90,9 @@ async def callback(interface: StateInterface[TestSideEffects]) -> None:
 
             interface.set_callback(callback)
 
+        async with action_scope.use():
+            pass  # Testing that callback stays put even through another .use()
+
     assert action_scope.side_effects.buf == b"bar"
 
 

diff --git a/chia/util/action_scope.py b/chia/util/action_scope.py
@@ -143,7 +143,7 @@ async def new_scope(
     async def use(self, _callbacks_allowed: bool = True) -> AsyncIterator[StateInterface[_T_SideEffects]]:
         async with self._resource_manager.use():
             side_effects = await self._resource_manager.get_resource(self._side_effects_format)
-            interface = StateInterface(side_effects, _callbacks_allowed)
+            interface = StateInterface(side_effects, _callbacks_allowed, self._callback)
 
             yield interface