e607ccbb00
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Cyprien <courtot.c@gmail.com> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.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: josephine-delas <57808586+josephine-delas@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from vibe.cli.update_notifier.update import do_update
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_do_update_runs_all_commands_even_when_first_succeeds() -> None:
|
|
mock_process = MagicMock()
|
|
mock_process.wait = AsyncMock(return_value=None)
|
|
mock_process.returncode = 0
|
|
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.UPDATE_COMMANDS", ["command_1", "command_2"]
|
|
):
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.asyncio.create_subprocess_shell"
|
|
) as mock_create:
|
|
mock_create.return_value = mock_process
|
|
|
|
result = await do_update()
|
|
|
|
assert result is True
|
|
assert mock_create.call_count == 2
|
|
assert "command_1" in mock_create.call_args_list[0][0][0]
|
|
assert "command_2" in mock_create.call_args_list[1][0][0]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_do_update_returns_true_when_second_command_succeeds() -> None:
|
|
mock_process_fail = MagicMock()
|
|
mock_process_fail.wait = AsyncMock(return_value=None)
|
|
mock_process_fail.returncode = 1
|
|
|
|
mock_process_success = MagicMock()
|
|
mock_process_success.wait = AsyncMock(return_value=None)
|
|
mock_process_success.returncode = 0
|
|
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.UPDATE_COMMANDS", ["command_1", "command_2"]
|
|
):
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.asyncio.create_subprocess_shell"
|
|
) as mock_create:
|
|
mock_create.side_effect = [mock_process_fail, mock_process_success]
|
|
|
|
result = await do_update()
|
|
|
|
assert result is True
|
|
assert mock_create.call_count == 2
|
|
assert "command_1" in mock_create.call_args_list[0][0][0]
|
|
assert "command_2" in mock_create.call_args_list[1][0][0]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_do_update_returns_false_when_all_commands_fail() -> None:
|
|
mock_process = MagicMock()
|
|
mock_process.wait = AsyncMock(return_value=None)
|
|
mock_process.returncode = 1
|
|
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.UPDATE_COMMANDS", ["command_1", "command_2"]
|
|
):
|
|
with patch(
|
|
"vibe.cli.update_notifier.update.asyncio.create_subprocess_shell"
|
|
) as mock_create:
|
|
mock_create.return_value = mock_process
|
|
|
|
result = await do_update()
|
|
|
|
assert result is False
|
|
assert mock_create.call_count == 2
|