[Findit] Adding central config test class for unit tests

appengine/findit/waterfall/test/schedule_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 testing_utils import testing
-
 from common import buildbucket_client
 from model.wf_try_job import WfTryJob
-from waterfall import waterfall_config
 from waterfall.schedule_try_job_pipeline import ScheduleTryJobPipeline
+from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
-class ScheduleTryjobPipelineTest(testing.AppengineTestCase):
-
-  def _Mock_GetTrybotForWaterfallBuilder(self, *_):
-    def MockedGetTrybotForWaterfallBuilder(*_):
-      return 'linux_chromium_variable', 'master.tryserver.chromium.linux'
-    self.mock(waterfall_config, 'GetTrybotForWaterfallBuilder',
-              MockedGetTrybotForWaterfallBuilder)
+class ScheduleTryjobPipelineTest(wf_testcase.WaterfallTestCase):
 
   def _Mock_TriggerTryJobs(self, responses):
     def MockedTriggerTryJobs(*_):
       results = []
       for response in responses:
         if response.get('error'):  # pragma: no cover
           results.append((
               buildbucket_client.BuildbucketError(response['error']), None))
         else:
           results.append((
@@ skipping 58 matching lines @@
 
     responses = [
         {
             'build': {
                 'id': '1',
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
-    self._Mock_GetTrybotForWaterfallBuilder(master_name, builder_name)
+
     self._Mock_TriggerTryJobs(responses)
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
 
     try_job_pipeline = ScheduleTryJobPipeline()
     try_job_id = try_job_pipeline.run(
         master_name, builder_name, build_number, good_revision, bad_revision,
         TryJobType.COMPILE, None, None)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
@@ skipping 13 matching lines @@
 
     responses = [
         {
             'build': {
                 'id': '1',
                 'url': 'url',
                 'status': 'SCHEDULED',
             }
         }
     ]
-    self._Mock_GetTrybotForWaterfallBuilder(master_name, builder_name)
     self._Mock_TriggerTryJobs(responses)
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
 
     try_job_pipeline = ScheduleTryJobPipeline()
     try_job_id = try_job_pipeline.run(
         master_name, builder_name, build_number, good_revision, bad_revision,
         TryJobType.TEST, None, targeted_tests)
 
     try_job = WfTryJob.Get(master_name, builder_name, build_number)
     self.assertEqual('1', try_job_id)
     self.assertEqual('1', try_job.test_results[-1]['try_job_id'])