Fix InputAnalyzerJob type and add regression test

TheHive-Project · Feb 17, 2025 · 58580ab · 58580ab
1 parent 3dc1a88
commit 58580ab
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/tests/test_cortex_endpoint.py b/tests/test_cortex_endpoint.py
@@ -1,8 +1,21 @@
 from thehive4py.client import TheHiveApi
 from thehive4py.types.case import OutputCase
+from thehive4py.types.observable import OutputObservable
 
 
 class TestCortexEndpoint:
+    def test_create_analyzer_job(
+        self, thehive: TheHiveApi, test_observable: OutputObservable
+    ):
+        output_analyzer_job = thehive.cortex.create_analyzer_job(
+            job={
+                "analyzerId": "example",
+                "cortexId": "cortex",
+                "artifactId": test_observable["_id"],
+            }
+        )
+        assert output_analyzer_job["_type"] == "case_artifact_job"
+
     def test_list_analyzers(self, thehive: TheHiveApi):
         analyzers = thehive.cortex.list_analyzers()
         assert analyzers == []

diff --git a/thehive4py/types/cortex.py b/thehive4py/types/cortex.py
@@ -86,7 +86,7 @@ class InputResponderAction(InputResponderActionRequired, total=False):
 class InputAnalyzerJobRequired(TypedDict):
     analyzerId: str
     cortexId: str
-    observableId: str
+    artifactId: str
 
 
 class InputAnalyzerJob(InputAnalyzerJobRequired, total=False):