mirrored from https://chromium.googlesource.com/infra/luci/recipes-py
-
Notifications
You must be signed in to change notification settings - Fork 17
/
raise_on_failure.py
67 lines (56 loc) · 2.02 KB
/
raise_on_failure.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Copyright 2021 The LUCI Authors. All rights reserved.
# Use of this source code is governed under the Apache License, Version 2.0
# that can be found in the LICENSE file.
from recipe_engine import recipe_api, recipe_test_api, post_process
DEPS = [
'properties',
'step',
]
PROPERTIES = {
'infra_step': recipe_api.Property(default=False),
'set_status_to_exception': recipe_api.Property(default=False),
}
def RunSteps(api, infra_step, set_status_to_exception):
def failure_step_test_data():
test_data = recipe_test_api.StepTestData()
test_data.retcode = 1
return test_data
result = api.step(
'non-raising step',
['bash', '-c', 'exit 1'],
infra_step=infra_step,
raise_on_failure=False,
step_test_data=failure_step_test_data)
status = None
if set_status_to_exception:
status = result.presentation.status
result.presentation.status = api.step.EXCEPTION
api.step('in-between step', [])
api.step.raise_on_failure(result, status_override=status)
def GenTests(api):
yield api.test(
'basic',
api.post_process(post_process.MustRun, 'in-between step'),
api.post_process(post_process.StepFailure, 'non-raising step'),
api.post_process(post_process.StatusFailure),
api.post_process(post_process.DropExpectation),
status='FAILURE',
)
yield api.test(
'infra-step',
api.properties(infra_step=True),
api.post_process(post_process.MustRun, 'in-between step'),
api.post_process(post_process.StepException, 'non-raising step'),
api.post_process(post_process.StatusException),
api.post_process(post_process.DropExpectation),
status='INFRA_FAILURE',
)
yield api.test(
'changed-status',
api.properties(set_status_to_exception=True),
api.post_process(post_process.MustRun, 'in-between step'),
api.post_process(post_process.StepException, 'non-raising step'),
api.post_process(post_process.StatusFailure),
api.post_process(post_process.DropExpectation),
status='FAILURE',
)