fix import

Author: matusdrobuliak66

packages/models-library/src/models_library/api_schemas_webserver/wallets.py

@@ -27,13 +27,13 @@ class WalletGet(OutputSchema):
             "examples": [
                 {
                     "wallet_id": 1,
-                    "name": "pm_0987654321",
-                    "description": "https://example.com/payment-method/form",
-                    "owner": "https://example.com/payment-method/form",
+                    "name": "My wallet",
+                    "description": "My description",
+                    "owner": 1,
                     "thumbnail": "https://example.com/payment-method/form",
-                    "status": "https://example.com/payment-method/form",
-                    "created": "https://example.com/payment-method/form",
-                    "modified": "https://example.com/payment-method/form",
+                    "status": "ACTIVE",
+                    "created": "2024-03-25T00:00:00",
+                    "modified": "2024-03-25T00:00:00",
                 }
             ]
         },

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/resource_usage_tracker/licensed_items_purchases.py

@@ -6,13 +6,13 @@
 )
 from models_library.api_schemas_resource_usage_tracker.licensed_items_purchases import (
     LicensedItemPurchaseGet,
-    LicensedItemPurchaseID,
     LicensedItemsPurchasesPage,
 )
 from models_library.basic_types import IDStr
 from models_library.products import ProductName
 from models_library.rabbitmq_basic_types import RPCMethodName
 from models_library.resource_tracker_licensed_items_purchases import (
+    LicensedItemPurchaseID,
     LicensedItemsPurchasesCreate,
 )
 from models_library.rest_ordering import OrderBy

services/web/server/tests/unit/with_dbs/04/licenses/test_licensed_items_handlers.py

@@ -7,8 +7,13 @@
 
 import pytest
 from aiohttp.test_utils import TestClient
+from models_library.api_schemas_resource_usage_tracker.pricing_plans import (
+    PricingUnitGet,
+)
 from models_library.api_schemas_webserver.licensed_items import LicensedItemGet
+from models_library.api_schemas_webserver.wallets import WalletGetWithAvailableCredits
 from models_library.licensed_items import LicensedResourceType
+from pytest_mock.plugin import MockerFixture
 from pytest_simcore.helpers.assert_checks import assert_status
 from pytest_simcore.helpers.webserver_login import UserInfoDict
 from servicelib.aiohttp import status
@@ -18,7 +23,7 @@
 
 
 @pytest.mark.parametrize("user_role,expected", [(UserRole.USER, status.HTTP_200_OK)])
-async def test_licensed_items_db_crud(
+async def test_licensed_items_listing(
     client: TestClient,
     logged_user: UserInfoDict,
     user_project: ProjectDict,
@@ -58,9 +63,77 @@ async def test_licensed_items_db_crud(
     data, _ = await assert_status(resp, status.HTTP_200_OK)
     assert LicensedItemGet(**data)
 
+
+@pytest.fixture
+def mock_licensed_items_purchase_functions(mocker: MockerFixture) -> tuple:
+    mock_wallet_credits = mocker.patch(
+        "simcore_service_webserver.licenses._licensed_items_api.get_wallet_with_available_credits_by_user_and_wallet",
+        spec=True,
+        return_value=WalletGetWithAvailableCredits.model_validate(
+            WalletGetWithAvailableCredits.model_config["json_schema_extra"]["examples"][
+                0
+            ]
+        ),
+    )
+    mock_get_pricing_unit = mocker.patch(
+        "simcore_service_webserver.licenses._licensed_items_api.get_pricing_plan_unit",
+        spec=True,
+        return_value=PricingUnitGet.model_validate(
+            PricingUnitGet.model_config["json_schema_extra"]["examples"][0]
+        ),
+    )
+    mock_create_licensed_item_purchase = mocker.patch(
+        "simcore_service_webserver.licenses._licensed_items_api.licensed_items_purchases.create_licensed_item_purchase",
+        spec=True,
+    )
+
+    return (
+        mock_wallet_credits,
+        mock_get_pricing_unit,
+        mock_create_licensed_item_purchase,
+    )
+
+
+@pytest.mark.parametrize("user_role,expected", [(UserRole.USER, status.HTTP_200_OK)])
+async def test_licensed_items_purchase(
+    client: TestClient,
+    logged_user: UserInfoDict,
+    user_project: ProjectDict,
+    osparc_product_name: str,
+    expected: HTTPStatus,
+    pricing_plan_id: int,
+    mock_licensed_items_purchase_functions: tuple,
+):
+    assert client.app
+
+    licensed_item_db = await _licensed_items_db.create(
+        client.app,
+        product_name=osparc_product_name,
+        name="Model A",
+        licensed_resource_type=LicensedResourceType.VIP_MODEL,
+        pricing_plan_id=pricing_plan_id,
+    )
+    _licensed_item_id = licensed_item_db.licensed_item_id
+
+    # get
+    url = client.app.router["get_licensed_item"].url_for(
+        licensed_item_id=f"{_licensed_item_id}"
+    )
+    resp = await client.get(f"{url}")
+    data, _ = await assert_status(resp, status.HTTP_200_OK)
+    assert LicensedItemGet(**data)
+
     # purchase
     url = client.app.router["purchase_licensed_item"].url_for(
         licensed_item_id=f"{_licensed_item_id}"
     )
-    resp = await client.post(f"{url}", json={"wallet_id": 1, "num_of_seats": 5})
-    # NOTE: Not yet implemented
+    resp = await client.post(
+        f"{url}",
+        json={
+            "wallet_id": 1,
+            "num_of_seats": 5,
+            "pricing_plan_id": pricing_plan_id,
+            "pricing_unit_id": 1,
+        },
+    )
+    await assert_status(resp, status.HTTP_204_NO_CONTENT)