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

appengine/findit/waterfall/test/try_job_util_test.py

Index: appengine/findit/waterfall/test/try_job_util_test.py
diff --git a/appengine/findit/waterfall/test/try_job_util_test.py b/appengine/findit/waterfall/test/try_job_util_test.py
index 706a8c82916c1adf8d83231a60258153893844ff..d7c671fb8b923379ec6a64a173dbbe0d4e8a217d 100644
--- a/appengine/findit/waterfall/test/try_job_util_test.py
+++ b/appengine/findit/waterfall/test/try_job_util_test.py
@@ -423,7 +423,7 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
             signals, 'master1', 'builder1'),
         ['b.o'])
 
-  def testGetSuspectsForCompileFailureFromHeuristicResult(self):
+  def testGetSuspectsFromHeuristicResultForCompile(self):
     heuristic_result = {
         'failures': [
             {
@@ -437,13 +437,41 @@ class TryJobUtilTest(wf_testcase.WaterfallTestCase):
                     },
                 ],
             },
+        ]
+    }
+    expected_suspected_revisions = ['r1', 'r2']
+    self.assertEqual(
+        expected_suspected_revisions,
+        try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))
+
+  def testGetSuspectsFromHeuristicResultForTest(self):
+    heuristic_result = {
+        'failures': [
+            {
+                'step_name': 'step1',
+                'suspected_cls': [
+                    {
+                        'revision': 'r1',
+                    },
+                    {
+                        'revision': 'r2',
+                    },
+                ],
+            },
             {
-                'step_name': 'steps'
+                'step_name': 'step2',
+                'suspected_cls': [
+                    {
+                        'revision': 'r1',
+                    },
+                    {
+                        'revision': 'r3',
+                    },
+                ],
             },
         ]
     }
-    expected_suspected_revisions = ['r1', 'r2']
+    expected_suspected_revisions = ['r1', 'r2', 'r3']
     self.assertEqual(
         expected_suspected_revisions,
-        try_job_util._GetSuspectsForCompileFailureFromHeuristicResult(
-            heuristic_result))
+        try_job_util._GetSuspectsFromHeuristicResult(heuristic_result))