Index: scripts/slave/recipe_modules/step/api.py |
diff --git a/scripts/slave/recipe_modules/step/api.py b/scripts/slave/recipe_modules/step/api.py |
index b25aa916aaf07e29dc1a7d901a760d3bf2958409..b806c2253f95b7624f051fca8a81ee371b68f6f4 100644 |
--- a/scripts/slave/recipe_modules/step/api.py |
+++ b/scripts/slave/recipe_modules/step/api.py |
@@ -3,6 +3,7 @@ |
# found in the LICENSE file. |
from slave import recipe_api |
+from slave import recipe_util |
class StepApi(recipe_api.RecipeApi): |
def __init__(self, *args, **kwargs): |
@@ -37,10 +38,10 @@ class StepApi(recipe_api.RecipeApi): |
assert isinstance(cmd, list) |
if kwargs.get('abort_on_failure', False): |
- @recipe_api.wrap_followup(kwargs) |
+ @recipe_util.wrap_followup(kwargs) |
def assert_success(step_result): |
if step_result.presentation.status == 'FAILURE': |
- raise recipe_api.RecipeAbort( |
+ raise recipe_util.RecipeAbort( |
"Step(%s) failed and was marked as abort_on_failure" % name) |
kwargs['followup_fn'] = assert_success |