[Findit] Use heuristic analysis result for test try jobs(Findit side).

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
@@ skipping 390 matching lines @@
                 {'source': 'b.cc', 'target': 'b.o'},
             ]
         }
     }
 
     self.assertEqual(
         try_job_util._GetFailedTargetsFromSignals(
             signals, 'master1', 'builder1'),
         ['b.o'])
 
-  def testGetSuspectsForCompileFailureFromHeuristicResult(self):
+  def testGetSuspectsFromHeuristicResultForCompile(self):
     heuristic_result = {
         'failures': [
             {
                 'step_name': 'compile',
                 'suspected_cls': [
                     {
                         'revision': 'r1',
                     },
                     {
                         'revision': 'r2',
                     },
                 ],
             },
-            {
-                'step_name': 'steps'
-            },
         ]
     }
     expected_suspected_revisions = ['r1', 'r2']
     self.assertEqual(
         expected_suspected_revisions,
-        try_job_util._GetSuspectsForCompileFailureFromHeuristicResult(
-            heuristic_result))
+        try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))
+
+  def testGetSuspectsFromHeuristicResultForTest(self):
+    heuristic_result = {
+        'failures': [
+            {
+                'step_name': 'step1',
+                'suspected_cls': [
+                    {
+                        'revision': 'r1',
+                    },
+                    {
+                        'revision': 'r2',
+                    },
+                ],
+            },
+            {
+                'step_name': 'step2',
+                'suspected_cls': [
+                    {
+                        'revision': 'r1',
+                    },
+                    {
+                        'revision': 'r3',
+                    },
+                ],
+            },
+        ]
+    }
+    expected_suspected_revisions = ['r1', 'r2', 'r3']
+    self.assertEqual(
+        expected_suspected_revisions,
+        try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))