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. ('/oldui', OldUIHandler), ('/stats', stats_gviz.StatsSummaryHandler), ('/<page:(bot|botlist|task|tasklist|)>', UIHandler), # User pages. ('/user/tasks', TasksHandler), ('/user/task/<task_id:[0-9a-fA-F]+>', TaskHandler), ('/user/task/<task_id:[0-9a-fA-F]+>/cancel', TaskCancelHandler), ('/user/task/<task_id:[0-9a-fA-F]+>/retry', TaskRetryHandler), # Privileged user pages. ('/restricted/bots', BotsListHandler), ('/restricted/bot/<bot_id:[^/]+>', BotHandler), ('/restricted/bot/<bot_id:[^/]+>/delete', BotDeleteHandler), # Admin pages. ('/restricted/config', RestrictedConfigHandler), ('/restricted/cancel_pending', RestrictedCancelPendingHandler), ('/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() routes.extend(handlers_backend.get_routes()) routes.extend(handlers_bot.get_routes()) routes.extend(handlers_endpoints.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 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. self.app = webtest.TestApp( webapp2.WSGIApplication(handlers_bot.get_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): 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() 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(PRPCTest, self).setUp() # handlers_bot is necessary to run fake tasks. routes = handlers_prpc.get_routes() + handlers_bot.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._headers = { 'Content-Type': encoding.Encoding.JSON[1], 'Accept': encoding.Encoding.JSON[1], } self._enqueue_task_orig = self.mock( utils, 'enqueue_task', self._enqueue_task) self.now = datetime.datetime(2010, 1, 2, 3, 4, 5) self.mock_now(self.now) self.mock_default_pool_acl([])
def create_application(debug): routes = [] routes.extend(get_routes()) routes.extend(handlers_bot.get_routes()) routes.extend(handlers_endpoints.get_routes()) return webapp2.WSGIApplication(routes, debug=debug)