Merge pull request #391 from Limmen/import

unit_test import_util

simulation-system/libs/csle-common/tests/test_grpc_util.py

@@ -14,7 +14,6 @@ def test_grpc_server_on(self, mock_channel_ready_future) -> None:
  Test utility function to test if a given gRPC channel is working or not
 
  :param mock_channel_ready_future: mock_channel_ready_future
-
  :return: None
  """
  mock_future = MagicMock()
@@ -29,7 +28,6 @@ def test_grpc_server_on_timeout(self, mock_channel_ready_future) -> None:
  Test utility function to test if a given gRPC channel is not working
 
  :param mock_channel_ready_future: mock_channel_ready_future
-
  :return: None
  """
  mock_future = MagicMock()

simulation-system/libs/csle-common/tests/test_import_util.py

@@ -0,0 +1,64 @@
+from unittest.mock import patch, MagicMock
+from csle_common.util.import_util import ImportUtil
+
+
+class TestImportUtilSuite:
+ """
+ Test suite for import_util
+ """
+
+ @patch("os.path.exists")
+ @patch("csle_common.dao.system_identification.emulation_statistics.EmulationStatistics.from_json_file")
+ @patch("csle_common.metastore.metastore_facade.MetastoreFacade.save_emulation_statistic")
+ def test_import_emulation_statistics_from_disk_json(
+ self, mock_save_emulation_statistic, mock_from_json_file, mock_path_exists
+ ) -> None:
+ """
+ Test the method that imports emulation statistics from disk to the metastore
+
+ :param mock_save_emulation_statistic: mock_save_emulation_statistic
+ :param mock_from_json_file: mock_from_json_file
+ :param mock_path_exists: mock_path_exists
+
+ :return: None
+ """
+ mock_path_exists.return_value = True
+ mock_statistics = MagicMock()
+ mock_from_json_file.return_value = mock_statistics
+ input_file = "file.json"
+ emulation_name = "test_emulation"
+ ImportUtil.import_emulation_statistics_from_disk_json(input_file=input_file, emulation_name=emulation_name)
+
+ mock_path_exists.assert_called()
+ mock_from_json_file.assert_called_once_with(input_file)
+ assert mock_statistics.emulation_name == emulation_name
+ mock_save_emulation_statistic.assert_called()
+
+ @patch("os.path.exists")
+ @patch("csle_common.dao.emulation_config.emulation_trace.EmulationTrace.load_traces_from_disk")
+ @patch("csle_common.metastore.metastore_facade.MetastoreFacade.save_emulation_trace")
+ def test_import_emulation_traces_from_disk_json(
+ self, mock_save_emulation_trace, mock_load_traces_from_disk, mock_path_exists
+ ) -> None:
+ """
+ Test the method that imports emulation traces from disk to the metastore
+
+ :param mock_save_emulation_trace: mock_save_emulation_trac
+ :param mock_load_traces_from_disk: mock_load_traces_from_disk
+ :param mock_path_exists: mock_path_exists
+
+ :return: None
+ """
+ mock_path_exists.return_value = True
+ mock_trace_1 = MagicMock()
+ mock_trace_2 = MagicMock()
+ mock_load_traces_from_disk.return_value = [mock_trace_1, mock_trace_2]
+ input_file = "file.json"
+ emulation_name = "test_emulation"
+ ImportUtil.import_emulation_traces_from_disk_json(input_file=input_file, emulation_name=emulation_name)
+
+ mock_path_exists.assert_called()
+ mock_load_traces_from_disk.assert_called()
+ assert mock_trace_1.emulation_name == emulation_name
+ assert mock_trace_2.emulation_name == emulation_name
+ assert mock_save_emulation_trace.call_count == 2