recipe_modules/flutter_bcid/examples/full.py

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

DEPS = [
    'flutter/flutter_bcid',
    'recipe_engine/buildbucket',
    'recipe_engine/path',
    'recipe_engine/raw_io',
    'recipe_engine/runtime',
]


def RunSteps(api):
  api.flutter_bcid.report_stage('one')
  api.flutter_bcid.upload_provenance(
      api.path.cache_dir / 'file.zip', 'gs://bucket/final_path/file.txt'
  )
  api.flutter_bcid.is_official_build()
  api.flutter_bcid.is_prod_build()
  api.flutter_bcid.is_try_build()
  api.flutter_bcid.is_signable_build()
  api.flutter_bcid.download_and_verify_provenance(
      "artifact.zip", "flutter_infra", "release_artifacts/artifacts.zip"
  )


def GenTests(api):
  fake_bcid_response_success = '{"allowed": true, "verificationSummary": "This artifact is definitely legitimate!"}'
  artifacts_location = 'artifact.zip'
  yield api.test(
      'release_build',
      api.buildbucket.ci_build(
          project='dart-internal',
          bucket='flutter',
          git_repo='https://dart.googlesource.com/monorepo',
          git_ref='refs/heads/main'
      ),
      api.step_data(
          'Verify artifact.zip provenance.verify artifact.zip provenance',
          stdout=api.raw_io.output_text(fake_bcid_response_success)
      ),
  )

  yield api.test(
      'try_build',
      api.buildbucket.ci_build(
          project='flutter',
          bucket='try',
          git_repo='https://dart.googlesource.com/monorepo',
          git_ref='refs/heads/main'
      ),
  )

  yield api.test(
      'prod_build',
      api.buildbucket.ci_build(
          project='flutter',
          bucket='prod',
          git_repo='https://dart.googlesource.com/monorepo',
          git_ref='refs/heads/main'
      ),
  )

  yield api.test(
      'prod_build_experimental',
      api.runtime(is_experimental=True),
      api.buildbucket.ci_build(
          project='flutter',
          bucket='prod',
          git_repo='https://dart.googlesource.com/monorepo',
          git_ref='refs/heads/main'
      ),
  )