tools/run_test_like_ci

#!/usr/bin/env python
# Copyright (C) 2019 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from __future__ import print_function
import argparse
import sys
import shutil
import subprocess
import os

REPO_ROOT = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(os.path.join(REPO_ROOT, 'infra', 'ci'))
from config import JOB_CONFIGS, SANDBOX_IMG


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('config', choices=JOB_CONFIGS.keys())
    args = parser.parse_args()

    # Check that the directory is clean.
    git_cmd = ['git', '-C', REPO_ROOT, 'status', '--porcelain']
    modified_files = subprocess.check_output(git_cmd)
    if modified_files:
      print('The current Git repo has modified/untracked files.')
      print('The sandboxed VM will fetch the HEAD of your current git repo.')
      print('This is probably not the state you want to be in.')
      print('I suggest you press CTRL+C, commit and then re-run this script')
      print('Modified files:\n' + modified_files)
      raw_input('If you think you know what you are doing, press Enter instead')

    env = {
        'PERFETTO_TEST_GIT_REF': 'file:///ci/source',
    }
    env.update(JOB_CONFIGS[args.config])

    workdir = os.path.join(REPO_ROOT, 'out', 'tmp.ci')
    cmd = [
        'sudo', '--', 'docker', 'run', '-it', '--name', 'perfetto_ci',
        '--cap-add', 'SYS_PTRACE', '--rm', '--volume',
        '%s:/ci/ramdisk' % workdir, '--tmpfs', '/tmp:exec',
        '--volume=%s:/ci/source:ro' % REPO_ROOT
    ]
    for kv in env.items():
        cmd += ['--env', '%s=%s' % kv]
    cmd += [SANDBOX_IMG]
    cmd += [
        'bash', '-c',
        'cd /ci/ramdisk; bash /ci/init.sh || sudo -u perfetto -EH bash -i'
    ]

    print('About to run\n', ' '.join(cmd))
    print('')
    print('The VM will have read-only acess to: %s, mounted as /ci/source' %
          REPO_ROOT)
    print('The VM workdir /ci/ramdisk will be mounted into: %s' % workdir)
    print(
        'The contents of %s will be deleted before starting the VM' % workdir)
    raw_input('Press a key to continue')

    shutil.rmtree(workdir, ignore_errors=True)
    os.makedirs(workdir)
    os.execvp(cmd[0], cmd[1:])


if __name__ == '__main__':
    sys.exit(main())