fix tests

Author: matusdrobuliak66

services/resource-usage-tracker/src/simcore_service_resource_usage_tracker/api/rpc/_service_runs.py

@@ -1,3 +1,4 @@
+# pylint: disable=too-many-arguments
 from fastapi import FastAPI
 from models_library.api_schemas_resource_usage_tracker.service_runs import (
     OsparcCreditsAggregatedUsagesPage,

services/web/server/src/simcore_service_webserver/projects/_states_handlers.py

@@ -45,6 +45,7 @@
     ProjectNotFoundError,
     ProjectStartsTooManyDynamicNodesError,
     ProjectTooManyProjectOpenedError,
+    ProjectWalletDebtError,
 )
 
 _logger = logging.getLogger(__name__)
@@ -75,7 +76,7 @@ async def _wrapper(request: web.Request) -> web.StreamResponse:
         except ProjectTooManyProjectOpenedError as exc:
             raise web.HTTPConflict(reason=f"{exc}") from exc
 
-        except WalletNotEnoughCreditsError as exc:
+        except (WalletNotEnoughCreditsError, ProjectWalletDebtError) as exc:
             raise web.HTTPPaymentRequired(reason=f"{exc}") from exc
 
     return _wrapper

services/web/server/tests/unit/with_dbs/02/test_projects_wallet_handlers.py

@@ -6,12 +6,20 @@
 # pylint: disable=too-many-statements
 
 from collections.abc import Iterator
+from decimal import Decimal
 from http import HTTPStatus
 
 import pytest
 import sqlalchemy as sa
 from aiohttp.test_utils import TestClient
+from models_library.api_schemas_resource_usage_tracker.credit_transactions import (
+    WalletTotalCredits,
+)
+from models_library.api_schemas_resource_usage_tracker.service_runs import (
+    ServiceRunPage,
+)
 from models_library.api_schemas_webserver.wallets import WalletGet
+from pytest_mock import MockerFixture
 from pytest_simcore.helpers.assert_checks import assert_status
 from pytest_simcore.helpers.webserver_login import LoggedUser, UserInfoDict
 from servicelib.aiohttp import status
@@ -97,13 +105,37 @@ def setup_wallets_db(
         con.execute(wallets.delete())
 
 
+@pytest.fixture
+def mock_get_project_wallet_total_credits(
+    mocker: MockerFixture, setup_wallets_db: list[WalletGet]
+):
+    mocker.patch(
+        "simcore_service_webserver.projects._wallets_api.credit_transactions.get_project_wallet_total_credits",
+        spec=True,
+        return_value=WalletTotalCredits(
+            wallet_id=setup_wallets_db[0].wallet_id, available_osparc_credits=Decimal(0)
+        ),
+    )
+
+
+@pytest.fixture
+def mock_get_service_run_page(mocker: MockerFixture):
+    mocker.patch(
+        "simcore_service_webserver.projects._wallets_api.service_runs.get_service_run_page",
+        spec=True,
+        return_value=ServiceRunPage(items=[], total=0),
+    )
+
+
 @pytest.mark.parametrize("user_role,expected", [(UserRole.USER, status.HTTP_200_OK)])
 async def test_project_wallets_full_workflow(
     client: TestClient,
     logged_user: UserInfoDict,
     user_project: ProjectDict,
     expected: HTTPStatus,
     setup_wallets_db: list[WalletGet],
+    mock_get_project_wallet_total_credits: None,
+    mock_get_service_run_page: None,
 ):
     assert client.app