add exception handling to release endpoint

Author: bisgaard-itis

services/api-server/src/simcore_service_api_server/services_rpc/resource_usage_tracker.py

@@ -6,10 +6,14 @@
     LicensedItemCheckoutID,
 )
 from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
+from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
+    LicensedItemCheckoutNotFoundError as _LicensedItemCheckoutNotFoundError,
+)
 from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.licensed_items_checkouts import (
     get_licensed_item_checkout as _get_licensed_item_checkout,
 )
 
+from ..exceptions.backend_errors import LicensedItemCheckoutNotFoundError
 from ..exceptions.service_errors_utils import service_exception_mapper
 from ..models.schemas.model_adapter import LicensedItemCheckoutGet
 
@@ -22,7 +26,11 @@
 class ResourceUsageTrackerClient:
     _client: RabbitMQRPCClient
 
-    @_exception_mapper(rpc_exception_map={})
+    @_exception_mapper(
+        rpc_exception_map={
+            _LicensedItemCheckoutNotFoundError: LicensedItemCheckoutNotFoundError
+        }
+    )
     async def get_licensed_item_checkout(
         self, *, product_name: str, licensed_item_checkout_id: LicensedItemCheckoutID
     ) -> LicensedItemCheckoutGet:

services/api-server/src/simcore_service_api_server/services_rpc/wb_api_server.py

@@ -19,6 +19,9 @@
 from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
     CanNotCheckoutServiceIsNotRunningError as _CanNotCheckoutServiceIsNotRunningError,
 )
+from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
+    LicensedItemCheckoutNotFoundError as _LicensedItemCheckoutNotFoundError,
+)
 from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
     NotEnoughAvailableSeatsError,
 )
@@ -38,6 +41,7 @@
 from ..exceptions.backend_errors import (
     CanNotCheckoutServiceIsNotRunningError,
     InsufficientNumberOfSeatsError,
+    LicensedItemCheckoutNotFoundError,
 )
 from ..exceptions.service_errors_utils import service_exception_mapper
 from ..models.pagination import PaginationParams
@@ -144,6 +148,11 @@ async def checkout_licensed_item_for_wallet(
             num_of_seats=licensed_item_checkout_get.num_of_seats,
         )
 
+    @_exception_mapper(
+        rpc_exception_map={
+            _LicensedItemCheckoutNotFoundError: LicensedItemCheckoutNotFoundError
+        }
+    )
     async def release_licensed_item_for_wallet(
         self,
         *,

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

@@ -15,14 +15,15 @@
 from servicelib.rabbitmq import RPCRouter
 from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
     LICENSES_ERRORS,
+    LicensedItemCheckoutNotFoundError,
 )
 
 from ...services import licensed_items_checkouts
 
 router = RPCRouter()
 
 
-@router.expose(reraise_if_error_type=LICENSES_ERRORS)
+@router.expose(reraise_if_error_type=(LicensedItemCheckoutNotFoundError,))
 async def get_licensed_item_checkout(
     app: FastAPI,
     *,

services/web/server/src/simcore_service_webserver/licenses/_rpc.py

@@ -18,6 +18,7 @@
 from servicelib.rabbitmq.rpc_interfaces.resource_usage_tracker.errors import (
     LICENSES_ERRORS,
     CanNotCheckoutNotEnoughAvailableSeatsError,
+    LicensedItemCheckoutNotFoundError,
     NotEnoughAvailableSeatsError,
 )
 from simcore_service_api_server.exceptions.backend_errors import (
@@ -103,7 +104,7 @@ async def checkout_licensed_item_for_wallet(
     )
 
 
-@router.expose(reraise_if_error_type=LICENSES_ERRORS)
+@router.expose(reraise_if_error_type=(LicensedItemCheckoutNotFoundError,))
 async def release_licensed_item_for_wallet(
     app: web.Application,
     *,