| 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 | |
| Xixuan Wu | a17e328 | 2018-08-21 16:11:11 -0700 | [diff] [blame] | 14 | class TestAFESwarmingAPI(webapp2.RequestHandler): |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 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, |
| Prathmesh Prabhu | cf67074 | 2018-09-27 15:13:48 -0700 | [diff] [blame] | 22 | swarming_lib.SKYLAB_STAGING_SWARMING_SERVER) |
| Xixuan Wu | a17e328 | 2018-08-21 16:11:11 -0700 | [diff] [blame] | 23 | afe_response = swarming_runner.dummy_run() |
| 24 | afe_swarming_task_info = swarming_runner.query_task(afe_response['task_id']) |
| 25 | self.response.write(afe_swarming_task_info.name) |
| 26 | |
| 27 | |
| 28 | class TestSkylabSwarmingAPI(webapp2.RequestHandler): |
| 29 | |
| 30 | def get(self): |
| 31 | lab_config = config_reader.LabConfig(config_reader.ConfigReader( |
| 32 | file_getter.LAB_CONFIG_FILE)) |
| 33 | afe_server = lab_config.get_dedicated_afe() |
| 34 | swarming_runner = swarming_lib.SwarmingRunner( |
| 35 | afe_server, |
| Prathmesh Prabhu | cf67074 | 2018-09-27 15:13:48 -0700 | [diff] [blame] | 36 | swarming_lib.SKYLAB_STAGING_SWARMING_SERVER) |
| Xixuan Wu | a17e328 | 2018-08-21 16:11:11 -0700 | [diff] [blame] | 37 | skylab_response = swarming_runner.dummy_run(is_skylab=True) |
| 38 | skylab_swarming_task_info = swarming_runner.query_task( |
| 39 | skylab_response['task_id'], is_skylab=True) |
| 40 | self.response.write(skylab_swarming_task_info.name) |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 41 | |
| 42 | |
| Xixuan Wu | d9dd474 | 2018-08-23 15:57:03 -0700 | [diff] [blame] | 43 | class TestAFEWithSanity(webapp2.RequestHandler): |
| 44 | |
| 45 | def get(self): |
| 46 | lab_config = config_reader.LabConfig(config_reader.ConfigReader( |
| 47 | file_getter.LAB_CONFIG_FILE)) |
| 48 | afe_server = lab_config.get_dedicated_afe() |
| 49 | swarming_runner = swarming_lib.SwarmingRunner( |
| 50 | afe_server, |
| Prathmesh Prabhu | cf67074 | 2018-09-27 15:13:48 -0700 | [diff] [blame] | 51 | swarming_lib.SKYLAB_STAGING_SWARMING_SERVER) |
| Xixuan Wu | d9dd474 | 2018-08-23 15:57:03 -0700 | [diff] [blame] | 52 | skylab_response = swarming_runner.dummy_run_sanity(is_skylab=False) |
| 53 | skylab_swarming_task_info = swarming_runner.query_task( |
| 54 | skylab_response['task_id'], is_skylab=False) |
| 55 | self.response.write(skylab_swarming_task_info.tags) |
| 56 | |
| 57 | |
| 58 | class TestSkylabWithSanity(webapp2.RequestHandler): |
| 59 | |
| 60 | def get(self): |
| 61 | lab_config = config_reader.LabConfig(config_reader.ConfigReader( |
| 62 | file_getter.LAB_CONFIG_FILE)) |
| 63 | afe_server = lab_config.get_dedicated_afe() |
| 64 | swarming_runner = swarming_lib.SwarmingRunner( |
| 65 | afe_server, |
| Xixuan Wu | 028f673 | 2019-04-11 14:47:42 -0700 | [diff] [blame^] | 66 | swarming_lib.SKYLAB_SWARMING_SERVER) |
| Xixuan Wu | d9dd474 | 2018-08-23 15:57:03 -0700 | [diff] [blame] | 67 | skylab_response = swarming_runner.dummy_run_sanity(is_skylab=True) |
| 68 | skylab_swarming_task_info = swarming_runner.query_task( |
| 69 | skylab_response['task_id'], is_skylab=True) |
| 70 | self.response.write(skylab_swarming_task_info.tags) |
| 71 | |
| 72 | |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 73 | app = webapp2.WSGIApplication([ |
| Xixuan Wu | d9dd474 | 2018-08-23 15:57:03 -0700 | [diff] [blame] | 74 | ('/test_push/afe_dummy', TestAFESwarmingAPI), |
| 75 | ('/test_push/skylab_dummy', TestSkylabSwarmingAPI), |
| 76 | ('/test_push/afe_sanity', TestAFEWithSanity), |
| 77 | ('/test_push/skylab_sanity', TestSkylabWithSanity), |
| Xixuan Wu | 865fa28 | 2017-09-05 15:23:19 -0700 | [diff] [blame] | 78 | ], debug=True) |