# 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/40853548): 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)
Codebase Browser
chromium