[Findit] Adding central config test class for unit tests

appengine/findit/handlers/test/handlers_util_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.
 
-from testing_utils import testing
-
+from handlers import handlers_util
 from handlers import result_status
-from handlers import handlers_util
 from model import wf_analysis_status
 from model.wf_analysis import WfAnalysis
 from model.wf_swarming_task import WfSwarmingTask
 from model.wf_try_job import WfTryJob
-from waterfall import buildbot
-from waterfall import waterfall_config
+from waterfall.test import wf_testcase
 
 
-class HandlersUtilResultTest(testing.AppengineTestCase):
+class HandlersUtilResultTest(wf_testcase.WaterfallTestCase):
 
   def setUp(self):
     super(HandlersUtilResultTest, self).setUp()
     self.master_name = 'm'
     self.builder_name = 'b'
     self.build_number = 121
 
-    def MockedGetSwarmingSettings():
-      return {'server_host': 'chromium-swarm.appspot.com'}
-    self.mock(
-        waterfall_config, 'GetSwarmingSettings', MockedGetSwarmingSettings)
-
   def testGetSwarmingTaskInfoNoAnalysis(self):
     data = handlers_util.GetSwarmingTaskInfo(
         self.master_name, self.builder_name, self.build_number)
-
     self.assertEqual({}, data)
 
   def testGetSwarmingTaskInfoReturnEmptyIfNoFailureMap(self):
     WfAnalysis.Create(
         self.master_name, self.builder_name, self.build_number).put()
 
     data = handlers_util.GetSwarmingTaskInfo(
         self.master_name, self.builder_name, self.build_number)
 
     self.assertEqual({}, data)
@@ skipping 171 matching lines @@
                         'task_url': ('https://chromium-swarm.appspot.com/user'
                                      '/task/task1')
                     },
                     'all_tests': ['test2', 'test3', 'test4'],
                     'reliable_tests': ['test2'],
                     'flaky_tests': ['test3', 'test4'],
                     'ref_name': 'step1'
                 },
                 'm/b/120': {
                     'task_info': {
-                    'status': wf_analysis_status.ANALYZED,
-                    'task_id': 'task0',
-                    'task_url': (
-                        'https://chromium-swarm.appspot.com/user/task/task0')
+                        'status': wf_analysis_status.ANALYZED,
+                        'task_id': 'task0',
+                        'task_url': ('https://chromium-swarm.appspot.com/user/'
+                                     'task/task0')
                     },
                     'all_tests': ['test1'],
                     'reliable_tests': ['test1'],
                     'flaky_tests': [],
                     'ref_name': 'step1'
                 }
             }
         },
         'step2': {
             'swarming_tasks': {
@@ skipping 126 matching lines @@
     try_job.compile_results = [
         {
             'try_job_id': '1'
         }
     ]
     try_job.put()
 
     result = handlers_util._GetTryJobResultForCompile({'compile': 'm/b/121'})
 
     expected_result = {
-        'compile':  {
+        'compile': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/121',
                     'status': wf_analysis_status.ERROR
                 }
             ]
         }
     }
 
     self.assertEqual(expected_result, result)
@@ skipping 28 matching lines @@
                 }
             }
         }
     ]
     try_job.put()
 
     result = handlers_util.GetAllTryJobResults(
         self.master_name, self.builder_name, self.build_number)
 
     expected_result = {
-        'compile':  {
+        'compile': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/121',
                     'status': wf_analysis_status.ANALYZED,
                     'try_job_build_number': 121,
                     'try_job_url': (
                         'http://build.chromium.org/p/tryserver.chromium.'
                         'linux/builders/linux_chromium_variable/builds/121'),
                     'culprit': {
                         'revision': 'rev2',
@@ skipping 29 matching lines @@
             'url': ('http://build.chromium.org/p/tryserver.chromium.linux/'
                     'builders/linux_chromium_variable/builds/121')
         }
     ]
     try_job.put()
 
     result = handlers_util.GetAllTryJobResults(
         self.master_name, self.builder_name, self.build_number)
 
     expected_result = {
-        'compile':  {
+        'compile': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/121',
                     'status': wf_analysis_status.ANALYZED,
                     'try_job_build_number': 121,
                     'try_job_url': (
                         'http://build.chromium.org/p/tryserver.chromium.'
                         'linux/builders/linux_chromium_variable/builds/121')
                 }
             ]
         }
     }
 
     self.assertEqual(expected_result, result)
 
   def testGetAllTryJobResultsTestFailureNoTaskInfo(self):
     analysis = WfAnalysis.Create(
         self.master_name, self.builder_name, self.build_number)
     analysis.failure_result_map = {
         'step1': {
             'test1': 'm/b/118'
         }
     }
     analysis.put()
 
     result = handlers_util.GetAllTryJobResults(
         self.master_name, self.builder_name, self.build_number)
 
     expected_result = {
-        'step1':{
+        'step1': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/118',
                     'status': (
                         result_status.NO_TRY_JOB_REASON_MAP.get(
                             result_status.NO_SWARMING_TASK_FOUND)),
                     'tests': ['test1']
                 }
             ]
         }
