# Copyright (c) 2011 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.
"""Playback driver."""
import cgi
import simplejson as json
import os
import string
import sys
import threading
import urlparse
START_PAGE = """<html>
<script type="text/javascript">
var runCount = $run_count;
var results = [];
function run() {
var wnd = window.open('?resource=start_page_popup', '',
'width=$width, height=$height');
var timerId = setInterval(function() {
wnd.postMessage('ping', '$target_origin');
}, 300);
var handleMessage = function(event) {
clearInterval(timerId);
wnd.close();
document.writeln('<div>' + event.data + '</div>');
results.push(event.data);
runCount -= 1;
window.removeEventListener('message', handleMessage);
if (runCount > 0) {
run();
} else {
var xmlHttpRequest = new XMLHttpRequest();
xmlHttpRequest.open("POST", '/benchmark/', true);
xmlHttpRequest.setRequestHeader("Content-type", "application/json");
xmlHttpRequest.send(JSON.stringify({results: results}));
}
}
window.addEventListener('message', handleMessage, false);
}
run();
</script>
</html>
"""
START_PAGE_POPUP = """<html>
<script type="text/javascript">
window.setTimeout(function() {
console.log(window.innerWidth, window.innerHeight);
if (window.innerWidth == $width && window.innerHeight == $height) {
window.location = '$start_url';
} else {
window.resizeBy($width - window.innerWidth, $height - window.innerHeight);
window.location = window.location;
}
}, 200);
</script>
</html>
"""
DATA_JS = 'Benchmark.data = $data;'
def ReadFile(file_name, mode='r'):
f = open(file_name, mode)
data = f.read()
f.close()
return data
def ReadJSON(file_name):
f = open(file_name, 'r')
data = json.load(f)
f.close()
return data
class PlaybackRequestHandler(object):
"""This class is used to process HTTP requests during test playback.
Attributes:
test_dir: directory containing test files.
test_callback: function to be called when the test is finished.
script_dir: directory where javascript files are located.
"""
def __init__(self, test_dir, test_callback=None, script_dir=os.getcwd()):
self.test_dir = test_dir
self.test_callback = test_callback
self.script_dir = script_dir
def ProcessRequest(self, handler):
"Processes single HTTP request."
parse_result = urlparse.urlparse(handler.path)
if parse_result.path.endswith('/benchmark/'):
query = cgi.parse_qs(parse_result.query)
if 'run_test' in query:
run_count = 1
if 'run_count' in query:
run_count = query['run_count'][0]
self._StartTest(handler, self.test_dir, run_count)
elif 'resource' in query:
self._GetBenchmarkResource(query['resource'][0], handler)
else:
self._ProcessBenchmarkReport(handler.body, handler)
else:
self._GetApplicationResource(handler)
def _StartTest(self, handler, test_dir, run_count):
"Sends test start page to browser."
cache_data = ReadJSON(os.path.join(test_dir, 'cache.json'))
# Load cached responses.
self.cache = {}
responses_dir = os.path.join(test_dir, 'responses')
for request in cache_data['requests']:
response_file = os.path.join(responses_dir, request['response_file'])
response = ReadFile(response_file, 'rb')
key = (request['method'], request['path'])
self.cache[key] = {'response': response, 'headers': request['headers']}
# Load benchmark scripts.
self.benchmark_resources = {}
data = ReadFile(os.path.join(test_dir, 'data.json'))
data = string.Template(DATA_JS).substitute(data=data)
self.benchmark_resources['data.js'] = {'data': data,
'type': 'application/javascript'}
for resource in ('common.js', 'playback.js'):
resource_file = os.path.join(self.script_dir, resource)
self.benchmark_resources[resource] = {'data': ReadFile(resource_file),
'type': 'application/javascript'}
# Format start page.
parse_result = urlparse.urlparse(cache_data['start_url'])
target_origin = '%s://%s' % (parse_result.scheme, parse_result.netloc)
start_page = string.Template(START_PAGE).substitute(
run_count=run_count, target_origin=target_origin,
width=cache_data['width'], height=cache_data['height'])
self.benchmark_resources['start_page'] = {
'data': start_page,
'type': 'text/html; charset=UTF-8'
}
start_page_popup = string.Template(START_PAGE_POPUP).substitute(
start_url=cache_data['start_url'],
width=cache_data['width'], height=cache_data['height'])
self.benchmark_resources['start_page_popup'] = {
'data': start_page_popup,
'type': 'text/html; charset=UTF-8'
}
self._GetBenchmarkResource('start_page', handler)
def _GetBenchmarkResource(self, resource, handler):
"Sends requested resource to browser."
if resource in self.benchmark_resources:
resource = self.benchmark_resources[resource]
handler.send_response(200)
handler.send_header('content-length', len(resource['data']))
handler.send_header('content-type', resource['type'])
handler.end_headers()
handler.wfile.write(resource['data'])
else:
handler.send_response(404)
handler.end_headers()
def _ProcessBenchmarkReport(self, content, handler):
"Reads benchmark score from report content and invokes callback."
handler.send_response(204)
handler.end_headers()
content = json.loads(content)
if 'results' in content:
results = content['results']
sys.stdout.write('Results: %s\n' % results)
if self.test_callback: self.test_callback(results)
elif 'error' in content:
sys.stderr.write('Error: %s\n' % content['error'])
def _GetApplicationResource(self, handler):
"Searches for response in cache. If not found, responds with 204."
key = (handler.command, handler.path)
if key in self.cache:
sys.stdout.write('%s %s -> found\n' % key)
handler.wfile.write(self.cache[key]['response'])
else:
sys.stderr.write('%s %s -> not found\n' % key)
handler.send_response(204, "not in cache")
handler.end_headers()