shaka-packager/tools/perf/benchmarks/spaceport.py

67 lines
2.4 KiB
Python

# Copyright (c) 2012 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 spaceport.io's PerfMarks benchmark."""
import logging
import os
from telemetry import test
from telemetry.core import util
from telemetry.page import page_measurement
from telemetry.page import page_set
class SpaceportMeasurement(page_measurement.PageMeasurement):
def CustomizeBrowserOptions(self, options):
options.extra_browser_args.extend(['--disable-gpu-vsync'])
def MeasurePage(self, _, tab, results):
util.WaitFor(lambda: tab.EvaluateJavaScript(
'!document.getElementById("start-performance-tests").disabled'), 60)
tab.ExecuteJavaScript("""
window.__results = {};
window.console.log = function(str) {
if (!str) return;
var key_val = str.split(': ');
if (!key_val.length == 2) return;
__results[key_val[0]] = key_val[1];
};
document.getElementById('start-performance-tests').click();
""")
js_get_results = 'JSON.stringify(window.__results)'
num_tests_complete = [0] # A list to work around closure issue.
def _IsDone():
num_tests_in_measurement = 24
num_results = len(eval(tab.EvaluateJavaScript(js_get_results)))
if num_results > num_tests_complete[0]:
num_tests_complete[0] = num_results
logging.info('Completed measurement %d of %d'
% (num_tests_complete[0],
num_tests_in_measurement))
return num_tests_complete[0] >= num_tests_in_measurement
util.WaitFor(_IsDone, 1200, poll_interval=5)
result_dict = eval(tab.EvaluateJavaScript(js_get_results))
for key in result_dict:
chart, trace = key.split('.', 1)
results.Add(trace, 'objects (bigger is better)', float(result_dict[key]),
chart_name=chart, data_type='unimportant')
results.Add('Score', 'objects (bigger is better)',
[float(x) for x in result_dict.values()])
class Spaceport(test.Test):
"""spaceport.io's PerfMarks benchmark."""
test = SpaceportMeasurement
def CreatePageSet(self, options):
spaceport_dir = os.path.join(util.GetChromiumSrcDir(), 'chrome', 'test',
'data', 'third_party', 'spaceport')
return page_set.PageSet.FromDict(
{'pages': [{'url': 'file:///index.html'}]},
spaceport_dir)