[Findit] Adding central config test class for unit tests

appengine/findit/waterfall/test/monitor_try_job_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 datetime import datetime
 import json
-from testing_utils import testing
 import time
 
 from common import buildbucket_client
 from model import wf_analysis_status
 from model.wf_try_job import WfTryJob
 from model.wf_try_job_data import WfTryJobData
-from waterfall import waterfall_config
 from waterfall.monitor_try_job_pipeline import MonitorTryJobPipeline
+from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
 # A counter to enable different responses to requests in a loop.
 REQUEST_COUNTER = {
     '1': 0,
     '2': 0,
     '3': 0
 }
 
 
-class MonitorTryJobPipelineTest(testing.AppengineTestCase):
+class MonitorTryJobPipelineTest(wf_testcase.WaterfallTestCase):
 
   def _MockGetTryJobs(self, build_id):
     def Mocked_GetTryJobs(*_):
       data = {
           '1': [
               {
                   'build': {
                       'id': '1',
                       'url': 'url',
                       'status': 'COMPLETED',
@@ skipping 76 matching lines @@
         try_job_results.append((
             buildbucket_client.BuildbucketError(build_error['error']), None))
       else:
         try_job_results.append((
             None, buildbucket_client.BuildbucketBuild(build_error['build'])))
       REQUEST_COUNTER[build_id] += 1
       return try_job_results
 
     self.mock(buildbucket_client, 'GetTryJobs', Mocked_GetTryJobs)
 
-  def _MockGetTryJobSettings(self):
-    return {
-        'server_query_interval_seconds': 60,
-        'job_timeout_hours': 5,
-        'allowed_response_error_times': 5
-    }
-
   def setUp(self):
     super(MonitorTryJobPipelineTest, self).setUp()
-    self.mock(
-        waterfall_config, 'GetTryJobSettings', self._MockGetTryJobSettings)
     self.mock(time, 'sleep', lambda x: None)
 
   def testMicrosecondsToDatetime(self):
     self.assertEqual(
         datetime(2016, 2, 1, 22, 59, 34),
         MonitorTryJobPipeline._MicrosecondsToDatetime(1454367574000000))
     self.assertIsNone(MonitorTryJobPipeline._MicrosecondsToDatetime(None))
 
   def testUpdateTryJobMetadataForBuildError(self):
     error_data = {
@@ skipping 63 matching lines @@
     try_job.compile_results = [
         {
             'report': None,
             'url': 'url',
             'try_job_id': '1',
         }
     ]
     try_job.status = wf_analysis_status.ANALYZING
     try_job.put()
     self._MockGetTryJobs(try_job_id)
-    self._MockGetTryJobSettings()
 
     pipeline = MonitorTryJobPipeline()
     compile_result = pipeline.run(
         master_name, builder_name, build_number, TryJobType.COMPILE,
         try_job_id)
 
     expected_compile_result = {
         'report': {
             'result': {
                 'rev1': 'passed',
@@ skipping 26 matching lines @@
     try_job.test_results = [
         {
             'report': None,
             'url': 'url',
             'try_job_id': '3',
         }
     ]
     try_job.status = wf_analysis_status.ANALYZING
     try_job.put()
     self._MockGetTryJobs(try_job_id)
-    self._MockGetTryJobSettings()
 
     pipeline = MonitorTryJobPipeline()
     test_result = pipeline.run(
         master_name, builder_name, build_number, TryJobType.TEST,
         try_job_id)
 
     expected_test_result = {
         'report': {
             'result': {
                 'rev1': {
@@ skipping 27 matching lines @@
     try_job_id = '3'
 
     try_job = WfTryJob.Create(master_name, builder_name, build_number).put()
 
     pipeline = MonitorTryJobPipeline()
     pipeline._UpdateTryJobResult(
         buildbucket_client.BuildbucketBuild.STARTED, master_name, builder_name,
         build_number, TryJobType.TEST, try_job_id, 'url')
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual(wf_analysis_status.ANALYZING, try_job.status)