| Index: dashboard/dashboard/start_try_job_test.py
|
| diff --git a/dashboard/dashboard/start_try_job_test.py b/dashboard/dashboard/start_try_job_test.py
|
| index a4d9ff3e80e16dcf2e6ff3d31ab820c82a9665e1..2af159ec04fb8c4e6ecbdf36e26f9490a1bd9288 100644
|
| --- a/dashboard/dashboard/start_try_job_test.py
|
| +++ b/dashboard/dashboard/start_try_job_test.py
|
| @@ -13,10 +13,12 @@ import webtest
|
| from google.appengine.ext import ndb
|
|
|
| from dashboard import can_bisect
|
| +from dashboard import issue_tracker_service
|
| from dashboard import namespaced_stored_object
|
| from dashboard import rietveld_service
|
| from dashboard import start_try_job
|
| from dashboard import testing_common
|
| +from dashboard import utils
|
| from dashboard.models import bug_data
|
| from dashboard.models import graph_data
|
| from dashboard.models import try_job
|
| @@ -309,6 +311,8 @@ def _MockMakeRequest(path, *args, **kwargs): # pylint: disable=unused-argument
|
| assert False, 'Invalid url %s requested!' % path
|
|
|
|
|
| +@mock.patch('apiclient.discovery.build', mock.MagicMock())
|
| +@mock.patch.object(utils, 'ServiceAccountCredentials', mock.MagicMock())
|
| class StartBisectTest(testing_common.TestCase):
|
|
|
| def setUp(self):
|
| @@ -732,7 +736,9 @@ class StartBisectTest(testing_common.TestCase):
|
|
|
| @mock.patch.object(start_try_job.buildbucket_service, 'PutJob',
|
| mock.MagicMock(return_value='1234567'))
|
| - def testPerformBuildbucketBisect(self):
|
| + @mock.patch.object(
|
| + issue_tracker_service.IssueTrackerService, 'AddBugComment')
|
| + def testPerformBuildbucketBisect(self, add_bug_comment_mock):
|
| self.SetCurrentUser('foo@chromium.org')
|
|
|
| bug_data.Bug(id=12345).put()
|
| @@ -757,6 +763,9 @@ class StartBisectTest(testing_common.TestCase):
|
| try_job.TryJob.buildbucket_job_id == '1234567').fetch()
|
| self.assertEqual(1, len(job_entities))
|
| self.assertTrue(job_entities[0].use_buildbucket)
|
| + add_bug_comment_mock.assert_called_once_with(
|
| + 12345, 'Started bisect job https://None/buildbucket_job_status/1234567',
|
| + send_email=False)
|
|
|
| def testPerformBisect_InvalidConfig_ReturnsError(self):
|
| bisect_job = try_job.TryJob(
|
| @@ -770,13 +779,14 @@ class StartBisectTest(testing_common.TestCase):
|
| {'error': 'No "recipe_tester_name" given.'},
|
| start_try_job.PerformBisect(bisect_job))
|
|
|
| + @mock.patch.object(issue_tracker_service.IssueTrackerService, 'AddBugComment')
|
| @mock.patch(
|
| 'google.appengine.api.urlfetch.fetch',
|
| mock.MagicMock(side_effect=_MockFetch))
|
| @mock.patch.object(
|
| start_try_job.rietveld_service.RietveldService, 'MakeRequest',
|
| mock.MagicMock(side_effect=_MockMakeRequest))
|
| - def testPerformBisect(self):
|
| + def testPerformBisect(self, _):
|
| self.SetCurrentUser('foo@chromium.org')
|
|
|
| # Create bug.
|
| @@ -832,13 +842,14 @@ class StartBisectTest(testing_common.TestCase):
|
| response = self.testapp.post('/start_try_job', query_parameters)
|
| self.assertEqual(json.dumps({'issue_id': '33001'}), response.body)
|
|
|
| + @mock.patch.object(issue_tracker_service.IssueTrackerService, 'AddBugComment')
|
| @mock.patch(
|
| 'google.appengine.api.urlfetch.fetch',
|
| mock.MagicMock(side_effect=_MockFailedFetch))
|
| @mock.patch.object(
|
| start_try_job.rietveld_service.RietveldService, 'MakeRequest',
|
| mock.MagicMock(side_effect=_MockMakeRequest))
|
| - def testPerformBisectStep_DeleteJobOnFailedBisect(self):
|
| + def testPerformBisectStep_DeleteJobOnFailedBisect(self, _):
|
| self.SetCurrentUser('foo@chromium.org')
|
| query_parameters = {
|
| 'bisect_bot': 'linux_perf_bisect',
|
| @@ -894,7 +905,9 @@ class StartBisectTest(testing_common.TestCase):
|
| mock.MagicMock(return_value='my-dashboard.appspot.com'))
|
| @mock.patch.object(start_try_job.buildbucket_service, 'PutJob',
|
| mock.MagicMock(return_value='1234567'))
|
| - def testPerformBisectWithArchive(self):
|
| + @mock.patch.object(
|
| + issue_tracker_service.IssueTrackerService, 'AddBugComment')
|
| + def testPerformBisectWithArchive(self, _):
|
| self.SetCurrentUser('foo@chromium.org')
|
|
|
| # Create bug.
|
|
|