From 9fa9399c999bc67c6a505738c0a87e5500836a01 Mon Sep 17 00:00:00 2001 From: Lianjun Zhang Atlas Date: Tue, 29 Apr 2025 11:48:46 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0dt?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../tests/ut/api/test_taskd_wokrer_api.py | 30 +++++++++++++++++-- 1 file changed, 27 insertions(+), 3 deletions(-) diff --git a/component/taskd/tests/ut/api/test_taskd_wokrer_api.py b/component/taskd/tests/ut/api/test_taskd_wokrer_api.py index 2a5e9de1e..49bbac856 100644 --- a/component/taskd/tests/ut/api/test_taskd_wokrer_api.py +++ b/component/taskd/tests/ut/api/test_taskd_wokrer_api.py @@ -15,13 +15,16 @@ # limitations under the License. # ============================================================================== import unittest -from unittest.mock import patch +from unittest.mock import patch, MagicMock -from taskd.api.taskd_worker_api import init_taskd_worker +from taskd.api.taskd_worker_api import init_taskd_worker, start_taskd_worker + +taskd_worker = None +run_log = MagicMock() class WorkerTestCase(unittest.TestCase): - def test_init_taskd_worker_success(self, mock_worker): + def test_init_taskd_worker_success(self): rank_id = 'not_an_int' upper_limit = 5000 result = init_taskd_worker(rank_id, upper_limit) @@ -39,6 +42,27 @@ class WorkerTestCase(unittest.TestCase): result = init_taskd_worker(rank_id, upper_limit) self.assertFalse(result) + def test_worker_not_initialized(self): + result = start_taskd_worker() + self.assertEqual(result, False) + + @patch('taskd.api.taskd_worker_api.taskd_worker') + def test_worker_start_success(self, mock_worker): + global taskd_worker + taskd_worker = mock_worker + mock_worker.start.return_value = True + result = start_taskd_worker() + self.assertEqual(result, True) + mock_worker.start.assert_called_once() + + @patch('taskd.api.taskd_worker_api.taskd_worker') + def test_worker_start_failure(self, mock_worker): + global taskd_worker + taskd_worker = mock_worker + mock_worker.start.side_effect = Exception("Test exception") + result = start_taskd_worker() + self.assertEqual(result, False) + if __name__ == '__main__': unittest.main() -- Gitee