| OLD | NEW |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import mock | 5 import mock |
| 6 | 6 |
| 7 from common.waterfall import buildbucket_client | 7 from common.waterfall import buildbucket_client |
| 8 from common.waterfall import failure_type | 8 from common.waterfall import failure_type |
| 9 from model.wf_try_job import WfTryJob | 9 from model.wf_try_job import WfTryJob |
| 10 from model.wf_try_job_data import WfTryJobData | 10 from model.wf_try_job_data import WfTryJobData |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 '/%s/builds/%s') % ( | 51 '/%s/builds/%s') % ( |
| 52 master_name, builder_name, build_number) | 52 master_name, builder_name, build_number) |
| 53 } | 53 } |
| 54 try_job_pipeline = ScheduleTryJobPipeline() | 54 try_job_pipeline = ScheduleTryJobPipeline() |
| 55 properties = try_job_pipeline._GetBuildProperties( | 55 properties = try_job_pipeline._GetBuildProperties( |
| 56 master_name, builder_name, build_number, 1, 2, failure_type.TEST, None) | 56 master_name, builder_name, build_number, 1, 2, failure_type.TEST, None) |
| 57 | 57 |
| 58 self.assertEqual(properties, expected_properties) | 58 self.assertEqual(properties, expected_properties) |
| 59 | 59 |
| 60 @mock.patch.object(schedule_try_job_pipeline, 'buildbucket_client') | 60 @mock.patch.object(schedule_try_job_pipeline, 'buildbucket_client') |
| 61 def testTriggerTryJob(self, mock_other): | 61 def testTriggerTryJob(self, mock_module): |
| 62 master_name = 'm' | 62 master_name = 'm' |
| 63 builder_name = 'b' | 63 builder_name = 'b' |
| 64 response = { | 64 response = { |
| 65 'build': { | 65 'build': { |
| 66 'id': '1', | 66 'id': '1', |
| 67 'url': 'url', | 67 'url': 'url', |
| 68 'status': 'SCHEDULED', | 68 'status': 'SCHEDULED', |
| 69 } | 69 } |
| 70 } | 70 } |
| 71 results = [(None, buildbucket_client.BuildbucketBuild(response['build']))] | 71 results = [(None, buildbucket_client.BuildbucketBuild(response['build']))] |
| 72 mock_other.TriggerTryJobs.return_value = results | 72 mock_module.TriggerTryJobs.return_value = results |
| 73 | 73 |
| 74 try_job_pipeline = ScheduleTryJobPipeline() | 74 try_job_pipeline = ScheduleTryJobPipeline() |
| 75 build_id = try_job_pipeline._TriggerTryJob( | 75 build_id = try_job_pipeline._TriggerTryJob( |
| 76 master_name, builder_name, {}, []) | 76 master_name, builder_name, {}, []) |
| 77 | 77 |
| 78 self.assertEqual(build_id, '1') | 78 self.assertEqual(build_id, '1') |
| 79 | 79 |
| 80 | 80 |
| 81 def testCreateTryJobData(self): | 81 def testCreateTryJobData(self): |
| 82 build_id = 1 | 82 build_id = 1 |
| 83 master_name = 'm' | 83 master_name = 'm' |
| 84 builder_name = 'b' | 84 builder_name = 'b' |
| 85 build_number = 1 | 85 build_number = 1 |
| 86 try_job_type = failure_type.GetDescriptionForFailureType(failure_type.TEST) | 86 try_job_type = failure_type.GetDescriptionForFailureType(failure_type.TEST) |
| 87 | 87 |
| 88 ScheduleTryJobPipeline()._CreateTryJobData( | 88 ScheduleTryJobPipeline()._CreateTryJobData( |
| 89 build_id, master_name, builder_name, build_number, try_job_type, | 89 build_id, master_name, builder_name, build_number, try_job_type, |
| 90 False, False) | 90 False, False) |
| 91 | 91 |
| 92 data = WfTryJobData.Get(build_id) | 92 data = WfTryJobData.Get(build_id) |
| 93 | 93 |
| 94 self.assertIsNotNone(data) | 94 self.assertIsNotNone(data) |
| 95 self.assertEqual(data.master_name, master_name) | 95 self.assertEqual(data.master_name, master_name) |
| 96 self.assertEqual(data.builder_name, builder_name) | 96 self.assertEqual(data.builder_name, builder_name) |
| 97 self.assertEqual(data.build_number, build_number) | 97 self.assertEqual(data.build_number, build_number) |
| 98 self.assertEqual(data.try_job_type, try_job_type) | 98 self.assertEqual(data.try_job_type, try_job_type) |
| 99 self.assertFalse(data.has_compile_targets) | 99 self.assertFalse(data.has_compile_targets) |
| 100 self.assertFalse(data.has_heuristic_results) | 100 self.assertFalse(data.has_heuristic_results) |
| OLD | NEW |