d33db9fff8
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai> 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> Co-Authored-By: Thaddee Tyl <thaddee.tyl@gmail.com> Co-Authored-By: David Brochart <david.brochart@gmail.com> Co-Authored-By: Joseph Guhlin <joseph.guhlin@gmail.com> Co-Authored-By: Thomas Kenbeek <thomaskenbeek@gmail.com> Co-Authored-By: Remenby31 <baptiste.cruvellier31@gmail.com>
103 lines
3 KiB
Python
103 lines
3 KiB
Python
from __future__ import annotations
|
|
|
|
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.ports.whoami_gateway import WhoAmIResponse
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_proposes_upgrade_without_call_when_api_key_is_empty() -> None:
|
|
gateway = FakeWhoAmIGateway(
|
|
WhoAmIResponse(
|
|
is_pro_plan=False,
|
|
advertise_pro_plan=False,
|
|
prompt_switching_to_pro_plan=False,
|
|
)
|
|
)
|
|
action = await decide_plan_offer("", gateway)
|
|
|
|
assert action is PlanOfferAction.UPGRADE
|
|
assert gateway.calls == []
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("response", "expected"),
|
|
[
|
|
(
|
|
WhoAmIResponse(
|
|
is_pro_plan=True,
|
|
advertise_pro_plan=False,
|
|
prompt_switching_to_pro_plan=False,
|
|
),
|
|
PlanOfferAction.NONE,
|
|
),
|
|
(
|
|
WhoAmIResponse(
|
|
is_pro_plan=False,
|
|
advertise_pro_plan=True,
|
|
prompt_switching_to_pro_plan=False,
|
|
),
|
|
PlanOfferAction.UPGRADE,
|
|
),
|
|
(
|
|
WhoAmIResponse(
|
|
is_pro_plan=False,
|
|
advertise_pro_plan=False,
|
|
prompt_switching_to_pro_plan=True,
|
|
),
|
|
PlanOfferAction.SWITCH_TO_PRO_KEY,
|
|
),
|
|
],
|
|
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
|
|
) -> None:
|
|
gateway = FakeWhoAmIGateway(response)
|
|
action = await decide_plan_offer("api-key", gateway)
|
|
|
|
assert action is expected
|
|
assert gateway.calls == ["api-key"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_proposes_nothing_when_nothing_is_suggested() -> None:
|
|
gateway = FakeWhoAmIGateway(
|
|
WhoAmIResponse(
|
|
is_pro_plan=False,
|
|
advertise_pro_plan=False,
|
|
prompt_switching_to_pro_plan=False,
|
|
)
|
|
)
|
|
|
|
action = await decide_plan_offer("api-key", gateway)
|
|
|
|
assert action is PlanOfferAction.NONE
|
|
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)
|
|
|
|
assert action is PlanOfferAction.UPGRADE
|
|
assert gateway.calls == ["bad-key"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_proposes_none_and_logs_warning_when_gateway_error_occurs(
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
gateway = FakeWhoAmIGateway(error=True)
|
|
with caplog.at_level(logging.WARNING):
|
|
action = await decide_plan_offer("api-key", gateway)
|
|
|
|
assert action is PlanOfferAction.NONE
|
|
assert gateway.calls == ["api-key"]
|
|
assert "Failed to fetch plan status." in caplog.text
|