fix tests for confluent kakfa

fkie-cad · Sep 26, 2024 · 68d2cdb · 68d2cdb
1 parent 3d4a39e
commit 68d2cdb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/unit/connector/test_confluent_kafka_input.py b/tests/unit/connector/test_confluent_kafka_input.py
@@ -298,6 +298,9 @@ def test_client_id_can_be_overwritten(self, mock_consumer):
         input_config = deepcopy(self.CONFIG)
         input_config["kafka_config"]["client.id"] = "thisclientid"
         kafka_input = Factory.create({"test": input_config})
+        metadata = mock.MagicMock()
+        metadata.topics = [kafka_input._config.topic]
+        kafka_input._consumer.list_topics.return_value = metadata
         kafka_input.setup()
         mock_consumer.assert_called()
         assert mock_consumer.call_args[0][0].get("client.id") == "thisclientid"
@@ -308,6 +311,9 @@ def test_statistics_interval_can_be_overwritten(self, mock_consumer):
         input_config = deepcopy(self.CONFIG)
         input_config["kafka_config"]["statistics.interval.ms"] = "999999999"
         kafka_input = Factory.create({"test": input_config})
+        metadata = mock.MagicMock()
+        metadata.topics = [kafka_input._config.topic]
+        kafka_input._consumer.list_topics.return_value = metadata
         kafka_input.setup()
         mock_consumer.assert_called()
         assert mock_consumer.call_args[0][0].get("statistics.interval.ms") == "999999999"