| 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 json | 5 import json |
| 6 | 6 |
| 7 from common import buildbucket_client | 7 from common import buildbucket_client |
| 8 from common.git_repository import GitRepository | 8 from common.git_repository import GitRepository |
| 9 from model import wf_analysis_status | 9 from model import analysis_status |
| 10 from model.wf_try_job import WfTryJob | 10 from model.wf_try_job import WfTryJob |
| 11 from pipeline_wrapper import pipeline_handlers | 11 from pipeline_wrapper import pipeline_handlers |
| 12 from waterfall.test import wf_testcase | 12 from waterfall.test import wf_testcase |
| 13 from waterfall.try_job_pipeline import TryJobPipeline | 13 from waterfall.try_job_pipeline import TryJobPipeline |
| 14 from waterfall.try_job_type import TryJobType | 14 from waterfall.try_job_type import TryJobType |
| 15 | 15 |
| 16 | 16 |
| 17 class TryJobPipelineTest(wf_testcase.WaterfallTestCase): | 17 class TryJobPipelineTest(wf_testcase.WaterfallTestCase): |
| 18 app_module = pipeline_handlers._APP | 18 app_module = pipeline_handlers._APP |
| 19 | 19 |
| (...skipping 123 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 143 build_number = 1 | 143 build_number = 1 |
| 144 | 144 |
| 145 WfTryJob.Create(master_name, builder_name, build_number).put() | 145 WfTryJob.Create(master_name, builder_name, build_number).put() |
| 146 | 146 |
| 147 root_pipeline = TryJobPipeline( | 147 root_pipeline = TryJobPipeline( |
| 148 master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'], | 148 master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'], |
| 149 TryJobType.COMPILE, []) | 149 TryJobType.COMPILE, []) |
| 150 root_pipeline._LogUnexpectedAbort(True) | 150 root_pipeline._LogUnexpectedAbort(True) |
| 151 | 151 |
| 152 try_job = WfTryJob.Get(master_name, builder_name, build_number) | 152 try_job = WfTryJob.Get(master_name, builder_name, build_number) |
| 153 self.assertEqual(wf_analysis_status.ERROR, try_job.status) | 153 self.assertEqual(analysis_status.ERROR, try_job.status) |
| 154 | 154 |
| 155 def testPipelineAbortedWithOutTryJobResult(self): | 155 def testPipelineAbortedWithOutTryJobResult(self): |
| 156 master_name = 'm' | 156 master_name = 'm' |
| 157 builder_name = 'b' | 157 builder_name = 'b' |
| 158 build_number = 1 | 158 build_number = 1 |
| 159 | 159 |
| 160 root_pipeline = TryJobPipeline( | 160 root_pipeline = TryJobPipeline( |
| 161 master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'], | 161 master_name, builder_name, build_number, 'rev1', 'rev2', ['rev2'], |
| 162 TryJobType.COMPILE, []) | 162 TryJobType.COMPILE, []) |
| 163 root_pipeline._LogUnexpectedAbort(True) | 163 root_pipeline._LogUnexpectedAbort(True) |
| 164 | 164 |
| 165 try_job = WfTryJob.Get(master_name, builder_name, build_number) | 165 try_job = WfTryJob.Get(master_name, builder_name, build_number) |
| 166 | 166 |
| 167 self.assertIsNone(try_job) | 167 self.assertIsNone(try_job) |
| OLD | NEW |