fix linter

Limmen · Jul 8, 2024 · 4b79964 · 4b79964
1 parent 00fe92a
commit 4b79964
Showing 1 changed file with 19 additions and 26 deletions.
diff --git a/simulation-system/libs/csle-common/tests/test_snort_ids_controller.py b/simulation-system/libs/csle-common/tests/test_snort_ids_controller.py
@@ -73,14 +73,12 @@ def test_stop_snort_ides(self, mock_stop_snort_ids) -> None:
         )
 
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.start_snort_manager")
-    @patch(
-        "csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_monitor_threads_statuses_by_ip_and_port"
-    )
+    @patch("csle_common.controllers.snort_ids_controller.SnortIDSController."
+           "get_snort_idses_monitor_threads_statuses_by_ip_and_port")
     @patch("csle_collector.snort_ids_manager.query_snort_ids_manager.start_snort_ids")
     @patch("grpc.insecure_channel")
-    def test_start_snort_ids(
-        self, mock_insecure_channel, mock_start_snort_ids, mock_get_statuses, mock_start_manager
-    ) -> None:
+    def test_start_snort_ids(self, mock_insecure_channel, mock_start_snort_ids, mock_get_statuses, mock_start_manager) \
+            -> None:
         """
         Test utility function for starting the Snort IDS on a specific IP
 
@@ -111,13 +109,12 @@ def test_start_snort_ids(
         mock_start_snort_ids.assert_called()
 
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.start_snort_manager")
-    @patch(
-        "csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_monitor_threads_statuses_by_ip_and_port"
-    )
+    @patch("csle_common.controllers.snort_ids_controller.SnortIDSController."
+           "get_snort_idses_monitor_threads_statuses_by_ip_and_port")
     @patch("csle_collector.snort_ids_manager.query_snort_ids_manager.stop_snort_ids")
     @patch("grpc.insecure_channel")
     def test_stop_snort_ids(
-        self, mock_insecure_channel, mock_stop_snort_ids, mock_get_statuses, mock_start_manager
+            self, mock_insecure_channel, mock_stop_snort_ids, mock_get_statuses, mock_start_manager
     ) -> None:
         """
         Test utility function for stopping the Snort IDS on a specific IP
@@ -245,14 +242,12 @@ def test_start_snort_idses_monitor_threads(self, mock_start_monitor_thread) -> N
         self.logger.info.assert_called_once_with("Starting Snort IDS monitor thread on IP: 10.0.0.1")
 
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.start_snort_manager")
-    @patch(
-        "csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_monitor_threads_statuses_by_ip_and_port"
-    )
+    @patch("csle_common.controllers.snort_ids_controller.SnortIDSController."
+           "get_snort_idses_monitor_threads_statuses_by_ip_and_port")
     @patch("csle_collector.snort_ids_manager.query_snort_ids_manager.start_snort_ids_monitor")
     @patch("grpc.insecure_channel")
-    def test_start_snort_idses_monitor_thread(
-        self, mock_insecure_channel, mock_start_monitor, mock_get_statuses, mock_start_manager
-    ):
+    def test_start_snort_idses_monitor_thread(self, mock_insecure_channel, mock_start_monitor, mock_get_statuses,
+                                              mock_start_manager) -> None:
         """
         Test a method that sends a request to the SnortIDSManager on a specific container that runs
         an IDS to start the IDS manager and the monitor thread
@@ -335,7 +330,8 @@ def test_stop_snort_idses_monitor_thread(self, mock_insecure_channel, mock_stop_
 
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.start_snort_managers")
     @patch(
-        "csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_monitor_threads_statuses_by_ip_and_port"
+        "csle_common.controllers.snort_ids_controller.SnortIDSController."
+        "get_snort_idses_monitor_threads_statuses_by_ip_and_port"
     )
     def test_get_snort_idses_monitor_threads_statuses(self, mock_get_statuses, mock_start_managers) -> None:
         """
@@ -365,9 +361,8 @@ def test_get_snort_idses_monitor_threads_statuses(self, mock_get_statuses, mock_
 
     @patch("grpc.insecure_channel")
     @patch("csle_collector.snort_ids_manager.query_snort_ids_manager.get_snort_ids_monitor_status")
-    def test_get_snort_idses_monitor_threads_statuses_by_ip_and_port(
-        self, mock_get_status, mock_insecure_channel
-    ) -> None:
+    def test_get_snort_idses_monitor_threads_statuses_by_ip_and_port(self, mock_get_status, mock_insecure_channel) \
+            -> None:
         """
         Test a method that sends a request to the SnortIDSManager with a specific port and ip
         to get the status of the IDS monitor thread
@@ -407,14 +402,13 @@ def test_get_snort_idses_managers_ports(self) -> None:
 
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_ids_managers_ips")
     @patch("csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_managers_ports")
-    @patch(
-        "csle_common.controllers.snort_ids_controller.SnortIDSController.get_snort_idses_monitor_threads_statuses_by_ip_and_port"
-    )
+    @patch("csle_common.controllers.snort_ids_controller.SnortIDSController."
+           "get_snort_idses_monitor_threads_statuses_by_ip_and_port")
     @patch("csle_common.util.emulation_util.EmulationUtil.physical_ip_match")
     @patch("csle_collector.snort_ids_manager.snort_ids_manager_util.SnortIdsManagerUtil.snort_ids_monitor_dto_empty")
     def test_get_snort_managers_info(
-        self, mock_snort_ids_monitor_dto_empty, mock_physical_ip_match, mock_get_statuses, mock_get_ports, mock_get_ips
-    ) -> None:
+            self, mock_snort_ids_monitor_dto_empty, mock_physical_ip_match, mock_get_statuses, mock_get_ports,
+            mock_get_ips) -> None:
         """
         Test the method that extracts the information of the Snort managers for a given emulation
 
@@ -445,7 +439,6 @@ def test_get_snort_managers_info(
             logger=self.logger,
             physical_server_ip=physical_server_ip,
         )
-
         mock_get_ips.assert_called_once_with(emulation_env_config=emulation_env_config)
         mock_get_ports.assert_called_once_with(emulation_env_config=emulation_env_config)
         mock_get_statuses.assert_any_call(port=50051, ip="10.0.0.1")