v2.6.0 (#524)
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Quentin <torroba.q@gmail.com> Co-authored-by: Simon <80467011+sorgfresser@users.noreply.github.com> Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: angelapopopo <angele.lenglemetz@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
parent
5103019b01
commit
eb580209d4
180 changed files with 11136 additions and 1030 deletions
77
tests/cli/test_feedback_bar.py
Normal file
77
tests/cli/test_feedback_bar.py
Normal file
|
|
@ -0,0 +1,77 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
from vibe.cli.textual_ui.widgets.feedback_bar import FeedbackBar
|
||||
|
||||
|
||||
class TestFeedbackBarState:
|
||||
def test_maybe_show_shows_when_random_below_threshold(self):
|
||||
bar = FeedbackBar()
|
||||
bar.display = False
|
||||
bar._set_active = MagicMock()
|
||||
|
||||
with patch(
|
||||
"vibe.cli.textual_ui.widgets.feedback_bar.random.random", return_value=0
|
||||
):
|
||||
bar.maybe_show()
|
||||
|
||||
bar._set_active.assert_called_once_with(True)
|
||||
|
||||
def test_maybe_show_does_not_show_when_random_above_threshold(self):
|
||||
bar = FeedbackBar()
|
||||
bar.display = False
|
||||
bar._set_active = MagicMock()
|
||||
|
||||
with patch(
|
||||
"vibe.cli.textual_ui.widgets.feedback_bar.random.random", return_value=1.0
|
||||
):
|
||||
bar.maybe_show()
|
||||
|
||||
bar._set_active.assert_not_called()
|
||||
|
||||
def test_hide_calls_set_active_false_when_displayed(self):
|
||||
bar = FeedbackBar()
|
||||
bar.display = True
|
||||
bar._set_active = MagicMock()
|
||||
|
||||
bar.hide()
|
||||
|
||||
bar._set_active.assert_called_once_with(False)
|
||||
|
||||
def test_hide_does_nothing_when_already_hidden(self):
|
||||
bar = FeedbackBar()
|
||||
bar.display = False
|
||||
bar._set_active = MagicMock()
|
||||
|
||||
bar.hide()
|
||||
|
||||
bar._set_active.assert_not_called()
|
||||
|
||||
def test_handle_feedback_key_posts_message_and_deactivates(self):
|
||||
bar = FeedbackBar()
|
||||
bar.set_timer = MagicMock()
|
||||
bar.post_message = MagicMock()
|
||||
bar.query_one = MagicMock()
|
||||
mock_text_area = MagicMock()
|
||||
mock_text_area.feedback_active = True
|
||||
mock_app = MagicMock()
|
||||
mock_app.query_one.return_value = mock_text_area
|
||||
|
||||
with patch.object(
|
||||
type(bar), "app", new_callable=lambda: property(lambda self: mock_app)
|
||||
):
|
||||
bar.handle_feedback_key(3)
|
||||
|
||||
assert mock_text_area.feedback_active is False
|
||||
bar.post_message.assert_called_once()
|
||||
msg = bar.post_message.call_args[0][0]
|
||||
assert isinstance(msg, FeedbackBar.FeedbackGiven)
|
||||
assert msg.rating == 3
|
||||
bar.set_timer.assert_called_once()
|
||||
|
||||
|
||||
class TestFeedbackGivenMessage:
|
||||
def test_message_stores_rating(self):
|
||||
msg = FeedbackBar.FeedbackGiven(rating=2)
|
||||
assert msg.rating == 2
|
||||
Loading…
Add table
Add a link
Reference in a new issue