add some tests for exceptions

Author: bisgaard-itis

services/api-server/src/simcore_service_api_server/api/routes/wallets.py

@@ -2,29 +2,21 @@
 from typing import Annotated, Any
 
 from fastapi import APIRouter, Depends, status
-from fastapi_pagination import Page
 from models_library.licensed_items import LicensedItemID
 from pydantic import PositiveInt
-from simcore_service_api_server.api.dependencies.authentication import (
-    get_current_user_id,
-    get_product_name,
-)
-from simcore_service_api_server.api.dependencies.webserver_rpc import (
-    get_wb_api_rpc_client,
-)
-from simcore_service_api_server.models.pagination import PaginationParams
-from simcore_service_api_server.models.schemas.licensed_items import (
-    LicensedItemCheckoutData,
-)
-from simcore_service_api_server.services_rpc.wb_api_server import WbApiRpcClient
 
+from ...api.dependencies.authentication import get_current_user_id, get_product_name
+from ...api.dependencies.webserver_rpc import get_wb_api_rpc_client
 from ...exceptions.service_errors_utils import DEFAULT_BACKEND_SERVICE_STATUS_CODES
+from ...models.pagination import Page, PaginationParams
 from ...models.schemas.errors import ErrorGet
+from ...models.schemas.licensed_items import LicensedItemCheckoutData
 from ...models.schemas.model_adapter import (
     LicensedItemCheckoutGet,
     LicensedItemGet,
     WalletGetWithAvailableCreditsLegacy,
 )
+from ...services_rpc.wb_api_server import WbApiRpcClient
 from ..dependencies.webserver_http import AuthSession, get_webserver_session
 from ._constants import FMSG_CHANGELOG_NEW_IN_VERSION
 

services/api-server/tests/unit/test_licensed_items.py

@@ -6,6 +6,7 @@
 from functools import partial
 
 import pytest
+from faker import Faker
 from fastapi import FastAPI, status
 from httpx import AsyncClient, BasicAuth
 from models_library.api_schemas_webserver.licensed_items import (
@@ -14,10 +15,16 @@
 from models_library.api_schemas_webserver.licensed_items import (
     LicensedItemGetPage as _LicensedItemGetPage,
 )
+from models_library.users import UserID
+from models_library.wallets import WalletID
 from pydantic import TypeAdapter
 from pytest_mock import MockerFixture
 from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
 from servicelib.rabbitmq._errors import RemoteMethodNotRegisteredError
+from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
+    CanNotCheckoutNotEnoughAvailableSeatsError,
+    NotEnoughAvailableSeatsError,
+)
 from simcore_service_api_server._meta import API_VTAG
 from simcore_service_api_server.api.dependencies.webserver_rpc import (
     get_wb_api_rpc_client,
@@ -96,3 +103,58 @@ async def test_get_licensed_items_502(
     )
     resp = await client.get(f"{API_VTAG}/licensed-items", auth=auth)
     assert resp.status_code == status.HTTP_502_BAD_GATEWAY
+
+
+@pytest.mark.parametrize(
+    "exception_to_raise,expected_api_server_status_code",
+    [
+        (None, status.HTTP_200_OK),
+        (NotEnoughAvailableSeatsError(), status.HTTP_409_CONFLICT),
+        (
+            CanNotCheckoutNotEnoughAvailableSeatsError(
+                licensed_item_id="2e1af95b-f664-4793-81ae-5512708db3b1",
+                num_of_seats=3,
+                available_num_of_seats=2,
+            ),
+            status.HTTP_409_CONFLICT,
+        ),
+    ],
+)
+async def test_get_licensed_items_for_wallet(
+    mock_wb_api_server_rcp: MockerFixture,
+    client: AsyncClient,
+    auth: BasicAuth,
+    exception_to_raise: Exception | None,
+    expected_api_server_status_code: int,
+    faker: Faker,
+):
+    _wallet_id = faker.pyint(min_value=1)
+
+    async def side_effect(
+        rabbitmq_rpc_client: RabbitMQRPCClient,
+        product_name: str,
+        wallet_id: WalletID,
+        user_id: UserID,
+        offset: int,
+        limit: int,
+    ) -> _LicensedItemGetPage:
+        assert _wallet_id == wallet_id
+        if exception_to_raise is not None:
+            raise exception_to_raise
+        extra = _LicensedItemGet.model_config.get("json_schema_extra")
+        assert isinstance(extra, dict)
+        examples = extra.get("examples")
+        assert isinstance(examples, list)
+        return _LicensedItemGetPage(
+            items=[_LicensedItemGet.model_validate(ex) for ex in examples],
+            total=len(examples),
+        )
+
+    mock_wb_api_server_rcp.patch(
+        "simcore_service_api_server.services_rpc.wb_api_server._get_available_licensed_items_for_wallet",
+        side_effect,
+    )
+    resp = await client.get(
+        f"{API_VTAG}/wallets/{_wallet_id}/licensed-items", auth=auth
+    )
+    assert resp.status_code == expected_api_server_status_code