228f3c65a9
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Corentin André <corentin.andre@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Peter Evers <pevers90@gmail.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: MichisGitIsKing <MichisGitIsKing@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from vibe.cli.textual_ui.session_exit import print_session_resume_message
|
|
from vibe.core.config import SessionLoggingConfig
|
|
from vibe.core.types import AgentStats
|
|
|
|
|
|
def test_print_session_resume_message_skips_output_without_session_id(
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
print_session_resume_message(None, AgentStats(), SessionLoggingConfig())
|
|
|
|
assert capsys.readouterr().out == ""
|
|
|
|
|
|
def test_print_session_resume_message_prints_resume_commands_and_usage(
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
print_session_resume_message(
|
|
"12345678-1234-1234-1234-123456789abc",
|
|
AgentStats(session_prompt_tokens=14_867, session_completion_tokens=6),
|
|
SessionLoggingConfig(),
|
|
)
|
|
|
|
assert capsys.readouterr().out == (
|
|
"\n"
|
|
"Total tokens used this session: input=14,867 output=6 (total=14,873)\n"
|
|
"\n"
|
|
"To continue this session, run: vibe --continue\n"
|
|
"Or: vibe --resume 12345678-1234-1234-1234-123456789abc\n"
|
|
)
|
|
|
|
|
|
def test_print_session_resume_message_prints_zero_usage_for_resumed_run_without_llm_activity(
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
print_session_resume_message("12345678", AgentStats(), SessionLoggingConfig())
|
|
|
|
assert capsys.readouterr().out == (
|
|
"\n"
|
|
"Total tokens used this session: input=0 output=0 (total=0)\n"
|
|
"\n"
|
|
"To continue this session, run: vibe --continue\n"
|
|
"Or: vibe --resume 12345678\n"
|
|
)
|