62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
|
# Copyright 2022 The Chromium Authors
|
||
|
|
# Use of this source code is governed by a BSD-style license that can be
|
||
|
|
# found in the LICENSE file.
|
||
|
|
"""Implements commands for running GPU tests."""
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
from typing import List, Optional
|
||
|
|
|
||
|
|
from common import DIR_SRC_ROOT
|
||
|
|
from test_runner import TestRunner
|
||
|
|
|
||
|
|
_GPU_TEST_SCRIPT = os.path.join(DIR_SRC_ROOT, 'content', 'test', 'gpu',
|
||
|
|
'run_gpu_integration_test.py')
|
||
|
|
_PERF_TEST_SCRIPT = os.path.join(DIR_SRC_ROOT, 'tools', 'perf',
|
||
|
|
'run_benchmark')
|
||
|
|
|
||
|
|
|
||
|
|
class TelemetryTestRunner(TestRunner):
|
||
|
|
"""Test runner for running GPU tests."""
|
||
|
|
|
||
|
|
def __init__(self, test_type: str, out_dir: str, test_args: List[str],
|
||
|
|
target_id: Optional[str]) -> None:
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument(
|
||
|
|
'--browser', help='The browser to use for Telemetry based tests.')
|
||
|
|
args, _ = parser.parse_known_args(test_args)
|
||
|
|
|
||
|
|
if args.browser == 'web-engine-shell':
|
||
|
|
packages = ['web_engine_shell']
|
||
|
|
elif args.browser == 'fuchsia-chrome':
|
||
|
|
packages = ['chrome']
|
||
|
|
elif args.browser == 'cast-streaming-shell':
|
||
|
|
packages = ['cast_streaming_shell']
|
||
|
|
else:
|
||
|
|
raise Exception('Unknown browser %s' % args.browser)
|
||
|
|
|
||
|
|
if test_type == 'gpu':
|
||
|
|
self._test_script = _GPU_TEST_SCRIPT
|
||
|
|
elif test_type == 'perf':
|
||
|
|
self._test_script = _PERF_TEST_SCRIPT
|
||
|
|
else:
|
||
|
|
raise ValueError('Test type can only be |gpu| or |perf|.')
|
||
|
|
|
||
|
|
super().__init__(out_dir, test_args, packages, target_id)
|
||
|
|
|
||
|
|
# TODO(crbug.com/1345390): Remove when Telemetry tests use CFv2 components.
|
||
|
|
@staticmethod
|
||
|
|
def is_cfv2() -> bool:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def run_test(self):
|
||
|
|
test_cmd = [self._test_script]
|
||
|
|
if self._test_args:
|
||
|
|
test_cmd.extend(self._test_args)
|
||
|
|
test_cmd.extend(['--chromium-output-directory', self._out_dir])
|
||
|
|
if self._target_id:
|
||
|
|
test_cmd.extend(['--fuchsia-target-id', self._target_id])
|
||
|
|
return subprocess.run(test_cmd, check=True)
|