| OLD | NEW |
| 1 # coding: utf-8 | 1 # coding: utf-8 |
| 2 # Copyright 2014 The LUCI Authors. All rights reserved. | 2 # Copyright 2014 The LUCI Authors. All rights reserved. |
| 3 # Use of this source code is governed under the Apache License, Version 2.0 | 3 # Use of this source code is governed under the Apache License, Version 2.0 |
| 4 # that can be found in the LICENSE file. | 4 # that can be found in the LICENSE file. |
| 5 | 5 |
| 6 """Base class for handlers_*_test.py""" | 6 """Base class for handlers_*_test.py""" |
| 7 | 7 |
| 8 import base64 | 8 import base64 |
| 9 import json | 9 import json |
| 10 import os | 10 import os |
| (...skipping 18 matching lines...) Expand all Loading... |
| 29 | 29 |
| 30 | 30 |
| 31 class AppTestBase(test_case.TestCase): | 31 class AppTestBase(test_case.TestCase): |
| 32 APP_DIR = test_env.APP_DIR | 32 APP_DIR = test_env.APP_DIR |
| 33 | 33 |
| 34 def setUp(self): | 34 def setUp(self): |
| 35 super(AppTestBase, self).setUp() | 35 super(AppTestBase, self).setUp() |
| 36 self.bot_version = None | 36 self.bot_version = None |
| 37 self.source_ip = '192.168.2.2' | 37 self.source_ip = '192.168.2.2' |
| 38 self.testbed.init_user_stub() | 38 self.testbed.init_user_stub() |
| 39 self.testbed.init_search_stub() | |
| 40 | 39 |
| 41 gae_ts_mon.reset_for_unittest(disable=True) | 40 gae_ts_mon.reset_for_unittest(disable=True) |
| 42 | 41 |
| 43 # By default requests in tests are coming from bot with fake IP. | 42 # By default requests in tests are coming from bot with fake IP. |
| 44 # WSGI app that implements auth REST API. | 43 # WSGI app that implements auth REST API. |
| 45 self.auth_app = webtest.TestApp( | 44 self.auth_app = webtest.TestApp( |
| 46 auth.create_wsgi_application(debug=True), | 45 auth.create_wsgi_application(debug=True), |
| 47 extra_environ={ | 46 extra_environ={ |
| 48 'REMOTE_ADDR': self.source_ip, | 47 'REMOTE_ADDR': self.source_ip, |
| 49 'SERVER_SOFTWARE': os.environ['SERVER_SOFTWARE'], | 48 'SERVER_SOFTWARE': os.environ['SERVER_SOFTWARE'], |
| (...skipping 207 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 257 | 256 |
| 258 def client_create_task_raw(self, properties=None, **kwargs): | 257 def client_create_task_raw(self, properties=None, **kwargs): |
| 259 """Creates a raw command TaskRequest via the Cloud Endpoints API.""" | 258 """Creates a raw command TaskRequest via the Cloud Endpoints API.""" |
| 260 properties = (properties or {}).copy() | 259 properties = (properties or {}).copy() |
| 261 properties['command'] = ['python', 'run_test.py'] | 260 properties['command'] = ['python', 'run_test.py'] |
| 262 return self._client_create_task(properties, **kwargs) | 261 return self._client_create_task(properties, **kwargs) |
| 263 | 262 |
| 264 def client_get_results(self, task_id): | 263 def client_get_results(self, task_id): |
| 265 api = test_case.Endpoints(handlers_endpoints.SwarmingTaskService) | 264 api = test_case.Endpoints(handlers_endpoints.SwarmingTaskService) |
| 266 return api.call_api('result', body={'task_id': task_id}).json | 265 return api.call_api('result', body={'task_id': task_id}).json |
| OLD | NEW |