romfast/echo-core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stage-6: model selection and advanced commands

Browse Source 
/model (show/change), /restart (owner), /logs, set_session_model API, model reset on /clear. 20 new tests (161 total).

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
MoltBot Service
2026-02-13 13:13:38 +00:00
parent a1a6ca9a3f
commit 5bdceff732
6 changed files with 475 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tests/test_claude_session.py
View File

@@ -20,6 +20,7 @@ from src.claude_session import (
list_sessions,
resume_session,
send_message,
set_session_model,
start_session,
)
@@ -574,3 +575,46 @@ class TestListSessions:
claude_session, "_SESSIONS_FILE", tmp_path / "nonexistent.json"
)
assert list_sessions() == {}
# ---------------------------------------------------------------------------
# set_session_model
# ---------------------------------------------------------------------------
class TestSetSessionModel:
def test_updates_model_in_active_json(self, tmp_path, monkeypatch):
sessions_dir = tmp_path / "sessions"
sessions_dir.mkdir()
sf = sessions_dir / "active.json"
monkeypatch.setattr(claude_session, "SESSIONS_DIR", sessions_dir)
monkeypatch.setattr(claude_session, "_SESSIONS_FILE", sf)
sf.write_text(json.dumps({
"general": {
"session_id": "abc",
"model": "sonnet",
"message_count": 1,
}
}))
result = set_session_model("general", "opus")
assert result is True
data = json.loads(sf.read_text())
assert data["general"]["model"] == "opus"
def test_returns_false_when_no_session(self, tmp_path, monkeypatch):
sessions_dir = tmp_path / "sessions"
sessions_dir.mkdir()
sf = sessions_dir / "active.json"
monkeypatch.setattr(claude_session, "SESSIONS_DIR", sessions_dir)
monkeypatch.setattr(claude_session, "_SESSIONS_FILE", sf)
sf.write_text("{}")
result = set_session_model("general", "opus")
assert result is False
def test_invalid_model_raises(self):
with pytest.raises(ValueError, match="Invalid model"):
set_session_model("general", "gpt4")

156
tests/test_discord.py
View File

