fix(SentinelOne): fix tests

SEKOIA-IO · Dec 16, 2024 · 9b6dc01 · 9b6dc01
1 parent 601687d
commit 9b6dc01
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/SentinelOne/tests/singularity/conftest.py b/SentinelOne/tests/singularity/conftest.py
@@ -1,4 +1,3 @@
-from pathlib import Path
 from unittest.mock import AsyncMock
 
 import pytest
@@ -24,7 +23,7 @@ def side_effect_return_input(events: list[str]) -> list[str]:
 
 
 @pytest.fixture
-def sentinel_module(symphony_storage: Path) -> SentinelOneModule:
+def sentinel_module() -> SentinelOneModule:
     """
     Create a SentinelOne module.
 

diff --git a/SentinelOne/tests/singularity/test_connectors.py b/SentinelOne/tests/singularity/test_connectors.py
@@ -1,5 +1,6 @@
 import itertools
 import time
+from pathlib import Path
 from multiprocessing import Process
 from typing import Any
 from unittest.mock import AsyncMock, MagicMock
@@ -16,9 +17,9 @@ class CustomTestConnector(AbstractSingularityConnector):
 
 @pytest.fixture
 def custom_test_connector(
-    sentinel_module: SentinelOneModule, mock_push_data_to_intakes: AsyncMock
+    sentinel_module: SentinelOneModule, symphony_storage: Path, mock_push_data_to_intakes: AsyncMock
 ) -> CustomTestConnector:
-    connector = CustomTestConnector(sentinel_module)
+    connector = CustomTestConnector(module=sentinel_module, data_path=symphony_storage)
     connector.configuration = {
         "intake_key": "test_key",
         "intake_server": "http://test_server.test",