def get_backend_routes(): # pragma: no cover
prpc_server = prpc.Server()
prpc_server.add_interceptor(auth.prpc_interceptor)
prpc_server.add_service(api.BuildsApi())
return [ # pragma: no branch
webapp2.Route(r'/internal/cron/buildbucket/expire_build_leases',
expiration.CronExpireBuildLeases),
webapp2.Route(r'/internal/cron/buildbucket/expire_builds',
expiration.CronExpireBuilds),
webapp2.Route(r'/internal/cron/buildbucket/delete_builds',
expiration.CronDeleteBuilds),
webapp2.Route(r'/internal/cron/buildbucket/update_buckets',
CronUpdateBuckets),
webapp2.Route(r'/internal/cron/buildbucket/bq-export-prod',
bq.CronExportBuildsProd),
webapp2.Route(r'/internal/cron/buildbucket/bq-export-experimental',
bq.CronExportBuildsExperimental),
webapp2.Route(r'/internal/cron/buildbucket/unregister-builders',
UnregisterBuilders),
webapp2.Route(r'/internal/task/buildbucket/notify/<build_id:\d+>',
notifications.TaskPublishNotification),
webapp2.Route(
r'/internal/task/buildbucket/cancel_swarming_task/<host>/<task_id>',
TaskCancelSwarmingTask),
] + bulkproc.get_routes() + prpc_server.get_routes()
def get_frontend_routes(): # pragma: no cover
endpoints_services = [
legacy_api.BuildBucketApi,
config_api.ConfigApi,
swarmbucket_api.SwarmbucketApi,
]
routes = [
webapp2.Route(r'/_ah/start', DummyStartHandler),
webapp2.Route(r'/', MainHandler),
webapp2.Route(r'/b/<build_id:\d+>', BuildRPCHandler),
webapp2.Route(r'/build/<build_id:\d+>', ViewBuildHandler),
]
routes.extend(endpoints_webapp2.api_routes(endpoints_services))
# /api routes should be removed once clients are hitting /_ah/api.
routes.extend(
endpoints_webapp2.api_routes(endpoints_services, base_path='/api')
)
prpc_server = prpc.Server()
prpc_server.add_interceptor(auth.prpc_interceptor)
if modules.get_current_module_name() == 'beefy':
prpc_server.add_interceptor(_beefy_service_interceptor)
prpc_server.add_service(access.AccessServicer())
prpc_server.add_service(api.BuildsApi())
routes += prpc_server.get_routes()
return routes
def setUp(self):
super(BaseTestCase, self).setUp()
self.patch('user.can_async', return_value=future(True))
self.patch(
'user.get_accessible_buckets_async',
autospec=True,
return_value=future({'chromium/try'}),
)
self.now = datetime.datetime(2015, 1, 1)
self.patch('components.utils.utcnow', side_effect=lambda: self.now)
self.api = api.BuildsApi()