Reimplement Machine Provider integration

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 itertools
 import json
 import logging
@@ skipping 444 matching lines @@
         r.template for r in self.app.app.router.match_routes
     ]
 
   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)
 
-    # Stub out task queue triggered by /internal/cron/machine_provider_pubsub.
-    self.mock(
-        handlers_frontend.handlers_backend.lease_management,
-        'process_pubsub', lambda *args, **kwargs: None)
-
     # For ereporter.
     for cron_job_url in cron_job_urls:
       self.app.get(
           cron_job_url, headers={'X-AppEngine-Cron': 'true'}, status=200)
 
       # Only cron job requests can be gets for this handler.
       response = self.app.get(cron_job_url, status=403)
       self.assertEqual(
           '403 Forbidden\n\nAccess was denied to this resource.\n\n '
           'Only internal cron jobs can do this  ',
@@ skipping 68 matching lines @@
 
   def testTaskQueueUrls(self):
     # Tests all the task queue tasks are securely handled.
     # TODO(maruel): Test mapreduce.
     task_queue_urls = sorted(
       r for r in self._GetRoutes() if r.startswith('/internal/taskqueue/')
       if r != '/internal/taskqueue/mapreduce/launch/<job_id:[^\\/]+>'
     )
     task_queues = [
       ('cleanup', '/internal/taskqueue/cleanup_data'),
-      ('machine-provider-pubsub',
-       '/internal/taskqueue/pubsub/machine_provider'),
+      ('machine-provider-manage',
+       '/internal/taskqueue/machine-provider-manage'),
       ('pubsub', '/internal/taskqueue/pubsub/<task_id:[0-9a-f]+>'),
     ]
     self.assertEqual(sorted(zip(*task_queues)[1]), task_queue_urls)
 
     for _, url in task_queues:
       url = url.replace('<task_id:[0-9a-f]+>', 'abcabcabc')
       self.app.post(
           url, 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()