swarming: mechanical changes to unit tests

appengine/swarming/handlers_test.py

 #!/usr/bin/env python
 # coding: utf-8
 # Copyright 2013 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 import datetime
 import json
 import logging
 import os
 import sys
 import unittest
 import urllib
 
 # Setups environment.
 import test_env_handlers
 
 import webtest
 
 import handlers_backend
 import handlers_frontend
 import template
+from components import utils
 from server import bot_code
 from server import bot_management
 from server import task_result
 
 
 class AppTestBase(test_env_handlers.AppTestBase):
   @staticmethod
   def wsgi_app():
     return None
 
@@ skipping 208 matching lines @@
   @staticmethod
   def wsgi_app():
     return handlers_backend.create_application(True)
 
   def _GetRoutes(self):
     """Returns the list of all routes handled."""
     return [
         r.template for r in self.app.app.router.match_routes
     ]
 
+  def setUp(self):
+    super(BackendTest, self).setUp()
+    self._enqueue_task_orig = self.mock(
+        utils, 'enqueue_task', self._enqueue_task)
+
+  def _enqueue_task(self, **kwargs):
+    return self._enqueue_task_orig(use_dedicated_module=False, **kwargs)
+
   def testCronJobTasks(self):
     # Tests all the cron tasks are securely handled.
     cron_job_urls = [
         r for r in self._GetRoutes() if r.startswith('/internal/cron/')
     ]
     self.assertTrue(cron_job_urls, cron_job_urls)
 
     # For ereporter.
     for cron_job_url in cron_job_urls:
       self.app.get(
@@ skipping 89 matching lines @@
           url+arg, headers={'X-AppEngine-QueueName': 'bogus name'}, status=403)
 
 
 if __name__ == '__main__':
   if '-v' in sys.argv:
     unittest.TestCase.maxDiff = None
   logging.basicConfig(
       level=logging.DEBUG if '-v' in sys.argv else logging.CRITICAL,
       format='%(levelname)-7s %(filename)s:%(lineno)3d %(message)s')
   unittest.main()