kc3-lang/angle/scripts/trigger.py

Branch : - branch - main - tag -

• Show log

  Commit

• Author : Ben Pastene
  Date : 2020-06-11 22:32:55
  Hash : 1496de21
  Message : Remove implied CIPD packages from scripts/trigger.py All of chromium's test pools should have these packages automatically added via task templates. So there's no need to add them ourselves in this script. Bug: chromium:812428 Change-Id: I283ad5963cdf237220545c5ed1f21892f6ade18e Reviewed-on: https://chromium-review.googlesource.com/c/angle/angle/+/2241901 Commit-Queue: Yuly Novikov <ynovikov@chromium.org> Reviewed-by: Yuly Novikov <ynovikov@chromium.org> Reviewed-by: Robbie Iannucci <iannucci@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 hashlib
  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')
      return parser.parse_known_args()
  
  
  def main():
      args, unknown = 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_cmd_path = os.path.join('tools', 'luci-go', 'isolate')
      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 = [
          isolate_cmd_path, 'archive', '-I', 'https://isolateserver.appspot.com', '-i', isolate_file,
          '-s', isolated_file
      ]
      subprocess.check_call(isolate_args)
      with open(isolated_file, 'rb') as f:
          sha = hashlib.sha1(f.read()).hexdigest()
  
      print('Got an isolated SHA of %s' % sha)
      swarming_script_path = os.path.join('tools', 'luci-go', 'swarming')
  
      swarming_args = [
          swarming_script_path, 'trigger', '-S', 'chromium-swarm.appspot.com', '-I',
          'https://isolateserver.appspot.com', '-d', 'os=' + args.os_dim, '-d', 'pool=' + args.pool,
          '-d', 'gpu=' + args.gpu_dim, '-s', sha
      ]
  
      for i in range(args.shards):
          shard_args = swarming_args[:]
          shard_args.extend([
              '--env',
              'GTEST_TOTAL_SHARDS=%d' % args.shards,
              '--env',
              'GTEST_SHARD_INDEX=%d' % i,
          ])
          if unknown:
              shard_args += ["--"] + unknown
  
          print(' '.join(shard_args))
          subprocess.call(shard_args)
      return 0
  
  
  if __name__ == '__main__':
      sys.exit(main())