recipes/recipe_modules/infra_cipd/examples/usage.py

# Copyright 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import json

from recipe_engine import post_process

PYTHON_VERSION_COMPATIBILITY = 'PY2+3'

DEPS = [
    'recipe_engine/assertions',
    'recipe_engine/buildbucket',
    'recipe_engine/context',
    'recipe_engine/json',
    'recipe_engine/path',
    'recipe_engine/platform',
    'recipe_engine/properties',
    'recipe_engine/step',
    'infra_cipd',
]


def RunSteps(api):
  url = 'https://chromium.googlesource.com/infra/infra'
  rev = 'deadbeef' * 5
  # Assume path is where infra/infra is repo is checked out.
  path = api.path['cache'].join('builder', 'assume', 'infra')
  with api.infra_cipd.context(
      path_to_repo=path,
      goos=api.properties.get('goos'),
      goarch=api.properties.get('goarch')):
    if api.platform.is_mac:
      api.infra_cipd.build_without_env_refresh(
          api.properties.get('signing_identity'))
    else:
      api.infra_cipd.build_without_env_refresh()
    api.infra_cipd.test()
    if not api.properties.get('no_buildnumbers'):
      api.infra_cipd.upload(api.infra_cipd.tags(url, rev))
    else:
      with api.assertions.assertRaises(ValueError):
        api.infra_cipd.upload(api.infra_cipd.tags(url, rev))


def GenTests(api):
  yield (api.test('luci-native') + api.buildbucket.ci_build(
      'infra-internal', 'ci', 'native', build_number=5))
  yield (api.test('luci-native_codesign') + api.platform.name('mac') +
         api.properties(
             signing_identity='AAAAAAAAAAAAABBBBBBBBBBBBBXXXXXXXXXXXXXX') +
         api.buildbucket.ci_build(
             'infra-internal', 'ci', 'native', build_number=5))
  yield (api.test('luci-cross') + api.properties(
      goos='linux',
      goarch='arm64',
  ) + api.buildbucket.ci_build('infra-internal', 'ci', 'cross', build_number=5))
  yield (
    api.test('no-buildnumbers') +
    api.properties(
      no_buildnumbers=True,
    ) +
    api.buildbucket.ci_build('infra-internal', 'ci', 'just-build-and-test') +
    api.post_process(post_process.DropExpectation))