Index: chrome/common/extensions/docs/server2/integration_test.py |
diff --git a/chrome/common/extensions/docs/server2/integration_test.py b/chrome/common/extensions/docs/server2/integration_test.py |
index cfc3f37bbaacbb711ec7a3840126efda6776d382..3724a9cb5725633d0922cb8eceee8f62d4eab43d 100755 |
--- a/chrome/common/extensions/docs/server2/integration_test.py |
+++ b/chrome/common/extensions/docs/server2/integration_test.py |
@@ -16,6 +16,7 @@ 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 ( |
@@ -92,6 +93,13 @@ class IntegrationTest(unittest.TestCase): |
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: |