Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1969)

Unified Diff: build/android/pylib/perf/test_runner.py

Issue 2392643003: Removes files from //build that we don't need (Closed)
Patch Set: Created 4 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « build/android/pylib/perf/test_options.py ('k') | build/android/pylib/perf/thermal_throttle.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: build/android/pylib/perf/test_runner.py
diff --git a/build/android/pylib/perf/test_runner.py b/build/android/pylib/perf/test_runner.py
deleted file mode 100644
index d21a9b71b10448654ad672ec0cd75be385924723..0000000000000000000000000000000000000000
--- a/build/android/pylib/perf/test_runner.py
+++ /dev/null
@@ -1,374 +0,0 @@
-# Copyright 2013 The Chromium Authors. All rights reserved.
-# Use of this source code is governed by a BSD-style license that can be
-# found in the LICENSE file.
-
-"""Runs perf tests.
-
-Our buildbot infrastructure requires each slave to run steps serially.
-This is sub-optimal for android, where these steps can run independently on
-multiple connected devices.
-
-The buildbots will run this script multiple times per cycle:
-- First: all steps listed in --steps in will be executed in parallel using all
-connected devices. Step results will be pickled to disk. Each step has a unique
-name. The result code will be ignored if the step name is listed in
---flaky-steps.
-The buildbot will treat this step as a regular step, and will not process any
-graph data.
-
-- Then, with -print-step STEP_NAME: at this stage, we'll simply print the file
-with the step results previously saved. The buildbot will then process the graph
-data accordingly.
-
-The JSON steps file contains a dictionary in the format:
-{ "version": int,
- "steps": {
- "foo": {
- "device_affinity": int,
- "cmd": "script_to_execute foo"
- },
- "bar": {
- "device_affinity": int,
- "cmd": "script_to_execute bar"
- }
- }
-}
-
-The JSON flaky steps file contains a list with step names which results should
-be ignored:
-[
- "step_name_foo",
- "step_name_bar"
-]
-
-Note that script_to_execute necessarily have to take at least the following
-option:
- --device: the serial number to be passed to all adb commands.
-"""
-
-import collections
-import datetime
-import json
-import logging
-import os
-import pickle
-import shutil
-import sys
-import tempfile
-import threading
-import time
-
-from pylib import cmd_helper
-from pylib import constants
-from pylib import forwarder
-from pylib.base import base_test_result
-from pylib.base import base_test_runner
-from pylib.device import battery_utils
-from pylib.device import device_errors
-
-
-def GetPersistedResult(test_name):
- file_name = os.path.join(constants.PERF_OUTPUT_DIR, test_name)
- if not os.path.exists(file_name):
- logging.error('File not found %s', file_name)
- return None
-
- with file(file_name, 'r') as f:
- return pickle.loads(f.read())
-
-
-def OutputJsonList(json_input, json_output):
- with file(json_input, 'r') as i:
- all_steps = json.load(i)
-
- step_values = []
- for k, v in all_steps['steps'].iteritems():
- data = {'test': k, 'device_affinity': v['device_affinity']}
-
- persisted_result = GetPersistedResult(k)
- if persisted_result:
- data['total_time'] = persisted_result['total_time']
- step_values.append(data)
-
- with file(json_output, 'w') as o:
- o.write(json.dumps(step_values))
- return 0
-
-
-def PrintTestOutput(test_name, json_file_name=None):
- """Helper method to print the output of previously executed test_name.
-
- Args:
- test_name: name of the test that has been previously executed.
- json_file_name: name of the file to output chartjson data to.
-
- Returns:
- exit code generated by the test step.
- """
- persisted_result = GetPersistedResult(test_name)
- if not persisted_result:
- return 1
- logging.info('*' * 80)
- logging.info('Output from:')
- logging.info(persisted_result['cmd'])
- logging.info('*' * 80)
- print persisted_result['output']
-
- if json_file_name:
- with file(json_file_name, 'w') as f:
- f.write(persisted_result['chartjson'])
-
- return persisted_result['exit_code']
-
-
-def PrintSummary(test_names):
- logging.info('*' * 80)
- logging.info('Sharding summary')
- device_total_time = collections.defaultdict(int)
- for test_name in test_names:
- file_name = os.path.join(constants.PERF_OUTPUT_DIR, test_name)
- if not os.path.exists(file_name):
- logging.info('%s : No status file found', test_name)
- continue
- with file(file_name, 'r') as f:
- result = pickle.loads(f.read())
- logging.info('%s : exit_code=%d in %d secs at %s',
- result['name'], result['exit_code'], result['total_time'],
- result['device'])
- device_total_time[result['device']] += result['total_time']
- for device, device_time in device_total_time.iteritems():
- logging.info('Total for device %s : %d secs', device, device_time)
- logging.info('Total steps time: %d secs', sum(device_total_time.values()))
-
-
-class _HeartBeatLogger(object):
- # How often to print the heartbeat on flush().
- _PRINT_INTERVAL = 30.0
-
- def __init__(self):
- """A file-like class for keeping the buildbot alive."""
- self._len = 0
- self._tick = time.time()
- self._stopped = threading.Event()
- self._timer = threading.Thread(target=self._runner)
- self._timer.start()
-
- def _runner(self):
- while not self._stopped.is_set():
- self.flush()
- self._stopped.wait(_HeartBeatLogger._PRINT_INTERVAL)
-
- def write(self, data):
- self._len += len(data)
-
- def flush(self):
- now = time.time()
- if now - self._tick >= _HeartBeatLogger._PRINT_INTERVAL:
- self._tick = now
- print '--single-step output length %d' % self._len
- sys.stdout.flush()
-
- def stop(self):
- self._stopped.set()
-
-
-class TestRunner(base_test_runner.BaseTestRunner):
- def __init__(self, test_options, device, shard_index, max_shard, tests,
- flaky_tests):
- """A TestRunner instance runs a perf test on a single device.
-
- Args:
- test_options: A PerfOptions object.
- device: Device to run the tests.
- shard_index: the index of this device.
- max_shards: the maximum shard index.
- tests: a dict mapping test_name to command.
- flaky_tests: a list of flaky test_name.
- """
- super(TestRunner, self).__init__(device, None)
- self._options = test_options
- self._shard_index = shard_index
- self._max_shard = max_shard
- self._tests = tests
- self._flaky_tests = flaky_tests
- self._output_dir = None
- self._device_battery = battery_utils.BatteryUtils(self.device)
-
- @staticmethod
- def _IsBetter(result):
- if result['actual_exit_code'] == 0:
- return True
- pickled = os.path.join(constants.PERF_OUTPUT_DIR,
- result['name'])
- if not os.path.exists(pickled):
- return True
- with file(pickled, 'r') as f:
- previous = pickle.loads(f.read())
- return result['actual_exit_code'] < previous['actual_exit_code']
-
- @staticmethod
- def _SaveResult(result):
- if TestRunner._IsBetter(result):
- with file(os.path.join(constants.PERF_OUTPUT_DIR,
- result['name']), 'w') as f:
- f.write(pickle.dumps(result))
-
- def _CheckDeviceAffinity(self, test_name):
- """Returns True if test_name has affinity for this shard."""
- affinity = (self._tests['steps'][test_name]['device_affinity'] %
- self._max_shard)
- if self._shard_index == affinity:
- return True
- logging.info('Skipping %s on %s (affinity is %s, device is %s)',
- test_name, self.device_serial, affinity, self._shard_index)
- return False
-
- def _CleanupOutputDirectory(self):
- if self._output_dir:
- shutil.rmtree(self._output_dir, ignore_errors=True)
- self._output_dir = None
-
- def _ReadChartjsonOutput(self):
- if not self._output_dir:
- return ''
-
- json_output_path = os.path.join(self._output_dir, 'results-chart.json')
- try:
- with open(json_output_path) as f:
- return f.read()
- except IOError:
- logging.exception('Exception when reading chartjson.')
- logging.error('This usually means that telemetry did not run, so it could'
- ' not generate the file. Please check the device running'
- ' the test.')
- return ''
-
- def _LaunchPerfTest(self, test_name):
- """Runs a perf test.
-
- Args:
- test_name: the name of the test to be executed.
-
- Returns:
- A tuple containing (Output, base_test_result.ResultType)
- """
- if not self._CheckDeviceAffinity(test_name):
- return '', base_test_result.ResultType.PASS
-
- try:
- logging.warning('Unmapping device ports')
- forwarder.Forwarder.UnmapAllDevicePorts(self.device)
- self.device.old_interface.RestartAdbdOnDevice()
- except Exception as e:
- logging.error('Exception when tearing down device %s', e)
-
- cmd = ('%s --device %s' %
- (self._tests['steps'][test_name]['cmd'],
- self.device_serial))
-
- if self._options.collect_chartjson_data:
- self._output_dir = tempfile.mkdtemp()
- cmd = cmd + ' --output-dir=%s' % self._output_dir
-
- logging.info(
- 'temperature: %s (0.1 C)',
- str(self._device_battery.GetBatteryInfo().get('temperature')))
- if self._options.max_battery_temp:
- self._device_battery.LetBatteryCoolToTemperature(
- self._options.max_battery_temp)
-
- logging.info('Charge level: %s%%',
- str(self._device_battery.GetBatteryInfo().get('level')))
- if self._options.min_battery_level:
- self._device_battery.ChargeDeviceToLevel(
- self._options.min_battery_level)
-
- logging.info('%s : %s', test_name, cmd)
- start_time = datetime.datetime.now()
-
- timeout = self._tests['steps'][test_name].get('timeout', 5400)
- if self._options.no_timeout:
- timeout = None
- logging.info('Timeout for %s test: %s', test_name, timeout)
- full_cmd = cmd
- if self._options.dry_run:
- full_cmd = 'echo %s' % cmd
-
- logfile = sys.stdout
- if self._options.single_step:
- # Just print a heart-beat so that the outer buildbot scripts won't timeout
- # without response.
- logfile = _HeartBeatLogger()
- cwd = os.path.abspath(constants.DIR_SOURCE_ROOT)
- if full_cmd.startswith('src/'):
- cwd = os.path.abspath(os.path.join(constants.DIR_SOURCE_ROOT, os.pardir))
- try:
- exit_code, output = cmd_helper.GetCmdStatusAndOutputWithTimeout(
- full_cmd, timeout, cwd=cwd, shell=True, logfile=logfile)
- json_output = self._ReadChartjsonOutput()
- except cmd_helper.TimeoutError as e:
- exit_code = -1
- output = str(e)
- json_output = ''
- finally:
- self._CleanupOutputDirectory()
- if self._options.single_step:
- logfile.stop()
- end_time = datetime.datetime.now()
- if exit_code is None:
- exit_code = -1
- logging.info('%s : exit_code=%d in %d secs at %s',
- test_name, exit_code, (end_time - start_time).seconds,
- self.device_serial)
-
- if exit_code == 0:
- result_type = base_test_result.ResultType.PASS
- else:
- result_type = base_test_result.ResultType.FAIL
- # Since perf tests use device affinity, give the device a chance to
- # recover if it is offline after a failure. Otherwise, the master sharder
- # will remove it from the pool and future tests on this device will fail.
- try:
- self.device.WaitUntilFullyBooted(timeout=120)
- except device_errors.CommandTimeoutError as e:
- logging.error('Device failed to return after %s: %s' % (test_name, e))
-
- actual_exit_code = exit_code
- if test_name in self._flaky_tests:
- # The exit_code is used at the second stage when printing the
- # test output. If the test is flaky, force to "0" to get that step green
- # whilst still gathering data to the perf dashboards.
- # The result_type is used by the test_dispatcher to retry the test.
- exit_code = 0
-
- persisted_result = {
- 'name': test_name,
- 'output': output,
- 'chartjson': json_output,
- 'exit_code': exit_code,
- 'actual_exit_code': actual_exit_code,
- 'result_type': result_type,
- 'total_time': (end_time - start_time).seconds,
- 'device': self.device_serial,
- 'cmd': cmd,
- }
- self._SaveResult(persisted_result)
-
- return (output, result_type)
-
- def RunTest(self, test_name):
- """Run a perf test on the device.
-
- Args:
- test_name: String to use for logging the test result.
-
- Returns:
- A tuple of (TestRunResults, retry).
- """
- _, result_type = self._LaunchPerfTest(test_name)
- results = base_test_result.TestRunResults()
- results.AddResult(base_test_result.BaseTestResult(test_name, result_type))
- retry = None
- if not results.DidRunPass():
- retry = test_name
- return results, retry
« no previous file with comments | « build/android/pylib/perf/test_options.py ('k') | build/android/pylib/perf/thermal_throttle.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698