[Findit] Refactoring monitor swarming task pipelines

appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py

Index: appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py
diff --git a/appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py b/appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py
index 9f94f76b7fa0c7099902221ae680ee3ce1039408..86370bbca2e1b8e520edd23e08c0d321aa390599 100644
--- a/appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py
+++ b/appengine/findit/waterfall/test/process_flake_swarming_task_result_pipeline_test.py
@@ -10,124 +10,15 @@ from model.flake.master_flake_analysis import MasterFlakeAnalysis
 from waterfall import swarming_util
 from waterfall.process_flake_swarming_task_result_pipeline import (
     ProcessFlakeSwarmingTaskResultPipeline)
+from waterfall.test import (
+    process_base_swarming_task_result_pipeline_test as base_test)
 from waterfall.test import wf_testcase
 
 
-_ISOLATED_SERVER = 'https://isolateserver.appspot.com'
-_ISOLATED_STORAGE_URL = 'isolateserver.storage.googleapis.com'
-_SAMPLE_FAILURE_LOG = {
-    'per_iteration_data': [
-        {
-            'TestSuite1.test1': [
-                {
-                    'status': 'SUCCESS',
-                    'other_info': 'N/A'
-                }
-            ],
-            'TestSuite1.test2': [
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'SUCCESS',
-                    'other_info': 'N/A'
-                }
-            ],
-            'TestSuite1.test3': [
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                }
-            ]
-        },
-        {
-            'TestSuite1.test1': [
-                {
-                    'status': 'SUCCESS',
-                    'other_info': 'N/A'
-                }
-            ],
-            'TestSuite1.test2': [
-                {
-                    'status': 'SUCCESS',
-                    'other_info': 'N/A'
-                }
-            ],
-            'TestSuite1.test3': [
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                },
-                {
-                    'status': 'FAILURE',
-                    'other_info': 'N/A'
-                }
-            ]
-        }
-    ]
-}
-_EXPECTED_TEST_STATUS = {
-    'TestSuite1.test1': {
-        'total_run': 2,
-        'SUCCESS': 2
-    },
-    'TestSuite1.test2': {
-        'total_run': 4,
-        'SUCCESS': 2,
-        'FAILURE': 2
-    },
-    'TestSuite1.test3': {
-        'total_run': 6,
-        'FAILURE': 6
-    }
-}
-
-
 class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
 
   def _MockedGetSwarmingTaskResultById(self, task_id, _):
-    swarming_task_results = {
-        'task_id1': {
-            'state': 'COMPLETED',
-            'exit_code': '1',
-            'tags': [
-                'priority:25',
-                'ref_name:abc_tests'
-            ],
-            'outputs_ref': {
-                'isolatedserver': _ISOLATED_SERVER,
-                'namespace': 'default-gzip',
-                'isolated': 'shard1_isolated'
-            },
-            'created_ts': '2016-02-10T18:32:06.538220',
-            'started_ts': '2016-02-10T18:32:09.090550',
-            'completed_ts': '2016-02-10T18:33:09'
-        },
-        'task_id2': {
-            'state': 'TIMED_OUT',
-            'outputs_ref': None
-        }
-    }
-
-    mocked_result = swarming_task_results[task_id]
-    return mocked_result
+    return base_test._SWARMING_TASK_RESULTS[task_id]
 
   def setUp(self):
     super(ProcessFlakeSwarmingTaskResultPipelineTest, self).setUp()
@@ -141,16 +32,6 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     self.mock(swarming_util, 'GetSwarmingTaskResultById',
               self._MockedGetSwarmingTaskResultById)
 
-  def testCheckTestsRunStatusesNoOutPutJson(self):
-    call_params = ProcessFlakeSwarmingTaskResultPipeline._GetArgs(
-        self.pipeline, self.master_name, self.builder_name,
-        self.build_number, self.step_name, self.build_number,
-        self.test_name, self.version_number)
-    tests_statuses = (
-        ProcessFlakeSwarmingTaskResultPipeline._CheckTestsRunStatuses(
-            self.pipeline, None, *call_params))
-    self.assertEqual({}, tests_statuses)
-
   def testCheckTestsRunStatuses(self):
     analysis = MasterFlakeAnalysis.Create(
         self.master_name, self.builder_name,
@@ -170,8 +51,8 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     tests_statuses = (
         ProcessFlakeSwarmingTaskResultPipeline._CheckTestsRunStatuses(
             self.pipeline,
-            _SAMPLE_FAILURE_LOG, *call_params))
-    self.assertEqual(_EXPECTED_TEST_STATUS, tests_statuses)
+            base_test._SAMPLE_FAILURE_LOG, *call_params))
+    self.assertEqual(base_test._EXPECTED_TESTS_STATUS, tests_statuses)
 
   def testCheckTestsRunStatusesWhenTestDoesNotExist(self):
     test_name = 'TestSuite1.new_test'
