def setUp(self):
super(ClientApiTest, self).setUp()
# By default requests in tests are coming from bot with fake IP.
routes = handlers_bot.get_routes() + handlers_api.get_routes()
app = webapp2.WSGIApplication(routes, debug=True)
self.app = webtest.TestApp(
app, extra_environ={"REMOTE_ADDR": self.source_ip, "SERVER_SOFTWARE": os.environ["SERVER_SOFTWARE"]}
)
self.mock(ereporter2, "log_request", lambda *args, **kwargs: self.fail("%s, %s" % (args, kwargs)))
# Client API test cases run by default as user.
self.set_as_user()
def create_application(debug):
template.bootstrap()
utils.set_task_queue_module('default')
routes = [
# Frontend pages. They return HTML.
# Public pages.
('/', RootHandler),
('/stats', stats_gviz.StatsSummaryHandler),
# User pages.
('/user/tasks', TasksHandler),
('/user/task/<task_id:[0-9a-fA-F]+>', TaskHandler),
('/user/task/<task_id:[0-9a-fA-F]+>/retry', TaskRetryHandler),
('/user/tasks/cancel', TaskCancelHandler),
# Privileged user pages.
('/restricted/bots', BotsListHandler),
('/restricted/bot/<bot_id:[^/]+>', BotHandler),
('/restricted/bot/<bot_id:[^/]+>/delete', BotDeleteHandler),
# Admin pages.
('/restricted/config', RestrictedConfigHandler),
('/restricted/upload/bot_config', UploadBotConfigHandler),
('/restricted/upload/bootstrap', UploadBootstrapHandler),
# Mapreduce related urls.
(r'/restricted/launch_mapreduce', RestrictedLaunchMapReduceJob),
# The new APIs:
('/swarming/api/v1/stats/summary/<resolution:[a-z]+>',
stats_gviz.StatsGvizSummaryHandler),
('/swarming/api/v1/stats/dimensions/<dimensions:.+>/<resolution:[a-z]+>',
stats_gviz.StatsGvizDimensionsHandler),
('/_ah/mail/<to:.+>', EmailHandler),
('/_ah/warmup', WarmupHandler),
]
routes = [webapp2.Route(*i) for i in routes]
# If running on a local dev server, allow bots to connect without prior
# groups configuration. Useful when running smoke test.
if utils.is_local_dev_server():
acl.bootstrap_dev_server_acls()
# TODO(maruel): Split backend into a separate module. For now add routes here.
routes.extend(handlers_backend.get_routes())
routes.extend(handlers_api.get_routes())
routes.extend(handlers_bot.get_routes())
return webapp2.WSGIApplication(routes, debug=debug)
def create_application(debug):
template.bootstrap()
utils.set_task_queue_module('default')
routes = [
# Frontend pages. They return HTML.
# Public pages.
('/', RootHandler),
('/stats', stats_gviz.StatsSummaryHandler),
# User pages.
('/user/tasks', TasksHandler),
('/user/task/<task_id:[0-9a-fA-F]+>', TaskHandler),
('/user/task/<task_id:[0-9a-fA-F]+>/retry', TaskRetryHandler),
('/user/tasks/cancel', TaskCancelHandler),
# Privileged user pages.
('/restricted/bots', BotsListHandler),
('/restricted/bot/<bot_id:[^/]+>', BotHandler),
('/restricted/bot/<bot_id:[^/]+>/delete', BotDeleteHandler),
# Admin pages.
('/restricted/config', RestrictedConfigHandler),
('/restricted/upload/bot_config', UploadBotConfigHandler),
('/restricted/upload/bootstrap', UploadBootstrapHandler),
# Mapreduce related urls.
(r'/restricted/launch_mapreduce', RestrictedLaunchMapReduceJob),
# The new APIs:
('/swarming/api/v1/stats/summary/<resolution:[a-z]+>',
stats_gviz.StatsGvizSummaryHandler),
('/swarming/api/v1/stats/dimensions/<dimensions:.+>/<resolution:[a-z]+>',
stats_gviz.StatsGvizDimensionsHandler),
('/_ah/mail/<to:.+>', EmailHandler),
('/_ah/warmup', WarmupHandler),
]
routes = [webapp2.Route(*i) for i in routes]
# If running on a local dev server, allow bots to connect without prior
# groups configuration. Useful when running smoke test.
if utils.is_local_dev_server():
acl.bootstrap_dev_server_acls()
# TODO(maruel): Split backend into a separate module. For now add routes here.
routes.extend(handlers_backend.get_routes())
routes.extend(handlers_api.get_routes())
routes.extend(handlers_bot.get_routes())
return webapp2.WSGIApplication(routes, debug=debug)
def create_application(debug):
"""Creates the url router.
The basic layouts is as follow:
- /restricted/.* requires being an instance administrator.
- /content/.* has the public HTTP API.
- /stats/.* has statistics.
"""
acl.bootstrap()
template.bootstrap()
routes = get_routes()
routes.extend(handlers_api.get_routes())
return webapp2.WSGIApplication(routes, debug=debug)
def create_application(debug): """Creates the url router. The basic layouts is as follow: - /restricted/.* requires being an instance administrator. - /content/.* has the public HTTP API. - /stats/.* has statistics. """ acl.bootstrap() template.bootstrap() routes = get_routes() routes.extend(handlers_api.get_routes()) return webapp2.WSGIApplication(routes, debug=debug)
def setUp(self):
super(BotApiTest, self).setUp()
# By default requests in tests are coming from bot with fake IP.
routes = handlers_bot.get_routes() + handlers_api.get_routes()
app = webapp2.WSGIApplication(routes, debug=True)
self.app = webtest.TestApp(app,
extra_environ={
'REMOTE_ADDR':
self.source_ip,
'SERVER_SOFTWARE':
os.environ['SERVER_SOFTWARE'],
})
self.mock(ereporter2, 'log_request',
lambda *args, **kwargs: self.fail('%s, %s' % (args, kwargs)))
# Bot API test cases run by default as bot.
self.set_as_bot()
def setUp(self):
super(BotApiTest, self).setUp()
# By default requests in tests are coming from bot with fake IP.
routes = handlers_bot.get_routes() + handlers_api.get_routes()
app = webapp2.WSGIApplication(routes, debug=True)
self.app = webtest.TestApp(
app,
extra_environ={
'REMOTE_ADDR': self.source_ip,
'SERVER_SOFTWARE': os.environ['SERVER_SOFTWARE'],
})
self.mock(
ereporter2, 'log_request',
lambda *args, **kwargs: self.fail('%s, %s' % (args, kwargs)))
# Bot API test cases run by default as bot.
self.set_as_bot()
def setUp(self):
test_case.EndpointsTestCase.setUp(self)
super(BaseTest, self).setUp()
self.mock(auth, 'is_group_member', lambda *_args, **_kwargs: True)
# handlers_bot is necessary to create fake tasks.
# TODO(maruel): Get rid of handlers_api.get_routes() here. The API should be
# self-sufficient.
routes = handlers_bot.get_routes() + handlers_api.get_routes()
self.app = webtest.TestApp(webapp2.WSGIApplication(routes, debug=True),
extra_environ={
'REMOTE_ADDR':
self.source_ip,
'SERVER_SOFTWARE':
os.environ['SERVER_SOFTWARE'],
})
self.mock(ereporter2, 'log_request',
lambda *args, **kwargs: self.fail('%s, %s' % (args, kwargs)))
# Client API test cases run by default as user.
self.set_as_user()
def setUp(self):
test_case.EndpointsTestCase.setUp(self)
super(BaseTest, self).setUp()
self.mock(auth, 'is_group_member', lambda *_args, **_kwargs: True)
# handlers_bot is necessary to create fake tasks.
# TODO(maruel): Get rid of handlers_api.get_routes() here. The API should be
# self-sufficient.
routes = handlers_bot.get_routes() + handlers_api.get_routes()
self.app = webtest.TestApp(
webapp2.WSGIApplication(routes, debug=True),
extra_environ={
'REMOTE_ADDR': self.source_ip,
'SERVER_SOFTWARE': os.environ['SERVER_SOFTWARE'],
})
self.mock(
ereporter2, 'log_request',
lambda *args, **kwargs: self.fail('%s, %s' % (args, kwargs)))
# Client API test cases run by default as user.
self.set_as_user()
def setUp(self):
"""Creates a new app instance for every test case."""
super(MainTest, self).setUp()
self.testbed.init_user_stub()
# When called during a taskqueue, the call to get_app_version() may fail so
# pre-fetch it.
version = utils.get_app_version()
self.mock(utils, 'get_task_queue_host', lambda: version)
self.source_ip = '192.168.0.1'
self.app_api = webtest.TestApp(
webapp2.WSGIApplication(handlers_api.get_routes(), debug=True),
extra_environ={'REMOTE_ADDR': self.source_ip})
# Do not use handlers_backend.create_application() because it also
# initializes ereporter2 cron jobs, which requires templates. We want to
# make sure templates are not needed for APIs.
self.app_backend = webtest.TestApp(
webapp2.WSGIApplication(handlers_backend.get_routes(), debug=True),
extra_environ={'REMOTE_ADDR': self.source_ip})
# Tasks are enqueued on the backend.
self.app = self.app_backend
def setUp(self):
"""Creates a new app instance for every test case."""
super(MainTest, self).setUp()
self.testbed.init_user_stub()
# When called during a taskqueue, the call to get_app_version() may fail so
# pre-fetch it.
version = utils.get_app_version()
self.mock(utils, 'get_task_queue_host', lambda: version)
self.source_ip = '192.168.0.1'
self.app_api = webtest.TestApp(
webapp2.WSGIApplication(handlers_api.get_routes(), debug=True),
extra_environ={'REMOTE_ADDR': self.source_ip})
# Do not use handlers_backend.create_application() because it also
# initializes ereporter2 cron jobs, which requires templates. We want to
# make sure templates are not needed for APIs.
self.app_backend = webtest.TestApp(
webapp2.WSGIApplication(handlers_backend.get_routes(), debug=True),
extra_environ={'REMOTE_ADDR': self.source_ip})
# Tasks are enqueued on the backend.
self.app = self.app_backend