@@ skipping 71 matching lines @@
                     'all_tests': ['test3']
                 }
             }
         }
     }
 
     result = handlers_util._GetAllTryJobResultsForTest(
         failure_result_map, tasks_info)
 
     expected_result = {
-        'step1':{
+        'step1': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/118',
                     'status': result_status.NO_TRY_JOB_REASON_MAP[
                         wf_analysis_status.PENDING],
                     'tests': ['test1']
                 },
                 {
                     'try_job_key': 'm/b/119',
                     'status': result_status.NO_TRY_JOB_REASON_MAP[
@@ skipping 107 matching lines @@
                 }
             }
         }
     ]
     try_job.put()
 
     result = handlers_util._GetAllTryJobResultsForTest(
         failure_result_map, tasks_info)
 
     expected_result = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': 'm/b/119',
                     'status': wf_analysis_status.ANALYZED,
                     'try_job_url': (
                         'http://build.chromium.org/p/tryserver.chromium.'
                         'linux/builders/linux_chromium_variable/builds/121'),
                     'try_job_build_number': 121,
                     'culprit': {
@@ skipping 81 matching lines @@
                 }
             }
         }
     ]
     try_job.put()
 
     result = handlers_util._GetAllTryJobResultsForTest(
         failure_result_map, tasks_info)
 
     expected_result = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'try_job_key': 'm/b/118',
                     'status': result_status.NO_TRY_JOB_REASON_MAP.get(
                         result_status.NO_SWARMING_TASK_FOUND),
                     'tests': ['test1']
                 },
                 {
                     'ref_name': 'step1',
                     'try_job_key': 'm/b/119',
@@ skipping 58 matching lines @@
 
     result = handlers_util._OrganizeTryJobResultByCulprits(try_job_culprits)
     expected_result = {
         'rev1': {
             'revision': 'rev1',
             'commit_position': '1',
             'review_url': 'url_1',
             'failed_tests': ['a_test2', 'a_test1']
         }
     }
-    self.assertEqual(expected_result,result)
+    self.assertEqual(expected_result, result)
 
   def testGetCulpritInfoForTryJobResultForTestTryJobNoResult(self):
     try_job_key = 'm/b/119'
     culprits_info = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3']
                 }
             ]
         }
     }
     WfTryJob.Create('m', 'b', '119').put()
     handlers_util._GetCulpritInfoForTryJobResultForTest(
         try_job_key, culprits_info)
 
     expected_culprits_info = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3'],
                     'status': wf_analysis_status.PENDING
                 }
             ]
         }
     }
     self.assertEqual(expected_culprits_info, culprits_info)
 
   def testGetCulpritInfoForTryJobResultForTestTryJobRunning(self):
     try_job_key = 'm/b/119'
     culprits_info = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3']
                 }
             ]
         }
     }
     try_job = WfTryJob.Create('m', 'b', '119')
     try_job.status = wf_analysis_status.ANALYZING
     try_job.test_results = [
         {
             'url': ('http://build.chromium.org/p/tryserver.chromium.'
                     'linux/builders/linux_chromium_variable/builds/121'),
             'try_job_id': '121'
         }
     ]
     try_job.put()
     handlers_util._GetCulpritInfoForTryJobResultForTest(
         try_job_key, culprits_info)
 
     expected_culprits_info = {
-        'step1 on platform':{
+        'step1 on platform': {
             'try_jobs': [
                 {
                     'ref_name': 'step1',
                     'try_job_key': try_job_key,
                     'tests': ['test2', 'test3'],
                     'status': wf_analysis_status.ANALYZING,
                     'try_job_url': (
                         'http://build.chromium.org/p/tryserver.chromium.'
                         'linux/builders/linux_chromium_variable/builds/121'),
                     'try_job_build_number': 121
                 }
             ]
         }
     }
     self.assertEqual(expected_culprits_info, culprits_info)