[Findit] Adding central config test class for unit tests

appengine/findit/waterfall/test/trigger_swarming_task_pipeline_test.py

 # Copyright 2016 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.
 
 import time
 
-from testing_utils import testing
-
 from model import wf_analysis_status
 from model.wf_swarming_task import WfSwarmingTask
 from waterfall import swarming_util
+from waterfall import trigger_swarming_task_pipeline
 from waterfall import waterfall_config
 from waterfall.swarming_task_request import SwarmingTaskRequest
-from waterfall import trigger_swarming_task_pipeline
+from waterfall.test import wf_testcase
 from waterfall.trigger_swarming_task_pipeline import TriggerSwarmingTaskPipeline
 
 
-_MOCK_SWARMING_SETTINGS = {
-    'task_timeout_hours': 23,
-    'server_query_interval_seconds': 60,
-    'iterations_to_rerun': 10,
-    'server_host': 'chromium-swarm.appspot.com',
-    'default_request_priority': 150,
-    'isolated_storage_url': 'isolateserver.storage.googleapis.com',
-    'isolated_server': 'https://isolateserver.appspot.com',
-    'request_expiration_hours': 20
-}
-
-
-class TriggerSwarmingTaskPipelineTest(testing.AppengineTestCase):
-
-  def _MockGetSwarmingSettings(self):
-    def MockedGetSwarmingSettings():
-      return _MOCK_SWARMING_SETTINGS
-    self.mock(waterfall_config, 'GetSwarmingSettings',
-              MockedGetSwarmingSettings)
+class TriggerSwarmingTaskPipelineTest(wf_testcase.WaterfallTestCase):
 
   def testNoNewSwarmingTaskIsNeeded(self):
     master_name = 'm'
     builder_name = 'b'
     build_number = 1
     step_name = 's'
     tests = ['a.b']
 
     def MockedListSwarmingTasksDataByTags(*_):
       raise Exception('Fail.')  # pragma: no cover. Should never be reached.
@@ skipping 74 matching lines @@
     def MockedTriggerSwarmingTask(new_request, *_):
       new_request_json.update(new_request.Serialize())
       return 'new_task_id'
     self.mock(swarming_util, 'TriggerSwarmingTask', MockedTriggerSwarmingTask)
 
     def MockedGetSwarmingTaskName(*_):
       return 'new_task_name'
     self.mock(trigger_swarming_task_pipeline, '_GetSwarmingTaskName',
               MockedGetSwarmingTaskName)
 
-    self._MockGetSwarmingSettings()
-
     master_name = 'm'
     builder_name = 'b'
     build_number = 234
     step_name = 'a_tests on platform'
     tests = ['a.b', 'a.c']
 
     expected_new_request_json = {
         'expiration_secs': 3600,
         'name': 'new_task_name',
         'parent_task_id': '',
@@ skipping 27 matching lines @@
     new_task_id = pipeline.run(
         master_name, builder_name, build_number, step_name, tests)
     self.assertEqual('new_task_id', new_task_id)
     self.assertEqual(expected_new_request_json, new_request_json)
 
     swarming_task = WfSwarmingTask.Get(
         master_name, builder_name, build_number, step_name)
     self.assertIsNotNone(swarming_task)
     self.assertEqual('new_task_id', swarming_task.task_id)
     self.assertEqual(tests, swarming_task.parameters['tests'])
-    self.assertEqual(_MOCK_SWARMING_SETTINGS['iterations_to_rerun'],
-                     swarming_task.parameters['iterations_to_rerun'])
+    self.assertEqual(
+        waterfall_config.GetSwarmingSettings()['iterations_to_rerun'],
+        swarming_task.parameters['iterations_to_rerun'])