forked from Rockachopa/Timmy-time-dashboard
982c42ba45
TIMMY_STATUS_PROMPT was defined in timmy/prompts.py and covered by tests, but never wired into the application. The CLI status command was passing a hardcoded inline string instead. Replace the inline string with the canonical prompt and add two CLI tests that verify the correct prompt is used. https://claude.ai/code/session_01DMjQ5qMZ8iHeyix1j3GS7c
30 lines
889 B
Python
30 lines
889 B
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from typer.testing import CliRunner
|
|
|
|
from timmy.cli import app
|
|
from timmy.prompts import TIMMY_STATUS_PROMPT
|
|
|
|
runner = CliRunner()
|
|
|
|
|
|
def test_status_uses_status_prompt():
|
|
"""status command must pass TIMMY_STATUS_PROMPT to the agent."""
|
|
mock_timmy = MagicMock()
|
|
|
|
with patch("timmy.cli.create_timmy", return_value=mock_timmy):
|
|
runner.invoke(app, ["status"])
|
|
|
|
mock_timmy.print_response.assert_called_once_with(TIMMY_STATUS_PROMPT, stream=False)
|
|
|
|
|
|
def test_status_does_not_use_inline_string():
|
|
"""status command must not pass the old inline hardcoded string."""
|
|
mock_timmy = MagicMock()
|
|
|
|
with patch("timmy.cli.create_timmy", return_value=mock_timmy):
|
|
runner.invoke(app, ["status"])
|
|
|
|
call_args = mock_timmy.print_response.call_args
|
|
assert call_args[0][0] != "Brief status report — one sentence."
|