Findit] Flake Checker: Pipeline to trigger try jobs to identify flake culprits

appengine/findit/waterfall/flake/test/process_flake_try_job_result_pipeline_test.py

Index: appengine/findit/waterfall/flake/test/process_flake_try_job_result_pipeline_test.py
diff --git a/appengine/findit/waterfall/flake/test/process_flake_try_job_result_pipeline_test.py b/appengine/findit/waterfall/flake/test/process_flake_try_job_result_pipeline_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..56770e3d909ac60efdd8a983ea8a2187129f61e7
--- /dev/null
+++ b/appengine/findit/waterfall/flake/test/process_flake_try_job_result_pipeline_test.py
@@ -0,0 +1,114 @@
+# Copyright 2016 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 gae_libs.testcase import TestCase
+
+from model.flake.flake_try_job import FlakeTryJob
+from model.flake.master_flake_analysis import MasterFlakeAnalysis
+from waterfall.flake.process_flake_try_job_result_pipeline import (
+    ProcessFlakeTryJobResultPipeline)
+
+
+class ProcessFlakeTryJobResultPipelineTest(TestCase):
+
+  def testProcessFlakeTryJobResultPipeline(self):
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = 100
+    step_name = 's'
+    test_name = 't'
+    revision = 'r4'
+    commit_position = 4
+    try_job_id = 'try_job_id'
+    url = 'url'
+
+    try_job_result = {
+        'report': {
+            'result': {
+                revision: {
+                    step_name: {
+                        'status': 'failed',
+                        'failures': [test_name],
+                        'valid': True,
+                        'pass_fail_counts': {
+                            test_name: {
+                                'pass_count': 20,
+                                'fail_count': 80
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    analysis = MasterFlakeAnalysis.Create(
+        master_name, builder_name, build_number, step_name, test_name)
+    analysis.Save()
+
+    try_job = FlakeTryJob.Create(
+        master_name, builder_name, step_name, test_name, revision)
+    try_job.flake_results = [{
+        'url': url,
+        'report': try_job_result,
+        'try_job_id': try_job_id
+    }]
+    try_job.try_job_ids = [try_job_id]
+    try_job.put()
+
+    ProcessFlakeTryJobResultPipeline().run(
+        revision, commit_position, try_job_result, try_job.key.urlsafe(),
+        analysis.key.urlsafe())
+
+    resulting_data_point = analysis.data_points[-1]
+    self.assertEqual(0.2, resulting_data_point.pass_rate)
+    self.assertEqual(commit_position, resulting_data_point.commit_position)
+    self.assertEqual(url, resulting_data_point.try_job_url)
+
+  def testProcessFlakeTryJobResultPipelineTestDoesNotExist(self):
+    master_name = 'm'
+    builder_name = 'b'
+    build_number = 100
+    step_name = 's'
+    test_name = 't'
+    revision = 'r4'
+    commit_position = 4
+    try_job_id = 'try_job_id'
+    url = 'url'
+
+    try_job_result = {
+        'report': {
+            'result': {
+                revision: {
+                    step_name: {
+                        'status': 'skipped',
+                        'valid': True
+                    }
+                }
+            }
+        }
+    }
+
+    analysis = MasterFlakeAnalysis.Create(
+        master_name, builder_name, build_number, step_name, test_name)
+    analysis.Save()
+
+    try_job = FlakeTryJob.Create(
+        master_name, builder_name, step_name, test_name, revision)
+    try_job.flake_results = [{
+        'url': url,
+        'report': try_job_result,
+        'try_job_id': try_job_id
+    }]
+    try_job.try_job_ids = [try_job_id]
+    try_job.put()
+
+    ProcessFlakeTryJobResultPipeline().run(
+        revision, commit_position, try_job_result, try_job.key.urlsafe(),
+        analysis.key.urlsafe())
+
+    resulting_data_point = analysis.data_points[-1]
+    self.assertEqual(-1, resulting_data_point.pass_rate)
+    self.assertEqual(commit_position, resulting_data_point.commit_position)
+    self.assertEqual(url, resulting_data_point.try_job_url)