@@ -2,6 +2,7 @@
import json
import logging
import signal
import pytest
from unittest.mock import AsyncMock, MagicMock, patch
@@ -500,3 +501,158 @@ class TestStatusSlashCommand:
msg = interaction.response.send_message.call_args
assert "no active session" in msg.args[0].lower()
# --- /clear mentions model reset ---
class TestClearMentionsModelReset:
@pytest.mark.asyncio
@patch("src.adapters.discord_bot.clear_session")
async def test_clear_mentions_model_reset(self, mock_clear, owned_bot):
mock_clear.return_value = True
cmd = _find_command(owned_bot.tree, "clear")
interaction = _mock_interaction(channel_id="900")
await cmd.callback(interaction)
msg = interaction.response.send_message.call_args
text = msg.args[0]
assert "model reset" in text.lower()
assert "sonnet" in text.lower()
# --- /model slash command ---
class TestModelSlashCommand:
@pytest.mark.asyncio
@patch("src.adapters.discord_bot.get_active_session")
async def test_model_no_args_shows_current_with_session(self, mock_get, owned_bot):
mock_get.return_value = {"model": "opus"}
cmd = _find_command(owned_bot.tree, "model")
interaction = _mock_interaction(channel_id="900")
await cmd.callback(interaction, choice=None)
msg = interaction.response.send_message.call_args
text = msg.args[0]
assert "opus" in text.lower()
assert "haiku" in text
assert "sonnet" in text
assert msg.kwargs.get("ephemeral") is True
@pytest.mark.asyncio
@patch("src.adapters.discord_bot.get_active_session")
async def test_model_no_args_shows_default_without_session(self, mock_get, owned_bot):
mock_get.return_value = None
cmd = _find_command(owned_bot.tree, "model")
interaction = _mock_interaction(channel_id="900")
await cmd.callback(interaction, choice=None)
msg = interaction.response.send_message.call_args
text = msg.args[0]
assert "sonnet" in text # default from config
@pytest.mark.asyncio
@patch("src.adapters.discord_bot.set_session_model")
@patch("src.adapters.discord_bot.get_active_session")
async def test_model_with_choice_changes_existing_session(self, mock_get, mock_set, owned_bot):
mock_get.return_value = {"model": "sonnet", "session_id": "abc"}
cmd = _find_command(owned_bot.tree, "model")
interaction = _mock_interaction(channel_id="900")
choice = MagicMock()
choice.value = "opus"
await cmd.callback(interaction, choice=choice)
mock_set.assert_called_once_with("900", "opus")
msg = interaction.response.send_message.call_args
assert "opus" in msg.args[0]
@pytest.mark.asyncio
@patch("src.claude_session._save_sessions")
@patch("src.claude_session._load_sessions")
@patch("src.adapters.discord_bot.get_active_session")
async def test_model_with_choice_presets_when_no_session(self, mock_get, mock_load, mock_save, owned_bot):
mock_get.return_value = None
mock_load.return_value = {}
cmd = _find_command(owned_bot.tree, "model")
interaction = _mock_interaction(channel_id="900")
choice = MagicMock()
choice.value = "haiku"
await cmd.callback(interaction, choice=choice)
mock_save.assert_called_once()
saved_data = mock_save.call_args[0][0]
assert saved_data["900"]["model"] == "haiku"
assert saved_data["900"]["session_id"] == ""
msg = interaction.response.send_message.call_args
assert "haiku" in msg.args[0]
# --- /restart slash command ---
class TestRestartSlashCommand:
@pytest.mark.asyncio
async def test_restart_owner_succeeds(self, owned_bot, tmp_path):
pid_file = tmp_path / "echo-core.pid"
pid_file.write_text("12345")
with patch.object(discord_bot, "PROJECT_ROOT", tmp_path), \
patch("os.kill") as mock_kill:
cmd = _find_command(owned_bot.tree, "restart")
interaction = _mock_interaction(user_id="111")
await cmd.callback(interaction)
mock_kill.assert_called_once_with(12345, signal.SIGTERM)
msg = interaction.response.send_message.call_args
assert "restarting" in msg.args[0].lower()
@pytest.mark.asyncio
async def test_restart_non_owner_rejected(self, owned_bot):
cmd = _find_command(owned_bot.tree, "restart")
interaction = _mock_interaction(user_id="999")
await cmd.callback(interaction)
msg = interaction.response.send_message.call_args
assert "owner only" in msg.args[0].lower()
@pytest.mark.asyncio
async def test_restart_no_pid_file(self, owned_bot, tmp_path):
with patch.object(discord_bot, "PROJECT_ROOT", tmp_path):
cmd = _find_command(owned_bot.tree, "restart")
interaction = _mock_interaction(user_id="111")
await cmd.callback(interaction)
msg = interaction.response.send_message.call_args
assert "no pid file" in msg.args[0].lower()
# --- /logs slash command ---
class TestLogsSlashCommand:
@pytest.mark.asyncio
async def test_logs_returns_code_block(self, owned_bot, tmp_path):
log_dir = tmp_path / "logs"
log_dir.mkdir()
log_file = log_dir / "echo-core.log"
log_file.write_text("line1\nline2\nline3\n")
with patch.object(discord_bot, "PROJECT_ROOT", tmp_path):
cmd = _find_command(owned_bot.tree, "logs")
interaction = _mock_interaction()
await cmd.callback(interaction, n=10)
msg = interaction.response.send_message.call_args
text = msg.args[0]
assert "```" in text
assert "line1" in text
assert "line3" in text
@pytest.mark.asyncio
async def test_logs_no_file(self, owned_bot, tmp_path):
with patch.object(discord_bot, "PROJECT_ROOT", tmp_path):
cmd = _find_command(owned_bot.tree, "logs")
interaction = _mock_interaction()
await cmd.callback(interaction, n=10)
msg = interaction.response.send_message.call_args
assert "no log file" in msg.args[0].lower()

89
tests/test_router.py
View File

