[Findit] Add link to swarming tasks.

appengine/findit/handlers/flake/test/check_flake_test.py

 # Copyright 2016 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 datetime
 import mock
 import re
 
 import webapp2
 import webtest
@@ skipping 66 matching lines @@
     build_number = '123'
     step_name = 's'
     test_name = 't'
     success_rate = .9
 
     analysis = MasterFlakeAnalysis.Create(
         master_name, builder_name, build_number, step_name, test_name)
     data_point = DataPoint()
     data_point.build_number = int(build_number)
     data_point.pass_rate = success_rate
+    data_point.task_id = '1'
     analysis.data_points.append(data_point)
     analysis.status = analysis_status.COMPLETED
     analysis.suspected_flake_build_number = 100
     analysis.request_time = datetime.datetime(2016, 10, 01, 12, 10, 00)
     analysis.start_time = datetime.datetime(2016, 10, 01, 12, 10, 05)
     analysis.end_time = datetime.datetime(2016, 10, 01, 13, 10, 00)
     analysis.algorithm_parameters = {'iterations_to_rerun': 100}
     analysis.Save()
 
     response = self.test_app.get('/waterfall/check-flake', params={
         'key': analysis.key.urlsafe(),
         'format': 'json'})
 
+    task_url = '%s/task?id=%s' % (
+      check_flake.SWARMING_TASK_BASE_URL, data_point.task_id)
     expected_check_flake_result = {
-        'pass_rates': [[int(build_number), success_rate]],
+        'pass_rates': [[int(build_number), success_rate, task_url]],
         'analysis_status': STATUS_TO_DESCRIPTION.get(analysis.status),
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': int(build_number),
         'step_name': step_name,
         'test_name': test_name,
         'request_time': '2016-10-01 12:10:00 UTC',
         'task_number': 1,
         'error': None,
         'iterations_to_rerun': 100,
@@ skipping 67 matching lines @@
 
     response = self.test_app.get('/waterfall/check-flake', params={
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'step_name': step_name,
         'test_name': test_name,
         'format': 'json'})
 
     expected_check_flake_result = {
-        'pass_rates': [[build_number - 1, success_rate]],
+        'pass_rates': [[build_number - 1, success_rate, None]],
         'analysis_status': STATUS_TO_DESCRIPTION.get(previous_analysis.status),
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number - 1,
         'step_name': step_name,
         'test_name': test_name,
         'request_time': '2016-10-01 12:10:00 UTC',
         'task_number': 1,
         'error': None,
         'iterations_to_rerun': 100,
@@ skipping 101 matching lines @@
             'm', 'b', '1', None, 't', '').get('data', {}).get('error_message'),
         'Step name must be specified')
     self.assertEqual(
         CheckFlake()._ValidateInput(
             'm', 'b', '1', 's', None, '').get('data', {}).get('error_message'),
         'Test name must be specified')
     self.assertEqual(
         CheckFlake()._ValidateInput(
             'm', 'b', '1', 's', 't', 'a').get('data', {}).get('error_message'),
         'Bug id (optional) must be an int')