didericis-claude
18e610c7a8
runner.py: use 'from bot_bottle import api' (satisfies R0402) with type: ignore and pylint disable for the cross-branch dependency on bot_bottle.api (added in PR #318, which merges before this one). sidecar.py: add pylint disable for intentional broad-exception-caught. test_runner.py: annotate _make_api_stub(**overrides: object) -> Any and type stub variable as Any to allow attribute assignment without type: ignore per-line. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
"""Unit: ProgrammaticBottleRunner + slugify."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
import unittest
|
|
from typing import Any
|
|
from unittest.mock import MagicMock
|
|
|
|
from bot_bottle.orchestrator.runner import ProgrammaticBottleRunner, slugify
|
|
|
|
|
|
class SlugifyTest(unittest.TestCase):
|
|
def test_basic(self):
|
|
self.assertEqual("impl-didericis-bot-bottle-17",
|
|
slugify("impl-didericis-bot-bottle-17"))
|
|
|
|
def test_collapses_and_strips(self):
|
|
self.assertEqual("a-b-c", slugify(" A_B/C!! "))
|
|
|
|
|
|
def _make_api_stub(**overrides: object) -> Any:
|
|
"""Return a mock bot_bottle.api module with sensible defaults."""
|
|
stub: Any = types.ModuleType("bot_bottle.api")
|
|
stub.start_headless = MagicMock(return_value="impl-r-17")
|
|
stub.freeze = MagicMock()
|
|
stub.resume_headless = MagicMock()
|
|
stub.destroy = MagicMock()
|
|
for k, v in overrides.items():
|
|
setattr(stub, k, v)
|
|
return stub
|
|
|
|
|
|
class ProgrammaticRunnerTest(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self._api: Any = _make_api_stub()
|
|
sys.modules["bot_bottle.api"] = self._api
|
|
self.runner = ProgrammaticBottleRunner()
|
|
|
|
def tearDown(self) -> None:
|
|
sys.modules.pop("bot_bottle.api", None)
|
|
|
|
def test_start_returns_slug_from_api(self) -> None:
|
|
slug = self.runner.start(
|
|
agent="impl", bottles=["claude", "dev"], label="impl-r-17",
|
|
prompt="do it", forge_env={"FORGE_OWNER": "didericis"},
|
|
)
|
|
self.assertEqual("impl-r-17", slug)
|
|
|
|
def test_start_forwards_all_args(self) -> None:
|
|
self.runner.start(
|
|
agent="impl", bottles=["claude", "dev"], label="impl-r-17",
|
|
prompt="do it", forge_env={"FORGE_OWNER": "didericis"},
|
|
)
|
|
self._api.start_headless.assert_called_once_with(
|
|
"impl",
|
|
prompt="do it",
|
|
bottles=["claude", "dev"],
|
|
label="impl-r-17",
|
|
forge_env={"FORGE_OWNER": "didericis"},
|
|
)
|
|
|
|
def test_start_no_bottles_passes_none(self) -> None:
|
|
self.runner.start(agent="impl", bottles=[], label="l", prompt="p", forge_env={})
|
|
call_kwargs = self._api.start_headless.call_args[1]
|
|
self.assertIsNone(call_kwargs["bottles"])
|
|
|
|
def test_freeze_delegates_to_api(self) -> None:
|
|
self.runner.freeze("slug-1")
|
|
self._api.freeze.assert_called_once_with("slug-1")
|
|
|
|
def test_freeze_returns_none(self) -> None:
|
|
result = self.runner.freeze("slug-1")
|
|
self.assertIsNone(result)
|
|
|
|
def test_resume_delegates_to_api(self) -> None:
|
|
self.runner.resume("slug-1", "address review")
|
|
self._api.resume_headless.assert_called_once_with("slug-1", prompt="address review")
|
|
|
|
def test_resume_returns_none(self) -> None:
|
|
result = self.runner.resume("slug-1", "p")
|
|
self.assertIsNone(result)
|
|
|
|
def test_destroy_delegates_to_api(self) -> None:
|
|
self.runner.destroy("slug-7")
|
|
self._api.destroy.assert_called_once_with("slug-7")
|
|
|
|
def test_destroy_returns_none(self) -> None:
|
|
result = self.runner.destroy("slug-7")
|
|
self.assertIsNone(result)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|