fix kafka tests

fkie-cad · Sep 13, 2024 · 6b82dbc · 6b82dbc
1 parent 185e86d
commit 6b82dbc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/logprep/connector/confluent_kafka/input.py b/logprep/connector/confluent_kafka/input.py
@@ -527,3 +527,10 @@ def health(self) -> bool:
             self.metrics.number_of_errors += 1
             return False
         return super().health()
+
+    def setup(self) -> None:
+        """Set the component up."""
+        try:
+            super().setup()
+        except KafkaException as error:
+            raise FatalInputError(self, f"Could not setup kafka consumer: {error}") from error
diff --git a/logprep/connector/confluent_kafka/output.py b/logprep/connector/confluent_kafka/output.py
@@ -340,3 +340,10 @@ def health(self) -> bool:
             self.metrics.number_of_errors += 1
             return False
         return super().health()
+
+    def setup(self) -> None:
+        """Set the component up."""
+        try:
+            super().setup()
+        except KafkaException as error:
+            raise FatalOutputError(self, f"Could not setup kafka producer: {error}") from error