test: isolate dispatch tests from quota state

Author: Trecek

tests/server/_helpers.py

@@ -31,6 +31,18 @@ async def _noop_quota_refresher(config: Any, **kwargs) -> None:
     """Quota refresher stub: no-op."""
 
 
+def _patch_dispatch_quota_no_sleep(monkeypatch: Any) -> None:
+    """Patch dispatch_food_truck's quota dependencies for non-quota tests."""
+    monkeypatch.setattr(
+        "autoskillit.server._misc.check_and_sleep_if_needed",
+        _no_sleep_quota_checker,
+    )
+    monkeypatch.setattr(
+        "autoskillit.server._misc._refresh_quota_cache",
+        _noop_quota_refresher,
+    )
+
+
 def _make_recipe_info(name: str = "test-recipe"):
     return _fleet_make_recipe_info(name, path_prefix="/fake/recipes/")
 

tests/server/test_tools_dispatch_halt.py

@@ -9,7 +9,11 @@
 
 from autoskillit.fleet import FleetSemaphore
 from tests.fakes import InMemoryHeadlessExecutor, InMemoryRecipeRepository
-from tests.server._helpers import _make_recipe_info, _make_standard_recipe
+from tests.server._helpers import (
+    _make_recipe_info,
+    _make_standard_recipe,
+    _patch_dispatch_quota_no_sleep,
+)
 
 pytestmark = [pytest.mark.layer("server"), pytest.mark.medium, pytest.mark.feature("fleet")]
 
