recipe_modules/cq/tests/owner_is_googler.py

# Copyright 2022 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

DEPS = [
  'assertions',
  'buildbucket',
  'cq',
  'properties',
]


def RunSteps(api):
  api.assertions.assertEqual(
       api.cq.owner_is_googler,  api.properties['expected_owner_is_googler'])


def GenTests(api):
  yield (
    api.test('default')
    + api.cq(run_mode=api.cq.FULL_RUN)
    + api.buildbucket.try_build(project='chrome')
    + api.properties(expected_owner_is_googler=False)
    + api.post_process(post_process.DropExpectation)
  )
  yield (
    api.test('is googler')
    + api.cq(run_mode=api.cq.FULL_RUN, owner_is_googler=True)
    + api.buildbucket.try_build(project='chrome')
    + api.properties(expected_owner_is_googler=True)
    + api.post_process(post_process.DropExpectation)
  )
  yield (
    api.test('chrome-branch-project')
    + api.cq(run_mode=api.cq.FULL_RUN)
    + api.buildbucket.try_build(project='chrome-m100')
    + api.properties(expected_owner_is_googler=False)
    + api.post_process(post_process.DropExpectation)
  )
  yield (
    api.test('raise if not Chrome')
    + api.cq(run_mode=api.cq.FULL_RUN)
    + api.buildbucket.try_build(project='infra')
    + api.expect_exception('ValueError')
    + api.post_process(post_process.DropExpectation)
  )