recipe_modules/context/tests/infra_step.py

# Copyright 2017 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.

PYTHON_VERSION_COMPATIBILITY = 'PY2+3'

DEPS = [
  "context",
  "path",
  "step",
]

def RunSteps(api):
  was_infra_failure = None
  try:
    api.step('boom', ['echo', 'hello'])
  except api.step.InfraFailure:  # pragma: no cover
    assert False, 'impossible'
  except api.step.StepFailure:
    was_infra_failure = False

  assert was_infra_failure is False, 'got: %r' % was_infra_failure

  with api.context(infra_steps=True):
    was_infra_failure = None
    try:
      api.step('boom 2', ['echo', 'hello', 'subdir'])
    except api.step.InfraFailure:
      was_infra_failure = True
    except api.step.StepFailure:  # pragma: no cover
      assert False, 'impossible'
    assert was_infra_failure is True

def GenTests(api):
  yield (
    api.test('basic')
    + api.step_data('boom', retcode=1)
    + api.step_data('boom 2', retcode=1)
  )