class TestTreeherderApiGetAllJobsByStatus(unittest.TestCase): """Test TreeherderApi fetch all jobs(buildernames for now) by job's status.""" def setUp(self): self.query_api = TreeherderApi() self.repo_name = 'repo_mock' self.revision = 'revision_mock' @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % { 'result': "success", 'state': "completed" })) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=SUCCESS) def test_successful_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a successful job.""" self.assertEqual( self.query_api.find_all_jobs_by_status(self.repo_name, self.revision, SUCCESS), [ "Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6" ]) @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % { 'result': "testfailed", 'state': "completed" })) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=FAILURE) def test_failed_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a failed job.""" self.assertEqual( self.query_api.find_all_jobs_by_status(self.repo_name, self.revision, FAILURE), [ "Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6" ]) @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % { 'result': "unknown", 'state': "pending" })) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=PENDING) def test_pending_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a pending job.""" self.assertEqual( self.query_api.find_all_jobs_by_status(self.repo_name, self.revision, PENDING), [ "Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6" ])
class TestTreeherderApiGetAllJobsByStatus(unittest.TestCase): """Test TreeherderApi fetch all jobs(buildernames for now) by job's status.""" def setUp(self): self.query_api = TreeherderApi() self.repo_name = 'repo_mock' self.revision = 'revision_mock' @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % {'result': "success", 'state': "completed"})) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=SUCCESS) def test_successful_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a successful job.""" self.assertEqual(self.query_api.find_all_jobs_by_status( self.repo_name, self.revision, SUCCESS), ["Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6"]) @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % {'result': "testfailed", 'state': "completed"})) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=FAILURE) def test_failed_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a failed job.""" self.assertEqual(self.query_api.find_all_jobs_by_status( self.repo_name, self.revision, FAILURE), ["Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6"]) @patch('mozci.query_jobs.TreeherderApi.get_all_jobs', return_value=json.loads(MOCK_JOBS % {'result': "unknown", 'state': "pending"})) @patch('mozci.query_jobs.TreeherderApi.get_job_status', return_value=PENDING) def test_pending_job(self, get_all_jobs, get_job_status): """Test TreeherderApi find_all_jobs_by_status with a pending job.""" self.assertEqual(self.query_api.find_all_jobs_by_status( self.repo_name, self.revision, PENDING), ["Ubuntu VM 12.04 x64 mozilla-inbound opt test mochitest-1", "[TC] - Linux64 web-platform-tests-e10s-6"])