[Findit] Add a sub-pipeline to detect first-known failure.

appengine/findit/waterfall/test/build_failure_analysis_pipelines_test.py

 # Copyright (c) 2014 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 os
+
 from pipeline_utils.appengine_third_party_pipeline_src_pipeline import handlers
 from testing_utils import testing
 
-from model.build import Build
+from model.build_analysis import BuildAnalysis
 from model.build_analysis_status import BuildAnalysisStatus
 from waterfall import build_failure_analysis_pipelines
+from waterfall import buildbot
 
 
 class BuildFailureAnalysisPipelinesTest(testing.AppengineTestCase):
   app_module = handlers._APP
 
-  def _CreateAndSaveBuild(self, master_name, builder_name, build_number,
-                          analysis_status):
-    build = Build.CreateBuild(master_name, builder_name, build_number)
-    build.analysis_status = analysis_status
-    build.put()
+  def _CreateAndSaveBuildAnanlysis(
+      self, master_name, builder_name, build_number, status):
+    analysis = BuildAnalysis.CreateBuildAnalysis(
+        master_name, builder_name, build_number)
+    analysis.status = status
+    analysis.put()
 
   def testAnanlysIsNeededWhenBuildWasNeverAnalyzed(self):
     master_name = 'm'
     builder_name = 'b 1'
     build_number = 123
 
-    _, need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
+    need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
         master_name, builder_name, build_number, False)
 
     self.assertTrue(need_analysis)
 
   def testNewAnalysisIsNotNeededWhenNotForced(self):
     master_name = 'm'
     builder_name = 'b 1'
     build_number = 123
-    self._CreateAndSaveBuild(master_name, builder_name, build_number,
+    self._CreateAndSaveBuildAnanlysis(master_name, builder_name, build_number,
                              BuildAnalysisStatus.ANALYZED)
 
-    build, need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
+    need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
         master_name, builder_name, build_number, False)
 
     self.assertFalse(need_analysis)
-    self.assertEqual(BuildAnalysisStatus.ANALYZED, build.analysis_status)
 
   def testNewAnanlysIsNeededWhenForced(self):
     master_name = 'm'
     builder_name = 'b 1'
     build_number = 123
-    self._CreateAndSaveBuild(master_name, builder_name, build_number,
+    self._CreateAndSaveBuildAnanlysis(master_name, builder_name, build_number,
                              BuildAnalysisStatus.ANALYZED)
 
-    _, need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
+    need_analysis = build_failure_analysis_pipelines.NeedANewAnalysis(
         master_name, builder_name, build_number, True)
 
     self.assertTrue(need_analysis)
 
-  def testAnalysisScheduled(self):
+  def testSuccessfulAnalysisOfBuildFailure(self):
     master_name = 'm'
-    builder_name = 'b 1'
+    builder_name = 'b'
     build_number = 123
 
-    build_failure_analysis_pipelines.ScheduleAnalysisIfNeeded(
-        master_name, builder_name, build_number, False, 'default')
+    def _MockFunc(*_):
+      return True
 
-    self.execute_queued_tasks()
+    original_func = build_failure_analysis_pipelines._WaitUntilDownloadAllowed
+    try:
+      build_failure_analysis_pipelines._WaitUntilDownloadAllowed = _MockFunc
 
-    build = Build.GetBuild(master_name, builder_name, build_number)
-    self.assertIsNotNone(build)
-    self.assertEqual(BuildAnalysisStatus.ANALYZED, build.analysis_status)
+      with self.mock_urlfetch() as urlfetch:
+        for i in range(3):
+          file_name = os.path.join(os.path.dirname(__file__), 'data',
+                                   'm_b_%s.json' % (build_number - i))
+          with open(file_name, 'r') as f:
+            urlfetch.register_handler(
+                buildbot.CreateBuildUrl(
+                    master_name, builder_name, build_number - i, json_api=True),
+                f.read())

[qyearsley, 2015/01/07 06:59:46]

It's not very clear what's happening here --
1) Maybe you could extract a helper function?
2) What does urlfetch.register_handler do?

[stgao, 2015/01/09 01:28:23]

I add a comment about the purpose of the code.
urlfetch.register_handler is to map a url to the dummy data for HTTP response.

+
+      build_failure_analysis_pipelines.ScheduleAnalysisIfNeeded(
+          master_name, builder_name, build_number, False, 'default')
+
+      self.execute_queued_tasks()
+
+      analysis = BuildAnalysis.GetBuildAnalysis(
+          master_name, builder_name, build_number)
+      self.assertIsNotNone(analysis)
+      self.assertEqual(BuildAnalysisStatus.ANALYZED, analysis.status)
+    finally:
+      build_failure_analysis_pipelines._WaitUntilDownloadAllowed = original_func