|  | # 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. | 
|  |  | 
|  | import collections | 
|  | import json | 
|  |  | 
|  | from telemetry import multi_page_benchmark | 
|  | from telemetry import util | 
|  |  | 
|  |  | 
|  | class SunSpiderBenchark(multi_page_benchmark.MultiPageBenchmark): | 
|  | def MeasurePage(self, _, tab, results): | 
|  | js_is_done = """ | 
|  | window.location.pathname.indexOf('sunspider-results') >= 0""" | 
|  | def _IsDone(): | 
|  | return tab.runtime.Evaluate(js_is_done) | 
|  | util.WaitFor(_IsDone, 300, poll_interval=5) | 
|  |  | 
|  | js_get_results = 'JSON.stringify(output);' | 
|  | js_results = json.loads(tab.runtime.Evaluate(js_get_results)) | 
|  | r = collections.defaultdict(list) | 
|  | totals = [] | 
|  | # js_results is: [{'foo': v1, 'bar': v2}, | 
|  | #                 {'foo': v3, 'bar': v4}, | 
|  | #                 ...] | 
|  | for result in js_results: | 
|  | total = 0 | 
|  | for key, value in result.iteritems(): | 
|  | r[key].append(value) | 
|  | total += value | 
|  | totals.append(total) | 
|  | for key, values in r.iteritems(): | 
|  | results.Add('t', 'ms', values, chart_name=key, data_type='unimportant') | 
|  | results.Add('t', 'ms', totals, chart_name='total') |