|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from rp.__main__ import main
|
|
|
|
|
|
def test_main_version(capsys):
|
|
with patch("sys.argv", ["rp", "--version"]):
|
|
with pytest.raises(SystemExit):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "RP Assistant" in captured.out
|
|
|
|
|
|
def test_main_create_config_success(capsys):
|
|
with patch("rp.core.config_loader.create_default_config", return_value=True):
|
|
with patch("sys.argv", ["rp", "--create-config"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Configuration file created" in captured.out
|
|
|
|
|
|
def test_main_create_config_fail(capsys):
|
|
with patch("rp.core.config_loader.create_default_config", return_value=False):
|
|
with patch("sys.argv", ["rp", "--create-config"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Error creating configuration file" in captured.err
|
|
|
|
with patch("rp.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.list_sessions.return_value = []
|
|
with patch("sys.argv", ["rp", "--list-sessions"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "No saved sessions found" in captured.out
|
|
|
|
|
|
def test_main_list_sessions_with_sessions(capsys):
|
|
sessions = [{"name": "test", "created_at": "2023-01-01", "message_count": 5}]
|
|
with patch("pr.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.list_sessions.return_value = sessions
|
|
with patch("sys.argv", ["pr", "--list-sessions"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Found 1 saved sessions" in captured.out
|
|
assert "test" in captured.out
|
|
|
|
|
|
def test_main_delete_session_success(capsys):
|
|
with patch("pr.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.delete_session.return_value = True
|
|
with patch("sys.argv", ["pr", "--delete-session", "test"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Session 'test' deleted" in captured.out
|
|
|
|
|
|
def test_main_delete_session_fail(capsys):
|
|
with patch("pr.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.delete_session.return_value = False
|
|
with patch("sys.argv", ["pr", "--delete-session", "test"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Error deleting session 'test'" in captured.err
|
|
|
|
|
|
def test_main_export_session_json(capsys):
|
|
with patch("pr.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.export_session.return_value = True
|
|
with patch("sys.argv", ["pr", "--export-session", "test", "output.json"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Session exported to output.json" in captured.out
|
|
|
|
|
|
def test_main_export_session_md(capsys):
|
|
with patch("pr.core.session.SessionManager") as mock_sm:
|
|
mock_instance = mock_sm.return_value
|
|
mock_instance.export_session.return_value = True
|
|
with patch("sys.argv", ["pr", "--export-session", "test", "output.md"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Session exported to output.md" in captured.out
|
|
|
|
|
|
def test_main_usage(capsys):
|
|
usage = {"total_requests": 10, "total_tokens": 1000, "total_cost": 0.01}
|
|
with patch("pr.core.usage_tracker.UsageTracker.get_total_usage", return_value=usage):
|
|
with patch("sys.argv", ["pr", "--usage"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Total Usage Statistics" in captured.out
|
|
assert "Requests: 10" in captured.out
|
|
|
|
|
|
def test_main_plugins_no_plugins(capsys):
|
|
with patch("pr.plugins.loader.PluginLoader") as mock_loader:
|
|
mock_instance = mock_loader.return_value
|
|
mock_instance.load_plugins.return_value = None
|
|
mock_instance.list_loaded_plugins.return_value = []
|
|
with patch("sys.argv", ["pr", "--plugins"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "No plugins loaded" in captured.out
|
|
|
|
|
|
def test_main_plugins_with_plugins(capsys):
|
|
with patch("pr.plugins.loader.PluginLoader") as mock_loader:
|
|
mock_instance = mock_loader.return_value
|
|
mock_instance.load_plugins.return_value = None
|
|
mock_instance.list_loaded_plugins.return_value = ["plugin1", "plugin2"]
|
|
with patch("sys.argv", ["pr", "--plugins"]):
|
|
main()
|
|
captured = capsys.readouterr()
|
|
assert "Loaded 2 plugins" in captured.out
|
|
|
|
|
|
def test_main_run_assistant():
|
|
with patch("pr.__main__.Assistant") as mock_assistant:
|
|
mock_instance = mock_assistant.return_value
|
|
with patch("sys.argv", ["pr", "test message"]):
|
|
main()
|
|
mock_assistant.assert_called_once()
|
|
mock_instance.run.assert_called_once()
|