uvos/vibe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

v2.0.2

Browse source 
Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-Authored-By: Clément Siriex <clement.sirieix@mistral.ai>
Co-Authored-By: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
This commit is contained in:
Mathias Gesbert 2026-01-30 14:08:38 +01:00 • committed by Mathias Gesbert
parent bd3497b1c0
commit 9809cfc831
32 changed files with 622 additions and 305 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/cli/plan_offer/test_decide_plan_offer.py
View file

@ -5,7 +5,11 @@ import logging
import pytest
from tests.cli.plan_offer.adapters.fake_whoami_gateway import FakeWhoAmIGateway
from vibe.cli.plan_offer.decide_plan_offer import PlanOfferAction, decide_plan_offer
from vibe.cli.plan_offer.decide_plan_offer import (
PlanOfferAction,
PlanType,
decide_plan_offer,
)
from vibe.cli.plan_offer.ports.whoami_gateway import WhoAmIResponse
@ -18,14 +22,15 @@ async def test_proposes_upgrade_without_call_when_api_key_is_empty() -> None:
prompt_switching_to_pro_plan=False,
)
)
action = await decide_plan_offer("", gateway)
action, plan_type = await decide_plan_offer("", gateway)
assert action is PlanOfferAction.UPGRADE
assert plan_type is PlanType.FREE
assert gateway.calls == []
@pytest.mark.parametrize(
("response", "expected"),
("response", "expected_action", "expected_plan_type"),
[
(
WhoAmIResponse(
@ -34,6 +39,7 @@ async def test_proposes_upgrade_without_call_when_api_key_is_empty() -> None:
prompt_switching_to_pro_plan=False,
),
PlanOfferAction.NONE,
PlanType.PRO,
),
(
WhoAmIResponse(
@ -42,6 +48,7 @@ async def test_proposes_upgrade_without_call_when_api_key_is_empty() -> None:
prompt_switching_to_pro_plan=False,
),
PlanOfferAction.UPGRADE,
PlanType.FREE,
),
(
WhoAmIResponse(
@ -50,18 +57,22 @@ async def test_proposes_upgrade_without_call_when_api_key_is_empty() -> None:
prompt_switching_to_pro_plan=True,
),
PlanOfferAction.SWITCH_TO_PRO_KEY,
PlanType.PRO,
),
],
ids=["with-a-pro-plan", "without-a-pro-plan", "with-a-non-pro-key"],
)
@pytest.mark.asyncio
async def test_proposes_an_action_based_on_current_plan_status(
response: WhoAmIResponse, expected: PlanOfferAction
response: WhoAmIResponse,
expected_action: PlanOfferAction,
expected_plan_type: PlanType,
) -> None:
gateway = FakeWhoAmIGateway(response)
action = await decide_plan_offer("api-key", gateway)
action, plan_type = await decide_plan_offer("api-key", gateway)
assert action is expected
assert action is expected_action
assert plan_type is expected_plan_type
assert gateway.calls == ["api-key"]
@ -75,18 +86,20 @@ async def test_proposes_nothing_when_nothing_is_suggested() -> None:
)
)
action = await decide_plan_offer("api-key", gateway)
action, plan_type = await decide_plan_offer("api-key", gateway)
assert action is PlanOfferAction.NONE
assert plan_type is PlanType.UNKNOWN
assert gateway.calls == ["api-key"]
@pytest.mark.asyncio
async def test_proposes_upgrade_when_api_key_is_unauthorized() -> None:
gateway = FakeWhoAmIGateway(unauthorized=True)
action = await decide_plan_offer("bad-key", gateway)
action, plan_type = await decide_plan_offer("bad-key", gateway)
assert action is PlanOfferAction.UPGRADE
assert plan_type is PlanType.FREE
assert gateway.calls == ["bad-key"]
@ -96,8 +109,9 @@ async def test_proposes_none_and_logs_warning_when_gateway_error_occurs(
) -> None:
gateway = FakeWhoAmIGateway(error=True)
with caplog.at_level(logging.WARNING):
action = await decide_plan_offer("api-key", gateway)
action, plan_type = await decide_plan_offer("api-key", gateway)
assert action is PlanOfferAction.NONE
assert plan_type is PlanType.UNKNOWN
assert gateway.calls == ["api-key"]
assert "Failed to fetch plan status." in caplog.text