[Findit] Adding central config test class for unit tests

appengine/findit/waterfall/test/try_job_util_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 model import wf_analysis_status
-from model.wf_analysis import WfAnalysis
 from model.wf_try_job import WfTryJob
 from waterfall import try_job_util
-from waterfall import waterfall_config
+from waterfall.test import wf_testcase
 from waterfall.try_job_type import TryJobType
 
 
 class _MockRootPipeline(object):
   STARTED = False
 
   def __init__(self, *_):
     pass
 
   def start(self, *_, **__):
     _MockRootPipeline.STARTED = True
 
   @property
   def pipeline_status_path(self):
     return 'path'
 
 
-class TryJobUtilTest(testing.AppengineTestCase):
+class TryJobUtilTest(wf_testcase.WaterfallTestCase):
 
   def setUp(self):
     super(TryJobUtilTest, self).setUp()
 
-    def Mocked_GetTrybotForWaterfallBuilder(*_):
-      return 'tryserver.master', 'tryserver.builder'
-    self.mock(waterfall_config, 'GetTrybotForWaterfallBuilder',
-              Mocked_GetTrybotForWaterfallBuilder)
-
   def testNotNeedANewTryJobIfBuildIsNotCompletedYet(self):
     self.mock(
         try_job_util.swarming_tasks_to_try_job_pipeline,
         'SwarmingTasksToTryJobPipeline', _MockRootPipeline)
     _MockRootPipeline.STARTED = False
 
     failure_result_map = try_job_util.ScheduleTryJobIfNeeded(
         {}, build_completed=False)
 
     self.assertFalse(_MockRootPipeline.STARTED)
     self.assertEqual({}, failure_result_map)
 
   def testNotNeedANewTryJobIfBuilderIsNotSupportedYet(self):
-    def Mocked_GetTrybotForWaterfallBuilder(*_):
-      return None, None
-    self.mock(waterfall_config, 'GetTrybotForWaterfallBuilder',
-              Mocked_GetTrybotForWaterfallBuilder)
-
-    master_name = 'm'
-    builder_name = 'b'
+    master_name = 'master2'
+    builder_name = 'builder2'
     build_number = 223
     failure_info = {
         'master_name': master_name,
         'builder_name': builder_name,
         'build_number': build_number,
         'failed_steps': {
             'compile': {
                 'current_failure': 221,
                 'first_failure': 221,
                 'last_pass': 220
@@ skipping 325 matching lines @@
                 {'target': 'a.exe'},
                 {'source': 'b.cc',
                  'target': 'b.o'}]
         }
     }
 
     self.assertEqual(
         try_job_util._GetFailedTargetsFromSignals(signals, 'm', 'b'), ['a.exe'])
 
   def testUseObjectFilesAsFailedTargetIfStrictRegexUsed(self):
-    def Mocked_EnableStrictRegexForCompileLinkFailures(*_):
-      return True
-    self.mock(waterfall_config, 'EnableStrictRegexForCompileLinkFailures',
-              Mocked_EnableStrictRegexForCompileLinkFailures)
-
     signals = {
         'compile': {
             'failed_targets': [
                 {'source': 'b.cc', 'target': 'b.o'},
             ]
         }
     }
 
     self.assertEqual(
-        try_job_util._GetFailedTargetsFromSignals(signals, 'm', 'b'), ['b.o'])
+        try_job_util._GetFailedTargetsFromSignals(
+            signals, 'master1', 'builder1'),
+        ['b.o'])