| # Copyright 2019 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 json |
| |
| DEPS = [ |
| 'chromium_swarming', |
| 'recipe_engine/assertions', |
| 'recipe_engine/platform', |
| 'recipe_engine/properties', |
| 'recipe_engine/step', |
| ] |
| |
| from recipe_engine import post_process |
| from recipe_engine.recipe_api import Property |
| |
| PROPERTIES = { |
| 'task_to_retry': Property(default=None, kind=dict), |
| 'expected_value': Property(default=None, kind=str), |
| } |
| |
| |
| def RunSteps(api, task_to_retry, expected_value): |
| kwargs = {} |
| if task_to_retry: |
| class FakeTask: |
| def __init__(self): |
| self.trigger_output = task_to_retry |
| kwargs['task_to_retry'] = FakeTask() |
| task = api.chromium_swarming.task(name='test-task', |
| isolated='00deadbeef00', |
| **kwargs) |
| task._trigger_output = { |
| 'tasks': { |
| 0: { |
| 'shard_index': 0, |
| 'task_id': '10', |
| }, |
| 1: { |
| 'shard_index': 1, |
| 'task_id': '11', |
| }, |
| 2: { |
| 'shard_index': 2, |
| 'task_id': '12', |
| }, |
| }, |
| } |
| api.assertions.assertEqual( |
| ' '.join(t['task_id'] for t in task.trigger_output['tasks'].values()), |
| expected_value) |
| |
| |
| def GenTests(api): |
| yield api.test( |
| 'basic', |
| api.properties(expected_value='10 11 12'), |
| api.post_process(post_process.DropExpectation), |
| ) |
| |
| yield api.test( |
| 'task_to_retry', |
| api.properties( |
| task_to_retry={ |
| 'tasks': { |
| 0: { |
| 'shard_index': 0, |
| 'task_id': '90', |
| }, |
| 1: { |
| 'shard_index': 1, |
| 'task_id': '91', |
| }, |
| 2: { |
| 'shard_index': 2, |
| 'task_id': '92', |
| }, |
| 3: { |
| 'shard_index': 3, |
| 'task_id': '93', |
| }, |
| 4: { |
| 'shard_index': 4, |
| 'task_id': '94', |
| }, |
| }, |
| }, |
| expected_value='10 11 12 93 94'), |
| api.post_process(post_process.DropExpectation), |
| ) |
