| #!/usr/bin/env python |
| |
| import json |
| import logging |
| import sys |
| |
| from webkitcorepy import Timeout |
| |
| from webkitpy.benchmark_runner.benchmark_runner import BenchmarkRunner |
| from webkitpy.benchmark_runner.http_server_driver.http_server_driver_factory import HTTPServerDriverFactory |
| |
| 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): |
| 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) |
