65 lines
2.2 KiB
Python
Executable File
65 lines
2.2 KiB
Python
Executable File
#! /usr/bin/env python
|
|
import time
|
|
from threading import Event
|
|
|
|
from launch_time import LaunchTimeBenchmark, DefaultLaunchTimeHandler
|
|
|
|
|
|
class StartupBenchmark(LaunchTimeBenchmark):
|
|
def initialize(self):
|
|
self.benchmark_description = 'Measure startup time of a given browser.'
|
|
self.response_handler = StartupBenchmark.ResponseHandler(self)
|
|
self.start_time = None
|
|
self.stop_time = None
|
|
self.stop_signal_was_received = Event()
|
|
|
|
def run_iteration(self):
|
|
self.start_time = time.time() * 1000
|
|
self.open_tab()
|
|
while self.stop_time is None:
|
|
self.stop_signal_was_received.wait()
|
|
result = self.stop_time - self.start_time
|
|
self.stop_time = None
|
|
self.stop_signal_was_received.clear()
|
|
self.quit_browser()
|
|
return result
|
|
|
|
def get_test_name(self):
|
|
return "StartupBenchmark"
|
|
|
|
@staticmethod
|
|
def ResponseHandler(startup_benchmark):
|
|
class Handler(DefaultLaunchTimeHandler):
|
|
def get_test_page(self):
|
|
return '''<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>Startup Benchmark</title>
|
|
<meta http-equiv="Content-Type" content="text/html" />
|
|
<script>
|
|
function sendDone() {
|
|
const time = Date.now();
|
|
const request = new XMLHttpRequest();
|
|
request.open("POST", "done", false);
|
|
request.setRequestHeader('Content-Type', 'application/json');
|
|
request.send(JSON.stringify(time));
|
|
}
|
|
window.onload = sendDone;
|
|
</script>
|
|
</head>
|
|
<body>
|
|
<h1>Startup Benchmark</h1>
|
|
</body>
|
|
</html>
|
|
'''
|
|
|
|
def on_receive_stop_signal(self, data):
|
|
startup_benchmark.stop_time = float(data)
|
|
startup_benchmark.stop_signal_was_received.set()
|
|
|
|
return Handler
|
|
|
|
|
|
if __name__ == '__main__':
|
|
StartupBenchmark().run()
|