catapult/dashboard/dashboard/bisect_fyi_test.py

# Copyright 2015 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest

import mock
import webapp2
import webtest

from dashboard import bisect_fyi
from dashboard import issue_tracker_service
from dashboard import stored_object
from dashboard import testing_common
from dashboard import utils

TEST_FYI_CONFIGS = {
    'positive_culprit': {
        'bisect_config': {
            'bad_revision': '357672',
            'bug_id': 111,
            'command': ('python src/tools/perf/run_benchmark -v '
                        '--browser=release_x64 --output-format=chartjson '
                        '--upload-results --also-run-disabled-tests '
                        'blink_perf.bindings'),
            'good_revision': '357643',
            'gs_bucket': 'chrome-perf',
            'max_time_minutes': '20',
            'metric': 'create-element/create-element',
            'recipe_tester_name': 'win_x64_perf_bisect',
            'repeat_count': '10',
            'test_type': 'perf'
        },
        'expected_results': {
            'status': ['completed'],
            'culprit_data': {'cl': ['2a1781d64d']},
        }
    },
    'early_abort': {
        'bisect_config': {
            'bad_revision': '257672',
            'bug_id': 222,
            'command': ('python src/tools/perf/run_benchmark -v '
                        '--browser=release_x64 --output-format=chartjson '
                        '--upload-results --also-run-disabled-tests '
                        'blink_perf.bindings'),
            'good_revision': '257643',
            'gs_bucket': 'chrome-perf',
            'max_time_minutes': '20',
            'metric': 'create-element/create-element',
            'recipe_tester_name': 'win_x64_perf_bisect',
            'repeat_count': '10',
            'test_type': 'perf'
        },
        'expected_results': {
            'status': ['aborted'],
        }
    },
}


@mock.patch('apiclient.discovery.build', mock.MagicMock())
@mock.patch.object(utils, 'ServiceAccountHttp', mock.MagicMock())
class BisectFYITest(testing_common.TestCase):

  def setUp(self):
    super(BisectFYITest, self).setUp()
    app = webapp2.WSGIApplication(
        [('/bisect_fyi', bisect_fyi.BisectFYIHandler)])
    self.testapp = webtest.TestApp(app)
    stored_object.Set(
        bisect_fyi._BISECT_FYI_CONFIGS_KEY, TEST_FYI_CONFIGS)
    testing_common.SetIsInternalUser('internal@chromium.org', True)
    self.SetCurrentUser('internal@chromium.org')

  @mock.patch.object(
      issue_tracker_service.IssueTrackerService, 'AddBugComment')
  @mock.patch.object(bisect_fyi.start_try_job, '_PerformBuildbucketBisect')
  def testPost_FailedJobs_BisectFYI(self, mock_perform_bisect, _):
    mock_perform_bisect.return_value = {'error': 'PerformBisect Failed'}
    self.testapp.post('/bisect_fyi')
    messages = self.mail_stub.get_sent_messages()
    self.assertEqual(1, len(messages))

  @mock.patch.object(
      issue_tracker_service.IssueTrackerService, 'AddBugComment')
  @mock.patch.object(bisect_fyi.start_try_job, '_PerformBuildbucketBisect')
  def testPost_SuccessJobs_BisectFYI(self, mock_perform_bisect, mock_comment):
    mock_perform_bisect.return_value = {'issue_id': 'http://fake'}
    self.testapp.post('/bisect_fyi')
    messages = self.mail_stub.get_sent_messages()
    self.assertEqual(0, len(messages))
    mock_comment.assert_called_with(222, mock.ANY, send_email=False)


if __name__ == '__main__':
  unittest.main()