swarming: mechanical changes to unit tests

appengine/swarming/handlers_bot_test.py

Index: appengine/swarming/handlers_bot_test.py
diff --git a/appengine/swarming/handlers_bot_test.py b/appengine/swarming/handlers_bot_test.py
index 921c9ea3dc7ef0d58d4d32bc5e877efdb72435de..4cbffbcdbdd3f51dc5438eceae7a3b4f6e96a84f 100755
--- a/appengine/swarming/handlers_bot_test.py
+++ b/appengine/swarming/handlers_bot_test.py
@@ -23,6 +23,7 @@ from google.appengine.ext import ndb
 import webapp2
 import webtest
 
+import handlers_backend
 import handlers_bot
 from components import ereporter2
 from components import utils
@@ -54,6 +55,17 @@ class BotApiTest(test_env_handlers.AppTestBase):
         lambda *args, **kwargs: self.fail('%s, %s' % (args, kwargs)))
     # Bot API test cases run by default as bot.
     self.set_as_bot()
+    self._enqueue_task_orig = self.mock(
+        utils, 'enqueue_task', self._enqueue_task)
+
+  def _enqueue_task(self, url, queue_name, **kwargs):
+    if queue_name == 'task-dimensions':
+      # Call directly into it.
+      handlers_backend.TaskDimensionsHandler.tidy_stale(kwargs['payload'])

[Vadim Sh., 2017/04/26 21:30:20]

note that this may incorrectly inherit transaction context of the caller. Maybe
use ndb.non_transactional around _enqueue_task for completness.

[M-A Ruel, 2017/04/26 21:47:43]

Done.

+      return True
+    if queue_name == 'pubsub':
+      return True
+    self.fail(url)
 
   def test_handshake(self):
     errors = []