ad0d5c9520
Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Peter Evers <peter.evers@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
26 lines
896 B
Python
26 lines
896 B
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from vibe.cli.textual_ui.widgets.compact import CompactMessage
|
|
from vibe.core.session.session_id import shorten_session_id
|
|
|
|
|
|
class TestCompactMessage:
|
|
def test_get_content_includes_session_ids_after_compaction(self) -> None:
|
|
message = CompactMessage()
|
|
message.post_message = MagicMock()
|
|
|
|
message.set_complete(
|
|
old_session_id="11111111-1111-1111-1111-111111111111",
|
|
new_session_id="22222222-2222-2222-2222-222222222222",
|
|
)
|
|
|
|
assert message.get_content() == (
|
|
"Compaction completed.\n"
|
|
"session: "
|
|
f"{shorten_session_id('11111111-1111-1111-1111-111111111111')} "
|
|
"(before compaction) → "
|
|
f"{shorten_session_id('22222222-2222-2222-2222-222222222222')} "
|
|
"(after compaction)"
|
|
)
|