Docserver: Run tasks in testing

chrome/common/extensions/docs/server2/integration_test.py

 #!/usr/bin/env python
 # Copyright 2013 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.
 
 # Run build_server so that files needed by tests are copied to the local
 # third_party directory.
 import build_server
 build_server.main()
 
 import json
 import optparse
 import os
 import posixpath
 import sys
 import time
 import unittest
 
+from appengine_wrappers import SetTaskRunnerForTest
 from branch_utility import BranchUtility
 from chroot_file_system import ChrootFileSystem
 from extensions_paths import (
     CONTENT_PROVIDERS, CHROME_EXTENSIONS, PUBLIC_TEMPLATES)
 from fake_fetchers import ConfigureFakeFetchers
 from special_paths import SITE_VERIFICATION_FILE
 from handler import Handler
 from link_error_detector import LinkErrorDetector, StringifyBrokenLinks
 from local_file_system import LocalFileSystem
 from local_renderer import LocalRenderer
@@ skipping 56 matching lines @@
     ConfigureFakeFetchers()
 
   @EnableLogging('info')
   def testCronAndPublicFiles(self):
     '''Runs cron then requests every public file. Cron needs to be run first
     because the public file requests are offline.
     '''
     if _EXPLICIT_TEST_FILES is not None:
       return
 
+
+    def task_runner(url, commit=None):
+      arguments = { 'commit': commit } if commit else {}
+      Handler(Request.ForTest(url, arguments=arguments)).Get()
+
+    SetTaskRunnerForTest(task_runner)
+
     print('Running cron...')
     start_time = time.time()
     try:
       response = Handler(Request.ForTest('/_cron')).Get()
       if response:
         self.assertEqual(200, response.status)
         self.assertEqual('Success', response.content.ToString())
       else:
         self.fail('No response for _cron')
     finally:
@@ skipping 169 matching lines @@
   parser.add_option('-v', '--verbose', action='store_true', default=False,
                     help='Show verbose output like currently broken links')
   (opts, args) = parser.parse_args()
   if not opts.all:
     _EXPLICIT_TEST_FILES = args
   _REBASE = opts.rebase
   _VERBOSE = opts.verbose
   # Kill sys.argv because we have our own flags.
   sys.argv = [sys.argv[0]]
   unittest.main()