From a8f6f4b8d42f3ddc2952cc84de81e32a820c5650 Mon Sep 17 00:00:00 2001 From: Vallari Agrawal Date: Mon, 8 Jul 2024 18:46:34 +0530 Subject: [PATCH] Create teuthology/test/test_kill.py Add TestFindTargets in test_kill for different test cases of find_targets function. Signed-off-by: Vallari Agrawal --- teuthology/test/test_kill.py | 46 ++++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) create mode 100644 teuthology/test/test_kill.py diff --git a/teuthology/test/test_kill.py b/teuthology/test/test_kill.py new file mode 100644 index 0000000000..21ec997185 --- /dev/null +++ b/teuthology/test/test_kill.py @@ -0,0 +1,46 @@ +from unittest.mock import patch + +from teuthology.kill import find_targets + + +class TestFindTargets(object): + """ Tests for teuthology.kill.find_targets """ + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_missing_run_find_targets(self, m_get_jobs): + m_get_jobs.return_value = [] + run_targets = find_targets("run-name") + assert run_targets == {} + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_missing_job_find_targets(self, m_get_jobs): + m_get_jobs.return_value = {} + job_targets = find_targets("run-name", "3") + assert job_targets == {} + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_missing_run_targets_find_targets(self, m_get_jobs): + m_get_jobs.return_value = [{"targets": None, "status": "waiting"}] + run_targets = find_targets("run-name") + assert run_targets == {} + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_missing_job_targets_find_targets(self, m_get_jobs): + m_get_jobs.return_value = {"targets": None} + job_targets = find_targets("run-name", "3") + assert job_targets == {} + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_run_find_targets(self, m_get_jobs): + m_get_jobs.return_value = [{"targets": {"node1": ""}, "status": "running"}] + run_targets = find_targets("run-name") + assert run_targets == {"node1": ""} + m_get_jobs.return_value = [{"targets": {"node1": ""}}] + run_targets = find_targets("run-name") + assert run_targets == {} + + @patch('teuthology.kill.report.ResultsReporter.get_jobs') + def test_job_find_targets(self, m_get_jobs): + m_get_jobs.return_value = {"targets": {"node1": ""}} + job_targets = find_targets("run-name", "3") + assert job_targets == {"node1": ""} -- 2.39.5