Add test with fake stx client implementation

[pti/o2.git] / o2ims / domain / stx_object.py

diff --git a/o2ims/domain/stx_object.py b/o2ims/domain/stx_object.py
index ee4f718..0a17092 100644 (file)
--- a/o2ims/domain/stx_object.py
+++ b/o2ims/domain/stx_object.py
@@ -18,12 +18,13 @@ import json
 \r
 \r
 class StxGenericModel:\r
-    def __init__(self, api_response: dict) -> None:\r
-        self.id = api_response.uuid\r
-        self.content = json.dumps(api_response.to_dict())\r
-        self.updatetime = api_response.updated_at\r
-        self.createtime = api_response.created_at\r
-        self.name = api_response.name\r
+    def __init__(self, api_response: dict = None) -> None:\r
+        if api_response:\r
+            self.id = api_response.uuid\r
+            self.content = json.dumps(api_response.to_dict())\r
+            self.updatetime = api_response.updated_at\r
+            self.createtime = api_response.created_at\r
+            self.name = api_response.name\r
 \r
     # def __init__(self, id: str, name: str,\r
     #              lastupdate: datetime, content: str) -> None:\r