recipe_modules/step/tests/step_call_args.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.

from recipe_engine import post_process

PYTHON_VERSION_COMPATIBILITY = 'PY2+3'

DEPS = [
  'step',
]


def RunSteps(api):
  try:
    api.step('bad cmd', 'I\'m a str cmd')
  except ValueError as e:
    assert str(e) == 'cmd must be a list, got "I\'m a str cmd"', e

  try:
    api.step('bad arg', [{}])
  except ValueError as e:
    assert '\'dict\'> is not permitted. cmd is [{}]' in str(e), e

  try:
    api.step('bad cost', None, cost='I\'m a str cost')
  except ValueError as e:
    assert str(e) == (
      'cost must be a None or ResourceCost , got "I\'m a str cost"'), e


def GenTests(api):
  yield (
    api.test('basic') +
    api.post_process(post_process.DropExpectation)
  )