Create teuthology/test/test_kill.py

Add TestFindTargets in test_kill for different
test cases of find_targets function.

Signed-off-by: Vallari Agrawal <val.agl002@gmail.com>

diff --git a/teuthology/test/test_kill.py b/teuthology/test/test_kill.py
new file mode 100644 (file)
index 0000000..21ec997
--- /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": ""}