fix tests

fkie-cad · Aug 19, 2024 · 6325f73 · 6325f73
1 parent eafbe56
commit 6325f73
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/connector/test_http_input.py b/tests/unit/connector/test_http_input.py
@@ -17,6 +17,7 @@
 from logprep.abc.input import FatalInputError
 from logprep.connector.http.input import HttpInput
 from logprep.factory import Factory
+from logprep.framework.pipeline_manager import ThrottlingQueue
 from logprep.util.defaults import ENV_NAME_LOGPREP_CREDENTIALS_FILE
 from tests.unit.connector.base import BaseInputTestCase
 
@@ -48,7 +49,9 @@ def create_credentials(tmp_path):
 class TestHttpConnector(BaseInputTestCase):
 
     def setup_method(self):
-        HttpInput.messages = multiprocessing.Queue(maxsize=self.CONFIG.get("message_backlog_size"))
+        HttpInput.messages = ThrottlingQueue(
+            ctx=multiprocessing.get_context(), maxsize=self.CONFIG.get("message_backlog_size")
+        )
         super().setup_method()
         self.object.pipeline_index = 1
         self.object.setup()