| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 1 | # Copyright 2017 The Chromium OS Authors. All rights reserved. |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """Module for tests for handlers.""" |
| 6 | |
| Craig Bergstrom | 3212fb4 | 2018-04-17 19:24:15 -0600 | [diff] [blame] | 7 | import config_reader |
| 8 | import file_getter |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 9 | import swarming_lib |
| 10 | |
| 11 | import webapp2 |
| 12 | |
| 13 | |
| 14 | class TestSwarmingAPI(webapp2.RequestHandler): |
| 15 | |
| 16 | def get(self): |
| Craig Bergstrom | 3212fb4 | 2018-04-17 19:24:15 -0600 | [diff] [blame] | 17 | lab_config = config_reader.LabConfig(config_reader.ConfigReader( |
| 18 | file_getter.LAB_CONFIG_FILE)) |
| 19 | afe_server = lab_config.get_dedicated_afe() |
| Xixuan Wu | 5d700dc | 2018-08-21 15:13:10 -0700 | [diff] [blame^] | 20 | swarming_runner = swarming_lib.SwarmingRunner( |
| 21 | afe_server, |
| 22 | swarming_lib.CHROME_STAGING_SWARMING_SERVER) |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 23 | response = swarming_runner.dummy_run() |
| 24 | swarming_task_info = swarming_runner.query_task(response['task_id']) |
| 25 | self.response.write(swarming_task_info.name) |
| 26 | |
| 27 | |
| 28 | app = webapp2.WSGIApplication([ |
| 29 | ('/test_push/swarming', TestSwarmingAPI), |
| 30 | ], debug=True) |