blob: af99d0a83bcede1b96ce6664f62e29f5478f958a [file] [log] [blame]
#!/usr/bin/env python
import json
import logging
import sys
from webkitpy.benchmark_runner.benchmark_runner import BenchmarkRunner
from webkitpy.benchmark_runner.http_server_driver.http_server_driver_factory import HTTPServerDriverFactory
from webkitpy.common.timeout_context import Timeout
if sys.version_info > (3, 0):
from urllib.parse import urljoin
else:
from urlparse import urljoin
_log = logging.getLogger(__name__)
class WebServerBenchmarkRunner(BenchmarkRunner):
name = 'webserver'
def __init__(self, plan_file, local_copy, count_override, build_dir, output_file, platform, browser, browser_path, scale_unit=True, show_iteration_values=False, device_id=None, diagnose_dir=None):
self._http_server_driver = HTTPServerDriverFactory.create(platform)
self._http_server_driver.set_device_id(device_id)
super(WebServerBenchmarkRunner, self).__init__(plan_file, local_copy, count_override, build_dir, output_file, platform, browser, browser_path, scale_unit, show_iteration_values, device_id, diagnose_dir)
def _get_result(self, test_url):
result = self._browser_driver.add_additional_results(test_url, self._http_server_driver.fetch_result())
assert(not self._http_server_driver.get_return_code())
return result
def _run_one_test(self, web_root, test_file):
result = None
try:
self._http_server_driver.serve(web_root)
url = urljoin(self._http_server_driver.base_url(), self._plan_name + '/' + test_file)
self._browser_driver.launch_url(url, self._plan['options'], self._build_dir, self._browser_path)
with Timeout(self._plan['timeout']):
result = self._get_result(url)
except Exception as error:
self._browser_driver.diagnose_test_failure(self._diagnose_dir, error)
raise error
finally:
self._browser_driver.close_browsers()
self._http_server_driver.kill_server()
return json.loads(result)