recipe_modules/resultdb/examples/include.py

# Copyright 2020 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, StepSuccess,
  DoesNotRunRE)

from PB.go.chromium.org.luci.buildbucket.proto import build as build_pb2
from PB.go.chromium.org.luci.resultdb.proto.v1 import invocation as invocation_pb2

DEPS = [
  'buildbucket',
  'resultdb',
]


def RunSteps(api):
  inv_bundle = api.resultdb.chromium_derive(
    step_name='rdb chromium-derive',
    swarming_host='chromium-swarm.appspot.com',
    task_ids=['deadbeef'],
    variants_with_unexpected_results=True,
  )
  invocation_ids = inv_bundle.keys()
  api.resultdb.include_invocations(invocation_ids, step_name='rdb include')
  api.resultdb.exclude_invocations(invocation_ids, step_name='rdb exclude')


def GenTests(api):
  yield (
    api.test('noop') +
    api.buildbucket.ci_build() +
    api.resultdb.chromium_derive(step_name='rdb chromium-derive', results={}) +
    api.post_process(
        DoesNotRunRE, 'rdb include', 'rdb exclude') +
    api.post_process(DropExpectation)
  )
  inv_bundle = {
      'invid': api.resultdb.Invocation(
          proto=invocation_pb2.Invocation(
              state=invocation_pb2.Invocation.FINALIZED),
      ),
      'invid2': api.resultdb.Invocation(
          proto=invocation_pb2.Invocation(
              state=invocation_pb2.Invocation.FINALIZED),
      ),
  }

  yield (
    api.test('basic') +
    api.buildbucket.ci_build() +
    api.resultdb.chromium_derive(
        step_name='rdb chromium-derive', results=inv_bundle)
  )