[Findit] Adding central config test class for unit tests

appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py

Index: appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py
diff --git a/appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py b/appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py
index 34b633e69495d54c974cb6dfb87d8f68c7ef012f..26adc26ed8cdf72b3fb02904cd82c5f4bcbf7862 100644
--- a/appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py
+++ b/appengine/findit/waterfall/test/schedule_try_job_pipeline_test.py
@@ -2,22 +2,14 @@
 # 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(*_):
@@ -96,7 +88,7 @@ class ScheduleTryjobPipelineTest(testing.AppengineTestCase):
             }
         }
     ]
-    self._Mock_GetTrybotForWaterfallBuilder(master_name, builder_name)
+
     self._Mock_TriggerTryJobs(responses)
 
     WfTryJob.Create(master_name, builder_name, build_number).put()
@@ -130,7 +122,6 @@ class ScheduleTryjobPipelineTest(testing.AppengineTestCase):
             }
         }
     ]
-    self._Mock_GetTrybotForWaterfallBuilder(master_name, builder_name)
     self._Mock_TriggerTryJobs(responses)
 
     WfTryJob.Create(master_name, builder_name, build_number).put()