@@ -20,21 +20,94 @@ def reset_router_config():
class TestClearCommand:
@patch("src.router._get_config")
@patch("src.router.clear_session")
def test_clear_active_session(self, mock_clear):
def test_clear_active_session(self, mock_clear, mock_get_config):
mock_clear.return_value = True
mock_cfg = MagicMock()
mock_cfg.get.return_value = "sonnet"
mock_get_config.return_value = mock_cfg
response, is_cmd = route_message("ch-1", "user-1", "/clear")
assert response == "Session cleared."
assert response == "Session cleared. Model reset to sonnet."
assert is_cmd is True
mock_clear.assert_called_once_with("ch-1")
@patch("src.router._get_config")
@patch("src.router.clear_session")
def test_clear_no_session(self, mock_clear):
def test_clear_no_session(self, mock_clear, mock_get_config):
mock_clear.return_value = False
mock_cfg = MagicMock()
mock_cfg.get.return_value = "sonnet"
mock_get_config.return_value = mock_cfg
response, is_cmd = route_message("ch-1", "user-1", "/clear")
assert response == "No active session."
assert is_cmd is True
@patch("src.router._get_config")
@patch("src.router.clear_session")
def test_clear_mentions_model_reset(self, mock_clear, mock_get_config):
mock_clear.return_value = True
mock_cfg = MagicMock()
mock_cfg.get.return_value = "opus"
mock_get_config.return_value = mock_cfg
response, is_cmd = route_message("ch-1", "user-1", "/clear")
assert "model reset" in response.lower()
assert "opus" in response
# --- /model command ---
class TestModelCommand:
@patch("src.router.get_active_session")
def test_model_show_current_with_session(self, mock_get):
mock_get.return_value = {"model": "opus"}
response, is_cmd = route_message("ch-1", "user-1", "/model")
assert is_cmd is True
assert "opus" in response
assert "haiku" in response # available models listed
@patch("src.router._get_config")
@patch("src.router._get_channel_config")
@patch("src.router.get_active_session")
def test_model_show_current_no_session(self, mock_get, mock_chan_cfg, mock_get_config):
mock_get.return_value = None
mock_chan_cfg.return_value = None
mock_cfg = MagicMock()
mock_cfg.get.return_value = "sonnet"
mock_get_config.return_value = mock_cfg
response, is_cmd = route_message("ch-1", "user-1", "/model")
assert is_cmd is True
assert "sonnet" in response
@patch("src.router.set_session_model")
@patch("src.router.get_active_session")
def test_model_change_opus(self, mock_get, mock_set):
mock_get.return_value = {"model": "sonnet", "session_id": "abc"}
response, is_cmd = route_message("ch-1", "user-1", "/model opus")
assert is_cmd is True
mock_set.assert_called_once_with("ch-1", "opus")
assert "opus" in response
def test_model_invalid_choice(self):
response, is_cmd = route_message("ch-1", "user-1", "/model gpt4")
assert is_cmd is True
assert "invalid" in response.lower()
assert "gpt4" in response
@patch("src.claude_session._save_sessions")
@patch("src.claude_session._load_sessions")
@patch("src.router.get_active_session")
def test_model_change_no_session_presets(self, mock_get, mock_load, mock_save):
mock_get.return_value = None
mock_load.return_value = {}
response, is_cmd = route_message("ch-1", "user-1", "/model haiku")
assert is_cmd is True
mock_save.assert_called_once()
saved = mock_save.call_args[0][0]
assert saved["ch-1"]["model"] == "haiku"
assert "haiku" in response
# --- /status command ---
@@ -186,3 +259,13 @@ class TestModelResolution:
route_message("ch-1", "user-1", "hello")
mock_send.assert_called_once_with("ch-1", "hello", model="sonnet")
@patch("src.router.get_active_session")
@patch("src.router.send_message")
def test_session_model_takes_priority(self, mock_send, mock_get_session):
"""Session model takes priority over channel and global defaults."""
mock_send.return_value = "ok"
mock_get_session.return_value = {"model": "opus", "session_id": "abc"}
route_message("ch-1", "user-1", "hello")
mock_send.assert_called_once_with("ch-1", "hello", model="opus")