ci/kokoro/simulate.py

#!/usr/bin/env python3
# Copyright 2022 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from pathlib import Path
import argparse
import os
import subprocess
import tempfile

USAGE="""
Simulates a Kokoro run executing one of the build-* scripts.

Example:
  $ cd ./ci/kokoro
  $ ./simulate.py build-aarch64.sh
"""

CROSVM_ROOT = Path(__file__).parent.parent.parent.resolve()


def git_clone_source(source: Path, destination: Path):
  destination.mkdir(parents=True, exist_ok=True)
  print(f"Cloning {source} into {destination}:")
  subprocess.check_call(['git', 'clone', '-q', source, destination])


def run_kokoro_build_script(kokoro_root: Path, script_path: Path):
  print(f"Running {script_path}:")
  env=os.environ.copy()
  env['KOKORO_ARTIFACTS_DIR'] = str(kokoro_root / 'src')
  subprocess.check_call([script_path.resolve()], cwd=kokoro_root, env=env)


def simulate_kokoro(kokoro_root: Path, script_path: Path):
  git_clone_source(CROSVM_ROOT, kokoro_root / 'src/git/crosvm')
  run_kokoro_build_script(kokoro_root, script_path)


def main():
  parser = argparse.ArgumentParser(usage=USAGE)
  parser.add_argument("script_name", type=Path)
  args = parser.parse_args()

  script_path: Path= args.script_name
  if not script_path.exists():
    raise ValueError(f"Script '{script_path} not found.")

  with tempfile.TemporaryDirectory() as temp_dir:
    simulate_kokoro(Path(temp_dir), script_path)



if __name__ == '__main__':
  main()