[Findit] Pipeline change to save suspected cls to data store.

appengine/findit/waterfall/test/build_failure_analysis_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 datetime import datetime
 
 from common.blame import Blame
 from common.blame import Region
 from common.change_log import FileChangeInfo
 from common.diff import ChangeType
 from common.git_repository import GitRepository
+from common.waterfall import failure_type
 from waterfall import build_failure_analysis
 from waterfall.failure_signal import FailureSignal
 from waterfall.test import wf_testcase
 
 
 class BuildFailureAnalysisTest(wf_testcase.WaterfallTestCase):
 
   def _MockGetChangeLog(self, revision):
 
     class MockChangeLog(object):
@@ skipping 526 matching lines @@
         'failed': True,
         'chromium_revision': None,
     }
     result = build_failure_analysis.AnalyzeBuildFailure(
         failure_info, change_logs=None, deps_info=None, failure_signals=None)
     self.assertEqual(0, len(result['failures']))
 
   def testAnalyzeBuildFailure(self):
     failure_info = {
         'master_name': 'm',
+        'builder_name': 'b',
+        'build_number': 99,
+        'failure_type': failure_type.TEST,
         'failed': True,
         'chromium_revision': 'r99_2',
         'failed_steps': {
             'a': {
                 'current_failure': 99,
                 'first_failure': 98,
             },
             'b': {
                 'current_failure': 99,
                 'first_failure': 98,
@@ skipping 130 matching lines @@
 
     analysis_result = build_failure_analysis.AnalyzeBuildFailure(
         failure_info, change_logs, deps_info, failure_signals_json)
     self.assertEqual(expected_analysis_result, analysis_result)
 
   def testAnalyzeBuildFailureTestLevel(self):
     failure_info = {
         'failed': True,
         'chromium_revision': 'r99_2',
         'master_name': 'm',
+        'builder_name': 'b',
+        'build_number': 99,
+        'failure_type': failure_type.TEST,
         'failed_steps': {
             'a': {
                 'current_failure': 99,
                 'first_failure': 98,
             },
             'b': {
                 'current_failure': 99,
                 'first_failure': 98,
                 'last_pass': 96,
                 'list_isolated_data': [
@@ skipping 258 matching lines @@
     }
 
     analysis_result = build_failure_analysis.AnalyzeBuildFailure(
         failure_info, change_logs, deps_info, failure_signals_json)
 
     self.assertEqual(expected_analysis_result, analysis_result)
 
   def testAnalyzeBuildFailureForUnsupportedStep(self):
     failure_info = {
         'master_name': 'master1',
+        'builder_name': 'b',
+        'build_number': 99,
+        'failure_type': failure_type.TEST,
         'failed': True,
         'chromium_revision': 'r99_2',
         'failed_steps': {
             'unsupported_step1': {
                 'current_failure': 99,
                 'first_failure': 98,
             },
         },
         'builds': {
             '99': {
@@ skipping 145 matching lines @@
         touched_file, file_path_in_log, justification, file_name_occurrences,
         line_numbers, repo_info, commit_revision)
 
     expected_justification = {
         'score': 4,
         'hints': {
             'modified c.cc[1, 3] (and it was in log)': 4
         }
     }
     self.assertEqual(expected_justification, justification.ToDict())