From 6d35e5f738c358df15017a71d0a4a67646b69183 Mon Sep 17 00:00:00 2001
From: Michal Jagiello <michal.jagiello@t-mobile.pl>
Date: Thu, 25 May 2023 10:15:41 +0000
Subject: New version of CPS package

Use wrapper to handle cps exceptions
Update CPS tests

Issue-ID: INT-2187
Signed-off-by: Michal Jagiello <michal.jagiello@t-mobile.pl>
Change-Id: I3dec62532e89994545524a3695cfb01e4880c089
---
 src/onapsdk/cps/dataspace.py | 49 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 39 insertions(+), 10 deletions(-)

(limited to 'src/onapsdk')

diff --git a/src/onapsdk/cps/dataspace.py b/src/onapsdk/cps/dataspace.py
index d997f81..282880f 100644
--- a/src/onapsdk/cps/dataspace.py
+++ b/src/onapsdk/cps/dataspace.py
@@ -13,7 +13,9 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 
+from functools import wraps
 from typing import Any, Dict, Iterable
+from onapsdk.exceptions import (APIError, ResourceNotFound)
 
 from .anchor import Anchor
 from .cps_element import CpsElement
@@ -52,6 +54,22 @@ class Dataspace(CpsElement):
         """
         return f"{self._url}/dataspaces/{self.name}"
 
+    def exception_handler(function): # pylint: disable= no-self-argument
+        """Exception handler.
+
+        Handling APIError and throwing ResourceNotFound if Data space does not exist.
+
+        """
+        @wraps(function)
+        def wrapper(*args):
+            try:
+                return function(*args) # pylint: disable= not-callable
+            except APIError as error:
+                if (error.response_status_code == 400 and 'Dataspace not found' in str(error)):
+                    raise ResourceNotFound(error) from error
+                raise
+        return wrapper
+
     @classmethod
     def create(cls, dataspace_name: str) -> "Dataspace":
         """Create dataspace with given name.
@@ -71,6 +89,7 @@ class Dataspace(CpsElement):
         )
         return Dataspace(dataspace_name)
 
+    @exception_handler
     def create_anchor(self, schema_set: SchemaSet, anchor_name: str) -> Anchor:
         """Create anchor.
 
@@ -90,6 +109,7 @@ class Dataspace(CpsElement):
         )
         return Anchor(name=anchor_name, schema_set=schema_set)
 
+    @exception_handler
     def get_anchors(self) -> Iterable[Anchor]:
         """Get all dataspace's anchors.
 
@@ -99,16 +119,23 @@ class Dataspace(CpsElement):
             Iterator[Anchor]: Anchor object
 
         """
-        for anchor_data in self.send_message_json(\
-            "GET",\
-            "Get all CPS dataspace anchors",\
-            f"{self.url}/anchors",\
-            auth=self.auth\
-        ):
-            yield Anchor(name=anchor_data["name"],
-                         schema_set=SchemaSet(name=anchor_data["schemaSetName"],
-                                              dataspace=self))
-
+        try:
+            for anchor_data in self.send_message_json(\
+                "GET",\
+                "Get all CPS dataspace anchors",\
+                f"{self.url}/anchors",\
+                auth=self.auth\
+            ):
+                yield Anchor(name=anchor_data["name"],
+                             schema_set=SchemaSet(name=anchor_data["schemaSetName"],
+                                                  dataspace=self))
+        except APIError as error:
+            if (error.response_status_code == 400 and 'Dataspace not found' in str(error)):
+                raise ResourceNotFound(error) from error
+            raise
+
+
+    @exception_handler
     def get_anchor(self, anchor_name: str) -> Anchor:
         """Get dataspace anchor by name.
 
@@ -131,6 +158,7 @@ class Dataspace(CpsElement):
                       schema_set=SchemaSet(name=anchor_data["schemaSetName"],
                                            dataspace=self))
 
+    @exception_handler
     def get_schema_set(self, schema_set_name: str) -> SchemaSet:
         """Get schema set by name.
 
@@ -159,6 +187,7 @@ class Dataspace(CpsElement):
             ]
         )
 
+    @exception_handler
     def create_schema_set(self, schema_set_name: str, schema_set: bytes) -> SchemaSet:
         """Create schema set.
 
-- 
cgit 1.2.3-korg