| # Copyright 2020 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. |
| |
| from recipe_engine.recipe_api import Property |
| |
| DEPS = [ |
| 'flutter/retry', |
| 'recipe_engine/properties', |
| 'recipe_engine/raw_io', |
| 'recipe_engine/step', |
| 'recipe_engine/context', |
| ] |
| |
| PROPERTIES = { |
| "max_attempts": |
| Property( |
| kind=int, |
| help="How many times to try before giving up.", |
| default=1, |
| ), |
| } |
| |
| |
| def RunSteps(api, max_attempts): |
| api.retry.step( |
| 'test: Run FEMU Test Suite', [ |
| 'echo', 'hello', '>', |
| api.raw_io.output_text(name='syslog'), ';', 'echo', 'hello', '>', |
| api.raw_io.output_text(name='emulator_log') |
| ], |
| max_attempts=max_attempts, |
| step_test_data=( |
| lambda: api.raw_io.test_api.output_text('failure', name='syslog') |
| ) |
| ) |
| api.retry.step('test: mytest', ['ls', '-la'], max_attempts=max_attempts) |
| |
| def func1(): |
| with api.step.nest('nested'): |
| api.step('test: mytest_func', ['ls', '-a']) |
| |
| # pylint: disable=unused-argument |
| def func2(timeout=None): |
| api.step('test: mytest_func_basic', ['ls', '-a']) |
| |
| def func3(): |
| api.step('test: mytest_func_3', ['ls', '-a']) |
| |
| api.retry.wrap( |
| func1, step_name='nested.test: mytest_func', max_attempts=max_attempts |
| ) |
| api.retry.basic_wrap(func2, max_attempts=max_attempts) |
| api.retry.run_flutter_doctor() |
| api.retry.wrap(func3, max_attempts=max_attempts, retriable_codes=(1,)) |
| |
| |
| def GenTests(api): |
| yield api.test('passing', api.properties(max_attempts=1)) |
| yield api.test( |
| 'failing_step', |
| api.properties(max_attempts=1), |
| api.step_data('test: mytest', retcode=1), |
| status='FAILURE' |
| ) |
| yield api.test( |
| 'failing_wrap', |
| api.properties(max_attempts=1), |
| api.step_data('nested.test: mytest_func', retcode=1), |
| status='FAILURE' |
| ) |
| yield api.test( |
| 'failing_wrap_with_nested', |
| api.properties(max_attempts=1), |
| api.step_data('test: mytest_func_3', retcode=1), |
| status='FAILURE' |
| ) |
| yield api.test( |
| 'failing_basic_wrap', |
| api.properties(max_attempts=1), |
| api.step_data('test: mytest_func_basic', retcode=1), |
| status='FAILURE' |
| ) |
| yield api.test( |
| 'pass_with_retries', api.properties(max_attempts=2), |
| api.step_data('test: mytest', retcode=1), |
| api.step_data('nested.test: mytest_func', retcode=1), |
| api.step_data('test: mytest_func_basic', retcode=1) |
| ) |
