2.0.0

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>

tests/cli/plan_offer/test_decide_plan_offer.py

@@ -0,0 +1,103 @@
+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