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

DEPS = [
  'futures',
  'step',
]


def RunSteps(api):
  # This normal step checks against regressions in the greenlet spawning
  # function. Previously the engine would accidentally add the spawned greenlet
  # to this normal step, resulting in deadlock at `fut.result()` below.
  api.step('normal step', ['echo', 'I am pretty normal'])

  fut = api.futures.spawn(api.step, 'do work', cmd=['something'])
  if fut.exception():
    assert isinstance(fut.exception(), api.step.StepFailure), (
      'Some other exception?')
  fut.result()

  assert fut.done, 'What? The future must be done after getting its result.'

  api.step('run if success', cmd=None)

def GenTests(api):
  yield (
    api.test('success')
    + api.post_check(lambda check, steps: check(
        'run if success' in steps
    ))
  )

  yield (
    api.test('failure')
    + api.step_data('do work', retcode=1)
    + api.post_check(lambda check, steps: check(
        steps['do work'].status == 'FAILURE'
    ))
    + api.post_check(lambda check, steps: check(
        'run if success' not in steps
    ))
  )