@@ -195,6 +199,7 @@ async def test_no_result_block_failure_does_not_halt_campaign(
 
     def _setup_standard_dispatch(self, tool_ctx, monkeypatch):
         """Wire tool_ctx for a successful standard dispatch."""
+        _patch_dispatch_quota_no_sleep(monkeypatch)
         tool_ctx.fleet_lock = FleetSemaphore(max_concurrent=1)
         repo = InMemoryRecipeRepository()
         recipe_info = _make_recipe_info("test-recipe")
@@ -246,6 +251,7 @@ async def test_dispatch_food_truck_halts_after_failure_when_continue_on_failure_
 
     def _setup_standard_dispatch(self, tool_ctx, monkeypatch):
         """Wire tool_ctx for a standard dispatch."""
+        _patch_dispatch_quota_no_sleep(monkeypatch)
         tool_ctx.fleet_lock = FleetSemaphore(max_concurrent=1)
         repo = InMemoryRecipeRepository()
         recipe_info = _make_recipe_info("test-recipe")
@@ -329,6 +335,7 @@ async def test_dispatch_halts_when_no_dispatch_name_provided(
 
     def _setup_standard_dispatch(self, tool_ctx, monkeypatch):
         """Wire tool_ctx for a successful standard dispatch."""
+        _patch_dispatch_quota_no_sleep(monkeypatch)
         tool_ctx.fleet_lock = FleetSemaphore(max_concurrent=1)
         repo = InMemoryRecipeRepository()
         recipe_info = _make_recipe_info("test-recipe")
@@ -436,6 +443,7 @@ async def test_skip_when_resolves_campaign_ref(
 
     def _setup_standard_dispatch(self, tool_ctx, monkeypatch):
         """Wire tool_ctx for a successful standard dispatch."""
+        _patch_dispatch_quota_no_sleep(monkeypatch)
         tool_ctx.fleet_lock = FleetSemaphore(max_concurrent=1)
         repo = InMemoryRecipeRepository()
         recipe_info = _make_recipe_info("test-recipe")

tests/server/test_tools_dispatch_params.py

@@ -12,6 +12,7 @@
     _make_standard_recipe,
     _no_sleep_quota_checker,
     _noop_quota_refresher,
+    _patch_dispatch_quota_no_sleep,
     _simple_prompt_builder,
 )
 
@@ -26,6 +27,7 @@ async def test_dispatch_food_truck_tool_passes_resume_session_id_to_executor(
     from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
     monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
+    _patch_dispatch_quota_no_sleep(monkeypatch)
     tool_ctx_kitchen_open.fleet_lock = FleetSemaphore(max_concurrent=1)
     repo = InMemoryRecipeRepository()
     recipe_info = _make_recipe_info("test-recipe")
@@ -59,6 +61,7 @@ async def test_dispatch_food_truck_tool_passes_resume_message_to_executor(
     from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
     monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
+    _patch_dispatch_quota_no_sleep(monkeypatch)
     tool_ctx_kitchen_open.fleet_lock = FleetSemaphore(max_concurrent=1)
     repo = InMemoryRecipeRepository()
     recipe_info = _make_recipe_info("test-recipe")
@@ -93,6 +96,7 @@ async def test_dispatch_food_truck_tool_passes_caller_instructions_into_prompt(
     from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
     monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
+    _patch_dispatch_quota_no_sleep(monkeypatch)
     tool_ctx_kitchen_open.fleet_lock = FleetSemaphore(max_concurrent=1)
     repo = InMemoryRecipeRepository()
     recipe_info = _make_recipe_info("test-recipe")
@@ -121,6 +125,7 @@ async def test_dispatch_food_truck_no_caller_instructions_section_when_not_speci
     from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
     monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
+    _patch_dispatch_quota_no_sleep(monkeypatch)
     tool_ctx_kitchen_open.fleet_lock = FleetSemaphore(max_concurrent=1)
     repo = InMemoryRecipeRepository()
     recipe_info = _make_recipe_info("test-recipe")
@@ -146,8 +151,9 @@ class TestDispatchFoodTruckIdleTimeout:
     def _set_fleet_session(self, monkeypatch):
         monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
 
-    def _setup_dispatch(self, tool_ctx):
+    def _setup_dispatch(self, tool_ctx, monkeypatch):
         """Wire tool_ctx for a standard dispatch with InMemoryHeadlessExecutor."""
+        _patch_dispatch_quota_no_sleep(monkeypatch)
         tool_ctx.fleet_lock = FleetSemaphore(max_concurrent=1)
         repo = InMemoryRecipeRepository()
         recipe_info = _make_recipe_info("test-recipe")
@@ -163,7 +169,7 @@ async def test_dispatch_food_truck_passes_idle_output_timeout_to_executor(
         """dispatch_food_truck MCP tool forwards idle_output_timeout to executor."""
         from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
-        self._setup_dispatch(tool_ctx_kitchen_open)
+        self._setup_dispatch(tool_ctx_kitchen_open, monkeypatch)
 
         await dispatch_food_truck(
             recipe="test-recipe",
@@ -182,7 +188,7 @@ async def test_dispatch_food_truck_idle_timeout_none_when_not_specified(
         """When idle_output_timeout is not specified, executor receives None."""
         from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
-        self._setup_dispatch(tool_ctx_kitchen_open)
+        self._setup_dispatch(tool_ctx_kitchen_open, monkeypatch)
 
         await dispatch_food_truck(
             recipe="test-recipe",
@@ -200,7 +206,7 @@ async def test_dispatch_food_truck_idle_timeout_overrides_config_default(
         """Explicit idle_output_timeout=0 overrides the config default of 1000."""
         from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
-        self._setup_dispatch(tool_ctx_kitchen_open)
+        self._setup_dispatch(tool_ctx_kitchen_open, monkeypatch)
         # Config idle_output_timeout is 1000 (default from RunSkillConfig)
         assert tool_ctx_kitchen_open.config.run_skill.idle_output_timeout == 1000
 
@@ -222,7 +228,7 @@ async def test_execute_dispatch_passes_idle_output_timeout_to_executor(
         """execute_dispatch forwards idle_output_timeout to executor.dispatch_food_truck."""
         from autoskillit.fleet._api import execute_dispatch
 
-        self._setup_dispatch(tool_ctx)
+        self._setup_dispatch(tool_ctx, monkeypatch)
 
         await execute_dispatch(
             tool_ctx=tool_ctx,
@@ -252,6 +258,7 @@ async def test_dispatch_food_truck_returns_fleet_error_on_mcp_timeout(
     from autoskillit.server.tools.tools_fleet_dispatch import dispatch_food_truck
 
     monkeypatch.setenv("AUTOSKILLIT_SESSION_TYPE", "fleet")
+    _patch_dispatch_quota_no_sleep(monkeypatch)
     tool_ctx_kitchen_open.fleet_lock = FleetSemaphore(max_concurrent=1)
     repo = InMemoryRecipeRepository()
     recipe_info = _make_recipe_info("test-recipe")