Edit

kc3-lang/angle/scripts/trigger.py

Branch :

  • Show log

    Commit

  • Author : Geoff Lang
    Date : 2019-05-06 13:15:35
    Hash : d7d42395
    Message : Format all of ANGLE's python code. BUG=angleproject:3421 Change-Id: I1d7282ac513c046de5d8ed87f7789290780d30a6 Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/1595440 Reviewed-by: Jamie Madill <jmadill@chromium.org> Commit-Queue: Geoff Lang <geofflang@chromium.org>

  • scripts/trigger.py
  • #!/usr/bin/python2
    #
    # Copyright 2019 The ANGLE Project Authors. All rights reserved.
    # Use of this source code is governed by a BSD-style license that can be
    # found in the LICENSE file.
    #
    # trigger.py:
    #   Helper script for triggering GPU tests on swarming.
    
    import argparse
    import os
    import subprocess
    import sys
    
    
    def parse_args():
        parser = argparse.ArgumentParser(os.path.basename(sys.argv[0]))
        parser.add_argument('gn_path', help='path to GN. (e.g. out/Release)')
        parser.add_argument('test', help='test name. (e.g. angle_end2end_tests)')
        parser.add_argument('os_dim', help='OS dimension. (e.g. Windows-10)')
        parser.add_argument('gpu_dim', help='GPU dimension. (e.g. intel-hd-630-win10-stable)')
        parser.add_argument('-s', '--shards', default=1, help='number of shards', type=int)
        parser.add_argument('-p', '--pool', default='Chrome-GPU', help='swarming pool')
        parser.add_argument('extra_args', help='extra test command line arguments', nargs='*')
        return parser.parse_args()
    
    
    def main():
        args = parse_args()
        path = args.gn_path.replace('\\', '/')
        out_gn_path = '//' + path
        out_file_path = os.path.join(*path.split('/'))
    
        mb_script_path = os.path.join('tools', 'mb', 'mb.py')
        subprocess.call(['python', mb_script_path, 'isolate', out_gn_path, args.test])
    
        isolate_script_path = os.path.join('tools', 'swarming_client', 'isolate.py')
        isolate_file = os.path.join(out_file_path, '%s.isolate' % args.test)
        isolated_file = os.path.join(out_file_path, '%s.isolated' % args.test)
    
        isolate_args = [
            'python', isolate_script_path, 'archive', '-I', 'https://isolateserver.appspot.com', '-i',
            isolate_file, '-s', isolated_file
        ]
        stdout = subprocess.check_output(isolate_args)
        sha = stdout[:40]
    
        print('Got an isolated SHA of %s' % sha)
        swarming_script_path = os.path.join('tools', 'swarming_client', 'swarming.py')
    
        swarmings_args = [
            'python', swarming_script_path, 'trigger', '-S', 'chromium-swarm.appspot.com', '-I',
            'isolateserver.appspot.com', '-d', 'os', args.os_dim, '-d', 'pool', args.pool, '-d', 'gpu',
            args.gpu_dim,
            '--shards=%d' % args.shards, '-s', sha
        ]
    
        if args.extra_args:
            swarmings_args += ['--'] + args.extra_args
    
        print(' '.join(swarmings_args))
        subprocess.call(swarmings_args)
        return 0
    
    
    if __name__ == '__main__':
        sys.exit(main())