didericis-claude
5fcca2060f
Remove auto-migration from DbStore.__init__; add explicit check_migrations() and migrate() methods. The CLI now checks both stores on every startup and prompts the user to confirm before migrating. supervise.prepare() calls .migrate() directly now that __init__ no longer does it implicitly. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
"""Unit: top-level CLI dispatch in bot_bottle.cli.main (ADR 0004).
|
|
|
|
`cli/__init__.py` is dispatch + exit-code mapping, not interactive I/O,
|
|
so it carries real unit tests rather than being omitted like the
|
|
`cli/init` / `cli/tui` shells."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
import bot_bottle.cli as climod
|
|
from bot_bottle.cli import main
|
|
from bot_bottle.db_store import DbStore
|
|
from bot_bottle.log import Die
|
|
from bot_bottle.manifest import ManifestError
|
|
|
|
|
|
class TestMainDispatch(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
patcher = patch.object(DbStore, "check_migrations", return_value=True)
|
|
self._mock_check = patcher.start()
|
|
self.addCleanup(patcher.stop)
|
|
|
|
def test_no_args_prints_usage_returns_2(self) -> None:
|
|
with patch("sys.stderr", io.StringIO()):
|
|
self.assertEqual(2, main([]))
|
|
|
|
def test_help_flags_return_0(self) -> None:
|
|
with patch("sys.stderr", io.StringIO()):
|
|
self.assertEqual(0, main(["-h"]))
|
|
self.assertEqual(0, main(["--help"]))
|
|
|
|
def test_unknown_command_dies(self) -> None:
|
|
with patch("sys.stderr", io.StringIO()):
|
|
with self.assertRaises(Die):
|
|
main(["definitely-not-a-command"])
|
|
|
|
def test_handler_return_code_passthrough(self) -> None:
|
|
def handler(_rest: list[str]) -> int:
|
|
return 7
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": handler}):
|
|
self.assertEqual(7, main(["x"]))
|
|
|
|
def test_handler_none_return_becomes_0(self) -> None:
|
|
def handler(_rest: list[str]) -> int | None:
|
|
return None
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": handler}):
|
|
self.assertEqual(0, main(["x"]))
|
|
|
|
def test_args_forwarded_to_handler(self) -> None:
|
|
seen: list[list[str]] = []
|
|
|
|
def handler(rest: list[str]) -> int:
|
|
seen.append(rest)
|
|
return 0
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": handler}):
|
|
main(["x", "a", "b"])
|
|
self.assertEqual([["a", "b"]], seen)
|
|
|
|
def test_manifest_error_maps_to_1(self) -> None:
|
|
def boom(_rest: list[str]) -> int:
|
|
raise ManifestError("bad manifest")
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": boom}), patch("sys.stderr", io.StringIO()):
|
|
self.assertEqual(1, main(["x"]))
|
|
|
|
def test_die_maps_to_its_code(self) -> None:
|
|
def boom(_rest: list[str]) -> int:
|
|
raise Die(3)
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": boom}):
|
|
self.assertEqual(3, main(["x"]))
|
|
|
|
def test_keyboard_interrupt_maps_to_130(self) -> None:
|
|
def boom(_rest: list[str]) -> int:
|
|
raise KeyboardInterrupt()
|
|
|
|
with patch.dict(climod.COMMANDS, {"x": boom}):
|
|
self.assertEqual(130, main(["x"]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|