recipe_modules/step/tests/empty.py

# 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.post_process import (
    DropExpectation,
    HasLog,
    LogEquals,
    StepCommandEmpty,
    StepException,
    StepFailure,
    StepSuccess,
    StepTextEquals
)

DEPS = [
  'step',
]


def RunSteps(api):
  api.step.empty('hello', step_text='stuff', log_text='other\nstuff')

  api.step.empty('multi hello', log_text=['multi', 'line'])

  api.step.empty(
      'alternate log', log_name='details', log_text='some\nlog\ncontent')

  try:
    api.step.empty('bye', status=api.step.FAILURE)
    assert False, 'unreachable'  # pragma: no cover
  except api.step.StepFailure:
    pass

  try:
    api.step.empty('bigfail', status=api.step.INFRA_FAILURE)
    assert False, 'unreachable'  # pragma: no cover
  except api.step.StepFailure:
    pass

  api.step.empty('quiet fail', status=api.step.FAILURE, raise_on_failure=False)

def GenTests(api):
  yield api.test(
      'basic',
      api.post_process(StepSuccess, 'hello'),
      api.post_process(HasLog, 'hello', 'stdout'),
      api.post_process(LogEquals, 'hello', 'stdout', 'other\nstuff'),
      api.post_process(StepTextEquals, 'hello', 'stuff'),
      api.post_process(StepCommandEmpty, 'hello'),

      api.post_process(StepSuccess, 'multi hello'),
      api.post_process(LogEquals, 'multi hello', 'stdout', 'multi\nline'),

      api.post_process(StepSuccess, 'alternate log'),
      api.post_process(LogEquals, 'alternate log', 'details',
                       'some\nlog\ncontent'),

      api.post_process(StepException, 'bigfail'),

      api.post_process(StepFailure, 'quiet fail'),

      api.post_process(DropExpectation),
  )