[Findit] Implementing swarming task error detection

appengine/findit/waterfall/test/trigger_swarming_task_pipeline_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 time
 
 from model import analysis_status
 from model.wf_swarming_task import WfSwarmingTask
 from waterfall import swarming_util
 from waterfall import trigger_swarming_task_pipeline
@@ skipping 83 matching lines @@
           },
           'tags': ['master:a', 'buildername:b', 'name:a_tests'],
           'user': 'user',
       })
     self.mock(swarming_util, 'GetSwarmingTaskRequest',
               MockedGetSwarmingTaskRequest)
 
     new_request_json = {}
     def MockedTriggerSwarmingTask(new_request, *_):
       new_request_json.update(new_request.Serialize())
-      return 'new_task_id'
+      return 'new_task_id', None
     self.mock(swarming_util, 'TriggerSwarmingTask', MockedTriggerSwarmingTask)
 
     def MockedGetSwarmingTaskName(*_):
       return 'new_task_name'
     self.mock(trigger_swarming_task_pipeline, '_GetSwarmingTaskName',
               MockedGetSwarmingTaskName)
 
     master_name = 'm'
     builder_name = 'b'
     build_number = 234
@@ skipping 37 matching lines @@
     self.assertEqual(expected_new_request_json, new_request_json)
 
     swarming_task = WfSwarmingTask.Get(
         master_name, builder_name, build_number, step_name)
     self.assertIsNotNone(swarming_task)
     self.assertEqual('new_task_id', swarming_task.task_id)
     self.assertEqual(tests, swarming_task.parameters['tests'])
     self.assertEqual(
         waterfall_config.GetSwarmingSettings()['iterations_to_rerun'],
         swarming_task.parameters['iterations_to_rerun'])
+
+  def testTriggerANewSwarmingTaskError(self):
+    def MockedDownloadSwarmingTaskData(*_):
+      return [{'task_id': '1'}, {'task_id': '2'}]
+    self.mock(swarming_util, 'ListSwarmingTasksDataByTags',
+              MockedDownloadSwarmingTaskData)
+
+    def MockedGetSwarmingTaskRequest(ref_task_id, *_):
+      self.assertEqual('1', ref_task_id)
+      return SwarmingTaskRequest.Deserialize({
+          'expiration_secs': 3600,
+          'name': 'ref_task_request',
+          'parent_task_id': 'pti',
+          'priority': 25,
+          'properties': {
+              'command': 'cmd',
+              'dimensions': [{'key': 'k', 'value': 'v'}],
+              'env': [
+                  {'key': 'a', 'value': '1'},
+                  {'key': 'GTEST_SHARD_INDEX', 'value': '1'},
+                  {'key': 'GTEST_TOTAL_SHARDS', 'value': '5'},
+              ],
+              'execution_timeout_secs': 3600,
+              'extra_args': ['--flag=value', '--gtest_filter=d.f'],
+              'grace_period_secs': 30,
+              'idempotent': True,
+              'inputs_ref': {'a': 1},
+              'io_timeout_secs': 1200,
+          },
+          'tags': ['master:a', 'buildername:b', 'name:a_tests'],
+          'user': 'user',
+      })
+    self.mock(swarming_util, 'GetSwarmingTaskRequest',
+              MockedGetSwarmingTaskRequest)
+
+    new_request_json = {}
+    def MockedTriggerSwarmingTask(new_request, *_):
+      new_request_json.update(new_request.Serialize())
+      return None, {'code': swarming_util.UNKNOWN, 'message': 'error'}
+    self.mock(swarming_util, 'TriggerSwarmingTask', MockedTriggerSwarmingTask)
+
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = 234
+    step_name = 'a_tests on platform'
+    tests = ['a.b', 'a.c']
+
+    pipeline = TriggerSwarmingTaskPipeline()
+    pipeline.run(master_name, builder_name, build_number, step_name, tests)
+
+    swarming_task = WfSwarmingTask.Get(
+        master_name, builder_name, build_number, step_name)
+    self.assertIsNotNone(swarming_task)
+    self.assertEqual(swarming_task.error,
+                     {
+                         'code': swarming_util.UNKNOWN,
+                         'message': 'error'
+                     })