@@ -187,11 +68,11 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
 
     pipeline = ProcessFlakeSwarmingTaskResultPipeline()
     tests_statuses = pipeline._CheckTestsRunStatuses(
-        _SAMPLE_FAILURE_LOG, self.master_name, self.builder_name,
+        base_test._SAMPLE_FAILURE_LOG, self.master_name, self.builder_name,
         self.build_number, self.step_name, self.build_number, test_name,
         self.version_number)
 
-    self.assertEqual(_EXPECTED_TEST_STATUS, tests_statuses)
+    self.assertEqual(base_test._EXPECTED_TESTS_STATUS, tests_statuses)
 
     task = FlakeSwarmingTask.Get(
         self.master_name, self.builder_name,
@@ -205,10 +86,10 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
     self.assertTrue(analysis.data_points[-1].pass_rate < 0)
 
   def _MockedGetSwarmingTaskFailureLog(self, *_):
-    return _SAMPLE_FAILURE_LOG
+    return base_test._SAMPLE_FAILURE_LOG
 
   def testProcessFlakeSwarmingTaskResultPipeline(self):
-
+    # End to end test.
     self.mock(swarming_util, 'GetSwarmingTaskFailureLog',
               self._MockedGetSwarmingTaskFailureLog)
 
@@ -237,7 +118,7 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
         self.step_name, self.test_name)
 
     self.assertEqual(analysis_status.COMPLETED, task.status)
-    self.assertEqual(_EXPECTED_TEST_STATUS, task.tests_statuses)
+    self.assertEqual(base_test._EXPECTED_TESTS_STATUS, task.tests_statuses)
 
     self.assertEqual(datetime.datetime(2016, 2, 10, 18, 32, 6, 538220),
                      task.created_time)
@@ -245,55 +126,3 @@ class ProcessFlakeSwarmingTaskResultPipelineTest(wf_testcase.WaterfallTestCase):
                      task.started_time)
     self.assertEqual(datetime.datetime(2016, 2, 10, 18, 33, 9),
                      task.completed_time)
-
-  def testProcessFlakeSwarmingTaskResultPipelineTaskNotRunning(self):
-    task = FlakeSwarmingTask.Create(
-        self.master_name, self.builder_name,
-        self.build_number, self.step_name, self.test_name)
-    task.task_id = 'task_id2'
-    task.put()
-
-    analysis = MasterFlakeAnalysis.Create(
-        self.master_name, self.builder_name,
-        self.build_number, self.step_name, self.test_name)
-    analysis.Save()
-
-    pipeline = ProcessFlakeSwarmingTaskResultPipeline()
-    step_name, task_info = pipeline.run(
-        self.master_name, self.builder_name, self.build_number, self.step_name,
-        'task_id2', self.build_number, self.test_name, analysis.version_number)
-    self.assertEqual(None, task_info)
-    self.assertEqual(self.step_name, step_name)
-
-    task = FlakeSwarmingTask.Get(
-        self.master_name, self.builder_name, self.build_number, self.step_name,
-        self.test_name)
-
-    self.assertEqual(analysis_status.ERROR, task.status)
-
-  def testProcessFlakeSwarmingTaskResultPipelineTaskTimeOut(self):
-    # Override swarming config settings to force a timeout.
-    override_swarming_settings = {
-        'task_timeout_hours': -1
-    }
-    self.UpdateUnitTestConfigSettings(
-        'swarming_settings', override_swarming_settings)
-
-    task = FlakeSwarmingTask.Create(
-        self.master_name, self.builder_name, self.build_number, self.step_name,
-        self.test_name)
-    task.task_id = 'task_id1'
-    task.put()
-
-    pipeline = ProcessFlakeSwarmingTaskResultPipeline()
-    step_name, task_info = pipeline.run(
-        self.master_name, self.builder_name, self.build_number, self.step_name,
-        'task_id1', self.build_number, self.test_name, self.version_number)
-    self.assertEqual('abc_tests', task_info)
-    self.assertEqual(self.step_name, step_name)
-
-    task = FlakeSwarmingTask.Get(
-        self.master_name, self.builder_name, self.build_number, self.step_name,
-        self.test_name)
-    self.assertEqual(analysis_status.ERROR, task.status)
-    self.assertEqual({}, task.tests_statuses)