| # Copyright 2015 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| |
| from dashboard import buildbucket_job |
| from dashboard import testing_common |
| |
| |
| class BuildbucketJobTest(testing_common.TestCase): |
| |
| def setUp(self): |
| super(BuildbucketJobTest, self).setUp() |
| self._args_base = { |
| 'try_job_id': 1, |
| 'bisect_director': 'linux_perf_bisector', |
| 'recipe_tester_name': 'linux_perf_bisect', |
| 'good_revision': '1', |
| 'bad_revision': '2', |
| 'test_command': 'tools/perf/dummy_command', |
| 'metric': 'dummy_metric', |
| 'repeats': '5', |
| 'timeout_minutes': '10', |
| 'bug_id': None, |
| 'gs_bucket': 'dummy_bucket', |
| 'builder_host': None, |
| 'builder_port': None, |
| } |
| |
| def testCreateJob(self): |
| job = buildbucket_job.BisectJob(**self._args_base) |
| params = job.GetBuildParameters() |
| assert isinstance(params, dict) |
| self.assertIn('builder_name', params) |
| properties = params['properties'] |
| assert isinstance(properties, dict) |
| bisect_config = properties['bisect_config'] |
| assert isinstance(bisect_config, dict) |
| self.assertIn('test_type', bisect_config) |
| self.assertIn('command', bisect_config) |
| self.assertIn('src', bisect_config['command']) |
| self.assertIn('metric', bisect_config) |
| self.assertIn('good_revision', bisect_config) |
| self.assertIn('bad_revision', bisect_config) |
| self.assertIn('repeat_count', bisect_config) |
| self.assertIn('max_time_minutes', bisect_config) |
| |
| def testMissingRequiredArgs(self): |
| self._args_base['test_command'] = None |
| with self.assertRaises(ValueError): |
| job = buildbucket_job.BisectJob(**self._args_base) |
| _ = job.GetBuildParameters() |
| |
| if __name__ == '__main__': |
| unittest.main() |
