133 lines
5.3 KiB
Python
133 lines
5.3 KiB
Python
# Copyright 2016 The Chromium OS Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
# repohooks/pre-upload.py currently does not run pylint. But for developers who
|
|
# want to check their code manually we disable several harmless pylint warnings
|
|
# which just distract from more serious remaining issues.
|
|
#
|
|
# The instance variable _android_gts is not defined in __init__().
|
|
# pylint: disable=attribute-defined-outside-init
|
|
#
|
|
# Many short variable names don't follow the naming convention.
|
|
# pylint: disable=invalid-name
|
|
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
from autotest_lib.server import utils
|
|
from autotest_lib.server.cros.tradefed import tradefed_test
|
|
|
|
# Maximum default time allowed for each individual GTS module.
|
|
_GTS_TIMEOUT_SECONDS = 3600
|
|
_PARTNER_GTS_BUCKET = 'gs://chromeos-partner-gts/'
|
|
_PARTNER_GTS_LOCATION = _PARTNER_GTS_BUCKET + 'gts-8-R2-6955212.zip'
|
|
_PARTNER_GTS_AUTHKEY = _PARTNER_GTS_BUCKET + 'gts-arc.json'
|
|
_GTS_MEDIA_URI = ('https://storage.googleapis.com/youtube-test-media/gts/' +
|
|
'GtsYouTubeTestCases-media-1.2.zip')
|
|
_GTS_MEDIA_LOCALPATH = '/tmp/android-gts-media/GtsYouTubeTestCases'
|
|
|
|
|
|
class cheets_GTS(tradefed_test.TradefedTest):
|
|
"""Sets up tradefed to run GTS tests."""
|
|
version = 1
|
|
|
|
_SHARD_CMD = '--shard-count'
|
|
|
|
def _tradefed_retry_command(self, template, session_id):
|
|
"""Build tradefed 'retry' command from template."""
|
|
cmd = []
|
|
for arg in template:
|
|
cmd.append(arg.format(session_id=session_id))
|
|
return cmd
|
|
|
|
def _tradefed_run_command(self, template):
|
|
"""Build tradefed 'run' command from template."""
|
|
cmd = template[:]
|
|
# If we are running outside of the lab we can collect more data.
|
|
if not utils.is_in_container():
|
|
logging.info('Running outside of lab, adding extra debug options.')
|
|
cmd.append('--log-level-display=DEBUG')
|
|
|
|
return cmd
|
|
|
|
def _get_default_bundle_url(self, bundle):
|
|
return _PARTNER_GTS_LOCATION
|
|
|
|
def _get_default_authkey(self):
|
|
return _PARTNER_GTS_AUTHKEY
|
|
|
|
def _get_tradefed_base_dir(self):
|
|
return 'android-gts'
|
|
|
|
def _tradefed_cmd_path(self):
|
|
return os.path.join(self._repository, 'tools', 'gts-tradefed')
|
|
|
|
def _tradefed_env(self):
|
|
if self._authkey:
|
|
return dict(os.environ, APE_API_KEY=self._authkey)
|
|
return None
|
|
|
|
def run_once(self,
|
|
test_name,
|
|
run_template,
|
|
retry_template=None,
|
|
target_module=None,
|
|
target_plan=None,
|
|
needs_push_media=False,
|
|
enable_default_apps=False,
|
|
executable_test_count=None,
|
|
precondition_commands=[],
|
|
login_precondition_commands=[],
|
|
authkey=None,
|
|
prerequisites=[],
|
|
timeout=_GTS_TIMEOUT_SECONDS):
|
|
"""Runs the specified GTS once, but with several retries.
|
|
|
|
Run an arbitrary tradefed command.
|
|
|
|
@param test_name: the name of test. Used for logging.
|
|
@param run_template: the template to construct the run command.
|
|
Example: ['run', 'commandAndExit', 'cts',
|
|
'--skip-media-download']
|
|
@param retry_template: the template to construct the retry command.
|
|
Example: ['run', 'commandAndExit', 'retry',
|
|
'--skip-media-download', '--retry',
|
|
'{session_id}']
|
|
@param target_module: the name of test module to run.
|
|
@param target_plan: the name of the test plan to run.
|
|
@param needs_push_media: need to push test media streams.
|
|
@param executable_test_count: the known number of tests in the run.
|
|
@param timeout: time after which tradefed can be interrupted.
|
|
@param precondition_commands: a list of scripts to be run on the
|
|
dut before the test is run, the scripts must already be installed.
|
|
@param login_precondition_commands: a list of scripts to be run on the
|
|
dut before the log-in for the test is performed.
|
|
@param prerequisites: a list of prerequisites that identify rogue DUTs.
|
|
"""
|
|
# Download the GTS auth key to the local temp directory.
|
|
tmpdir = tempfile.mkdtemp()
|
|
try:
|
|
self._authkey = self._download_to_dir(
|
|
authkey or self._get_default_authkey(), tmpdir)
|
|
|
|
self._run_tradefed_with_retries(
|
|
test_name=test_name,
|
|
run_template=run_template,
|
|
retry_template=retry_template,
|
|
timeout=timeout,
|
|
target_module=target_module,
|
|
target_plan=target_plan,
|
|
media_asset=tradefed_test.MediaAsset(
|
|
_GTS_MEDIA_URI if needs_push_media else None,
|
|
_GTS_MEDIA_LOCALPATH),
|
|
enable_default_apps=enable_default_apps,
|
|
executable_test_count=executable_test_count,
|
|
login_precondition_commands=login_precondition_commands,
|
|
precondition_commands=precondition_commands,
|
|
prerequisites=prerequisites)
|
|
finally:
|
|
shutil.rmtree(tmpdir)
|