[Findit] Fixing when try job entities are created

appengine/findit/waterfall/test/try_job_util_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.
 
+from common.waterfall import failure_type
 from model import analysis_status
 from model.wf_try_job import WfTryJob
 from waterfall import try_job_util
 from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
 class _MockRootPipeline(object):
   STARTED = False
 
@@ skipping 60 matching lines @@
     master_name = 'master2'
     builder_name = 'builder2'
     build_number = 223
     failure_info = {
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'failed_steps': {
             'a_test': {}
         },
+        'failure_type': failure_type.TEST
     }
 
     failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
         failure_info, None, None)
 
     self.assertEqual({}, failure_result_map)
 
   def testNotNeedANewTryJobIfNotFirstTimeFailure(self):
     master_name = 'm'
     builder_name = 'b'
@@ skipping 19 matching lines @@
                 'chromium_revision': '221-2'
             },
             '222': {
                 'blame_list': ['222-1'],
                 'chromium_revision': '222-1'
             },
             '223': {
                 'blame_list': ['223-1', '223-2', '223-3'],
                 'chromium_revision': '223-3'
             }
-        }
+        },
+        'failure_type': failure_type.COMPILE
     }
 
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     try_job_util.ScheduleTryJobIfNeeded(failure_info, None, None)
 
     self.assertFalse(_MockRootPipeline.STARTED)
@@ skipping 213 matching lines @@
         },
         'builds': {
             '222': {
                 'blame_list': ['222-1'],
                 'chromium_revision': '222-1'
             },
             '223': {
                 'blame_list': ['223-1', '223-2', '223-3'],
                 'chromium_revision': '223-3'
             }
-        }
+        },
+        'failure_type': failure_type.COMPILE
     }
 
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     try_job_util.ScheduleTryJobIfNeeded(failure_info, None, None)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
@@ skipping 52 matching lines @@
             {
                 'step_name': 'steps'
             },
         ]
     }
     expected_suspected_revisions = ['r1', 'r2']
     self.assertEqual(
         expected_suspected_revisions,
         try_job_util._GetSuspectsForCompileFailureFromHeuristicResult(
             heuristic_result))