| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 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 from model.wf_swarming_task import WfSwarmingTask | 5 from model.wf_swarming_task import WfSwarmingTask |
| 6 from waterfall.test import wf_testcase | 6 from waterfall.test import wf_testcase |
| 7 from waterfall.trigger_swarming_task_pipeline import TriggerSwarmingTaskPipeline | 7 from waterfall.trigger_swarming_task_pipeline import TriggerSwarmingTaskPipeline |
| 8 | 8 |
| 9 | 9 |
| 10 class TriggerSwarmingTaskPipelineTest(wf_testcase.WaterfallTestCase): | 10 class TriggerSwarmingTaskPipelineTest(wf_testcase.WaterfallTestCase): |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 52 self.assertEqual(step_name, task.step_name) | 52 self.assertEqual(step_name, task.step_name) |
| 53 | 53 |
| 54 def testGetIterationsToRerun(self): | 54 def testGetIterationsToRerun(self): |
| 55 expected_iterations = 50 | 55 expected_iterations = 50 |
| 56 self.UpdateUnitTestConfigSettings( | 56 self.UpdateUnitTestConfigSettings( |
| 57 config_property='swarming_settings', | 57 config_property='swarming_settings', |
| 58 override_data={'iterations_to_rerun': expected_iterations}) | 58 override_data={'iterations_to_rerun': expected_iterations}) |
| 59 self.assertEqual( | 59 self.assertEqual( |
| 60 expected_iterations, | 60 expected_iterations, |
| 61 TriggerSwarmingTaskPipeline()._GetIterationsToRerun()) | 61 TriggerSwarmingTaskPipeline()._GetIterationsToRerun()) |
| 62 | |
| 63 | |
| 64 | |
| OLD | NEW |