[Findit] Adding Try Job suspected CLs and result status to analysis

appengine/findit/waterfall/test/identify_try_job_culprit_pipeline_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 testing_utils import testing
 
 from common.git_repository import GitRepository
+from model import wf_analysis_result_status
 from model import wf_analysis_status
+from model.wf_analysis import WfAnalysis
 from model.wf_try_job import WfTryJob
 from model.wf_try_job_data import WfTryJobData
+from waterfall import identify_try_job_culprit_pipeline
 from waterfall.identify_try_job_culprit_pipeline import(
     IdentifyTryJobCulpritPipeline)
 from waterfall.try_job_type import TryJobType
 
 
 class IdentifyTryJobCulpritPipelineTest(testing.AppengineTestCase):
 
   def _MockGetChangeLog(self, revision):
     class MockedChangeLog(object):
 
@@ skipping 48 matching lines @@
         IdentifyTryJobCulpritPipeline._GetFailedRevisionFromCompileResult(
             {
                 'report': {
                     'result': {
                         'rev1': 'passed',
                         'rev2': 'failed'
                     }
                 }
             }))
 
+  def testGetResultanalysisStatusWithTryJobCulpritNotFoundUntriaged(self):
+    # Heuristic analysis provided no results, but the try job found a culprit.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.result_status = wf_analysis_result_status.NOT_FOUND_UNTRIAGED
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': {
+                'revision': 'rev1',
+                'commit_position': '1',
+                'review_url': 'url_1',
+                'repo_name': 'chromium'
+            }
+        }
+    }
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+
+    self.assertEqual(status, wf_analysis_result_status.FOUND_UNTRIAGED)
+
+  def testGetResultanalysisStatusWithTryJobCulpritNotFoundCorrect(self):
+    # Heuristic analysis found no results, which was correct. In this case, the
+    # try job result is actually a false positive.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.result_status = wf_analysis_result_status.NOT_FOUND_CORRECT
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': {
+                'revision': 'rev1',
+                'commit_position': '1',
+                'review_url': 'url_1',
+                'repo_name': 'chromium'
+            }
+        }
+    }
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+
+    self.assertEqual(status, wf_analysis_result_status.FOUND_UNTRIAGED)
+
+  def testGetResultanalysisStatusWithTryJobCulpritNotFoundIncorrect(self):
+    # Heuristic analysis found no results and was triaged to incorrect before a
+    # try job result was found. In this case the try job result should override
+    # the heuristic result.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.result_status = wf_analysis_result_status.NOT_FOUND_INCORRECT
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': {
+                'revision': 'rev1',
+                'commit_position': '1',
+                'review_url': 'url_1',
+                'repo_name': 'chromium'
+            }
+        }
+    }
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+
+    self.assertEqual(status, wf_analysis_result_status.FOUND_UNTRIAGED)
+
+  def testGetResultanalysisStatusWithTryJobCulpritNoHeuristicResult(self):
+    # In this case, the try job found a result before the heuristic result is
+    # available. This case should generally never happen, as heuristic analysis
+    # is usually much faster than try jobs.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': {
+                'revision': 'rev1',
+                'commit_position': '1',
+                'review_url': 'url_1',
+                'repo_name': 'chromium'
+            }
+        }
+    }
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+
+    self.assertEqual(status, wf_analysis_result_status.FOUND_UNTRIAGED)
+
+  def testGetResultanalysisStatusWithNoTryJobCulpritNoHeuristicResult(self):
+    # In this case, the try job completed faster than heuristic analysis
+    # (which should never happen) but no results were found.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.put()
+
+    result = {}
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+    self.assertEqual(status, wf_analysis_result_status.NOT_FOUND_UNTRIAGED)
+
+  def testGetResultanalysisStatusWithTryJobCulpritAndHeuristicResult(self):
+    # In this case, heuristic analysis found the correct culprit. The try job
+    # result should not overwrite it.
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.result_status = wf_analysis_result_status.FOUND_CORRECT
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': {
+                'revision': 'rev1',
+                'commit_position': '1',
+                'review_url': 'url_1',
+                'repo_name': 'chromium'
+            }
+        }
+    }
+
+    status = identify_try_job_culprit_pipeline._GetResultAnalysisStatus(
+        analysis, result)
+    self.assertEqual(status, wf_analysis_result_status.FOUND_CORRECT)
+
+  def testGetSuspectedCLsForCompileTryJob(self):
+    heuristic_suspected_cl = {
+        'revision': 'rev1',
+        'commit_position': '1',
+        'review_url': 'url_1',
+        'repo_name': 'chromium'
+    }
+
+    compile_suspected_cl = {
+        'revision': 'rev2',
+        'commit_position': '2',
+        'review_url': 'url_2',
+        'repo_name': 'chromium'
+    }
+
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.suspected_cls = [heuristic_suspected_cl]
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': compile_suspected_cl
+        }
+    }
+
+    self.assertEqual(
+        identify_try_job_culprit_pipeline._GetSuspectedCLs(analysis, result),
+        [heuristic_suspected_cl, compile_suspected_cl])
+
+  def testGetSuspectedCLsForTestTryJobAndHeuristicResultsSame(self):
+    suspected_cl = {
+        'revision': 'rev1',
+        'commit_position': '1',
+        'review_url': 'url_1',
+        'repo_name': 'chromium'
+    }
+
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.suspected_cls = [suspected_cl]
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'compile': suspected_cl
+        }
+    }
+
+    self.assertEqual(
+        identify_try_job_culprit_pipeline._GetSuspectedCLs(analysis, result),
+        [suspected_cl])
+
+  def testGetSuspectedCLsForTestTryJob(self):
+    suspected_cl1 = {
+        'revision': 'rev1',
+        'commit_position': '1',
+        'review_url': 'url_1',
+        'repo_name': 'chromium'
+    }
+    suspected_cl2 = {
+        'revision': 'rev2',
+        'commit_position': '2',
+        'review_url': 'url_2',
+        'repo_name': 'chromium'
+    }
+    suspected_cl3 = {
+        'revision': 'rev3',
+        'commit_position': '3',
+        'review_url': 'url_3',
+        'repo_name': 'chromium'
+    }
+
+    analysis = WfAnalysis.Create('m', 'b', 1)
+    analysis.suspected_cls = []
+    analysis.put()
+
+    result = {
+        'culprit': {
+            'a_test': {
+                'tests': {
+                    'a_test1': suspected_cl1,
+                    'a_test2': suspected_cl1
+                }
+            },
+            'b_test': {
+                'tests': {
+                    'b_test1': suspected_cl2
+                }
+            },
+            'c_test': {
+                'revision': 'rev3',
+                'commit_position': '3',
+                'review_url': 'url_3',
+                'repo_name': 'chromium',
+                'tests': {}
+            }
+        }
+    }
+
+    self.assertEqual(
+        identify_try_job_culprit_pipeline._GetSuspectedCLs(analysis, result),
+        [suspected_cl3, suspected_cl2, suspected_cl1])
+
   def testIdentifyCulpritForCompileTryJobNoCulprit(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     try_job = WfTryJob.Create(master_name, builder_name, build_number)
     try_job.put()
     try_job_data = WfTryJobData.Create(try_job_id)
     try_job_data.put()
 
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
+
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev1'],
         TryJobType.COMPILE, '1', None)
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertEqual(wf_analysis_status.ANALYZED, try_job.status)
     self.assertEqual([], try_job.compile_results)
     self.assertIsNone(culprit)
     self.assertIsNone(try_job_data.culprits)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.NOT_FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [])
 
   def testIdentifyCulpritForCompileTryJobSuccess(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     compile_result = {
         'report': {
             'result': {
@@ skipping 11 matching lines @@
     try_job.compile_results = [{
         'report': {
             'result': {
                 'rev1': 'passed',
                 'rev2': 'failed'
             },
         },
         'try_job_id': try_job_id,
     }]
     try_job.put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev1'],
         TryJobType.COMPILE, '1', compile_result)
 
     expected_culprit = 'rev2'
+    expected_suspected_cl = {
+        'revision': 'rev2',
+        'commit_position': '2',
+        'review_url': 'url_2',
+        'repo_name': 'chromium'
+    }
     expected_compile_result = {
         'report': {
             'result': {
                 'rev1': 'passed',
                 'rev2': 'failed'
             }
         },
         'try_job_id': try_job_id,
         'culprit': {
-            'compile': {
-                'revision': 'rev2',
-                'commit_position': '2',
-                'review_url': 'url_2'
-            }
+            'compile': expected_suspected_cl
         }
     }
 
     self.assertEqual(expected_compile_result['culprit'], culprit)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual(expected_compile_result, try_job.compile_results[-1])
     self.assertEqual(wf_analysis_status.ANALYZED, try_job.status)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     self.assertEqual({'compile': expected_culprit}, try_job_data.culprits)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls,
+                     [expected_suspected_cl])
 
   def testIdentifyCulpritForCompileReturnNoneIfAllPassed(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     compile_result = {
         'report': {
             'result': {
                 'rev1': 'passed',
                 'rev2': 'passed'
             }
         },
         'url': 'url',
         'try_job_id': try_job_id,
     }
 
     WfTryJobData.Create(try_job_id).put()
     WfTryJob.Create(master_name, builder_name, build_number).put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev1'],
         TryJobType.COMPILE, '1', compile_result)
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
 
     self.assertIsNone(culprit)
     self.assertEqual(wf_analysis_status.ANALYZED, try_job.status)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     self.assertIsNone(try_job_data.culprits)
 
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.NOT_FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [])
+
   def testIdentifyCulpritForTestTryJobReturnNoneIfNoTryJobResult(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     WfTryJobData.Create(try_job_id).put()
     try_job = WfTryJob.Create(master_name, builder_name, build_number)
     try_job.status = wf_analysis_status.ANALYZING
     try_job.put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev1', 'rev2'],
         TryJobType.TEST, '1', None)
 
     self.assertIsNone(culprit)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     self.assertIsNone(try_job_data.culprits)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.NOT_FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [])
 
   def testIdentifyCulpritForTestTryJobReturnNoneIfNoRevisionToCheck(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     test_result = {
         'report': {
             'result': {
                 'rev1': {
                     'a_test': {
                         'status': 'failed',
                         'valid': True,
                         'failures': ['a_test1']
                     }
                 }
             }
         },
         'url': 'url',
         'try_job_id': try_job_id
     }
 
     WfTryJobData.Create(try_job_id).put()
     try_job = WfTryJob.Create(master_name, builder_name, build_number)
     try_job.status = wf_analysis_status.ANALYZING
     try_job.put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, [], TryJobType.TEST, '1',
         test_result)
 
     self.assertIsNone(culprit)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     self.assertIsNone(try_job_data.culprits)
 
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.NOT_FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [])
+
   def testIdentifyCulpritForTestTryJobReturnRevisionIfNoCulpritInfo(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     test_result = {
         'report': {
             'result': {
                 'rev3': {
                     'a_test': {
                         'status': 'failed',
                         'valid': True,
                         'failures': ['a_test1']
                     }
                 }
             }
         },
         'url': 'url',
         'try_job_id': try_job_id
     }
 
     WfTryJobData.Create(try_job_id).put()
     try_job = WfTryJob.Create(master_name, builder_name, build_number)
     try_job.status = wf_analysis_status.ANALYZING
     try_job.put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev3'], TryJobType.TEST,
         '1', test_result)
 
