feat(unrelated): SWFError: add error_code

* add SWFError.error_code
* @translate: fill it when given a `ClientError` instance
* add unit test

Signed-off-by: Yves Bastide <yves@botify.com>

Author: ybastide

simpleflow/swf/mapper/exceptions.py

@@ -9,7 +9,7 @@
 
 
 class SWFError(Exception):
-    def __init__(self, message: str = "", raw_error: str = "", *args) -> None:
+    def __init__(self, message: str = "", raw_error: str = "", error_code: str = "", *args) -> None:
         """
         Examples:
 
@@ -59,6 +59,7 @@ def __init__(self, message: str = "", raw_error: str = "", *args) -> None:
 
         self.kind = values[0].strip()
         self.type_ = self.kind.lower().strip().replace(" ", "_") if self.kind else None
+        self.error_code = error_code
 
     @property
     def message(self):
@@ -312,6 +313,8 @@ def translate(exceptions, to):
     """
 
     def throw(err, *args, **kwargs):
+        if isinstance(getattr(err, "response", None), dict) and issubclass(to, SWFError):
+            raise to(extract_message(err), error_code=extract_error_code(err)) from None
         raise to(extract_message(err))
 
     return catch(exceptions, handle_with=throw)

tests/test_simpleflow/swf/mapper/test_exceptions.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+import pytest
+from botocore.exceptions import ClientError
+
+from simpleflow.swf.mapper.exceptions import ResponseError, translate
+
+
+def test_translate():
+    def raiser():
+        raise ClientError(
+            error_response={
+                "Error": {
+                    "Code": "UnknownError",
+                    "Message": "unknown error",
+                }
+            },
+            operation_name="Foo",
+        )
+
+    error_ = translate(ClientError, to=ResponseError)(raiser)
+    with pytest.raises(ResponseError) as excinfo:
+        error_()
+    assert excinfo.value.error_code == "UnknownError"