Add test cases

Signed-off-by: Zelin Hao <zelinhao@amazon.com>
zelinh · Feb 9, 2024 · 33ccaba · 33ccaba
1 parent b985877
commit 33ccaba
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/src/test_workflow/test_recorder/test_recorder.py b/src/test_workflow/test_recorder/test_recorder.py
@@ -145,9 +145,9 @@ def save_test_result_data(self, test_result_data: TestResultData) -> None:
         self.parent_class._generate_std_files(test_result_data.stdout, test_result_data.stderr, dest_directory)
         self.parent_class._copy_log_files(test_result_data.log_files, dest_directory)
 
-
-    def generate_component_yml(self, test_result_data: TestResultData):
+    def generate_component_yml(self, test_result_data: TestResultData) -> None:
         dest_directory = self.parent_class._create_base_folder_structure(test_result_data.component_name, test_result_data.component_test_config)
         self.parent_class._generate_yml(test_result_data, dest_directory)
 
+
 TestRecorder.__test__ = False  # type:ignore
diff --git a/...s/tests_test_workflow/test_integ_workflow/integ_test/test_integ_test_runner_opensearch.py b/...s/tests_test_workflow/test_integ_workflow/integ_test/test_integ_test_runner_opensearch.py
@@ -69,7 +69,7 @@ def test_with_integ_test(self, mock_temp: Mock, mock_test_recorder: Mock, mock_s
         self.assertEqual(results["sql"], mock_test_results)
 
         mock_suite_object.result_data.__iter__.assert_called()
-        mock_test_recorder_object.test_results_logs.save_test_result_data.assert_called()
+        mock_test_recorder_object.test_results_logs.generate_component_yml.assert_called()
 
         mock_suite.assert_called_once_with(
             mock_properties_object.dependency_installer,

diff --git a/...t_workflow/test_integ_workflow/integ_test/test_integ_test_runner_opensearch_dashboards.py b/...t_workflow/test_integ_workflow/integ_test/test_integ_test_runner_opensearch_dashboards.py
@@ -76,7 +76,7 @@ def test_with_integ_test(self, mock_temp: Mock, mock_test_recorder: Mock, mock_s
         self.assertEqual(results["sql"], mock_test_results)
 
         mock_suite_object.result_data.__iter__.assert_called()
-        mock_test_recorder_object.test_results_logs.save_test_result_data.assert_called()
+        mock_test_recorder_object.test_results_logs.generate_component_yml.assert_called()
 
         mock_suite.assert_called_once_with(
             mock_properties_dependency_object.dependency_installer,

diff --git a/tests/tests_test_workflow/test_recorder/test_test_recorder.py b/tests/tests_test_workflow/test_recorder/test_test_recorder.py
@@ -191,4 +191,7 @@ def test(self) -> None:
         logs.save_test_result_data(mock_test_result_data)
 
         mock_parent_class._copy_log_files.assert_called_once_with(mock_test_result_data.log_files, dest_directory)
+
+        logs.generate_component_yml(mock_test_result_data)
+
         mock_parent_class._generate_yml.assert_called_once_with(mock_test_result_data, dest_directory)