mirror of
https://chromium.googlesource.com/crosvm/crosvm
synced 2025-02-05 18:20:34 +00:00
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
|
#!/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()
|