+    expected_suspected_cl = {
+        'revision': 'rev3',
+        'repo_name': 'chromium'
+    }
+
     expected_culprit = {
         'a_test': {
             'tests': {
-                'a_test1': {
-                    'revision': 'rev3'
-                }
+                'a_test1': expected_suspected_cl
             }
         }
     }
     self.assertEqual(expected_culprit, culprit)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     expected_culprit_data = {
         'a_test': {
             'a_test1': 'rev3'
         }
     }
     self.assertEqual(expected_culprit_data, try_job_data.culprits)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls, [expected_suspected_cl])
 
   def testIdentifyCulpritForTestTryJobSuccess(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     try_job_id = '1'
 
     test_result = {
         'report': {
             'result': {
@@ skipping 33 matching lines @@
         },
         'url': 'url',
         'try_job_id': try_job_id
     }
 
     WfTryJobData.Create(try_job_id).put()
     try_job = WfTryJob.Create(master_name, builder_name, build_number)
     try_job.status = wf_analysis_status.ANALYZING
     try_job.test_results = [test_result]
     try_job.put()
+    analysis = WfAnalysis.Create(master_name, builder_name, build_number)
+    analysis.put()
 
     pipeline = IdentifyTryJobCulpritPipeline()
     culprit = pipeline.run(
         master_name, builder_name, build_number, ['rev1', 'rev2'],
         TryJobType.TEST, '1', test_result)
 
+    a_test1_suspected_cl = {
+        'revision': 'rev1',
+        'commit_position': '1',
+        'review_url': 'url_1',
+        'repo_name': 'chromium'
+    }
+    a_test2_suspected_cl = {
+        'revision': 'rev2',
+        'commit_position': '2',
+        'review_url': 'url_2',
+        'repo_name': 'chromium'
+    }
+    b_test1_suspected_cl = a_test1_suspected_cl
+
     expected_test_result = {
         'report': {
             'result': {
                 'rev1': {
                     'a_test': {
                         'status': 'failed',
                         'valid': True,
                         'failures': ['a_test1']
                     },
                     'b_test': {
@@ skipping 22 matching lines @@
                         'failures': []
                     }
                 }
             }
         },
         'url': 'url',
         'try_job_id': try_job_id,
         'culprit': {
             'a_test': {
                 'tests': {
-                    'a_test1': {
-                        'revision': 'rev1',
-                        'commit_position': '1',
-                        'review_url': 'url_1'
-                    },
-                    'a_test2': {
-                        'revision': 'rev2',
-                        'commit_position': '2',
-                        'review_url': 'url_2'
-                    }
+                    'a_test1': a_test1_suspected_cl,
+                    'a_test2': a_test2_suspected_cl
                 }
             },
             'b_test': {
                 'tests': {
-                    'b_test1': {
-                        'revision': 'rev1',
-                        'commit_position': '1',
-                        'review_url': 'url_1'
-                    }
+                    'b_test1': b_test1_suspected_cl
                 }
             },
             'c_test': {
                 'revision': 'rev2',
                 'commit_position': '2',
                 'review_url': 'url_2',
+                'repo_name': 'chromium',
                 'tests': {}
             }
         }
     }
 
     self.assertEqual(expected_test_result['culprit'], culprit)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual(expected_test_result, try_job.test_results[-1])
     self.assertEqual(wf_analysis_status.ANALYZED, try_job.status)
 
     try_job_data = WfTryJobData.Get(try_job_id)
     expected_culprit_data = {
         'a_test': {
             'a_test1': 'rev1',
             'a_test2': 'rev2',
         },
         'b_test': {
             'b_test1': 'rev1',
         },
         'c_test': 'rev2'
         }
     self.assertEqual(expected_culprit_data, try_job_data.culprits)
+    self.assertEqual(analysis.result_status,
+                     wf_analysis_result_status.FOUND_UNTRIAGED)
+    self.assertEqual(analysis.suspected_cls,
+                     [a_test2_suspected_cl, a_test1_suspected_cl])