Add list to validate response

CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## 1.18.2 - 2024-12-03
+
+### Add
+
+- Add the support to list type for action response
+
 ## 1.18.1 - 2024-12-03
 
 ### Add

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "sekoia-automation-sdk"
 
-version = "1.18.1"
+version = "1.18.2"
 description = "SDK to create Sekoia.io playbook modules"
 license = "MIT"
 readme = "README.md"

sekoia_automation/action.py

@@ -212,6 +212,16 @@ def validate_results(self):
                 except Exception:
                     sentry_sdk.capture_exception()
 
+            # Add a check for list
+            if isinstance(self._results, list):
+                for result in self._results:
+                    if isinstance(result, dict):
+                        try:
+                            orjson.dumps(result)
+                        except Exception:
+                            sentry_sdk.capture_exception()
+                return
+
             # If we reached this point, the results are invalid
             self._error = f"Results are invalid: '{self._results}'"
             self._results = None

tests/test_action.py

@@ -171,6 +171,24 @@ def test_validate_results_none():
     assert action.error_message is None
 
 
+def test_validate_list_results(mock_volume):
+    class ListAction(Action):
+        def run(self, arguments):
+            return [{"key1": "value1"}, {"key2": "value2"}]
+
+    action = ListAction()
+
+    with requests_mock.Mocker() as rmock:
+        rmock.patch(FAKE_URL)
+
+        action.execute()
+
+        assert rmock.last_request.json()["results"] == [
+            {"key1": "value1"},
+            {"key2": "value2"},
+        ]
+
+
 def test_action_results_invalid(mock_volume):
     class TestAction(Action):
         def run(self, arguments):