1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
|
#!/usr/bin/env python3
# Copyright (C) 2021 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
import re
import os
import sys
import time
import atexit
import threading
import subprocess
import http.server
from pathlib import Path
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.chrome.service import Service
import argparse
class WasmTestRunner:
def __init__(self, args: dict):
self.server_process = None
self.browser_process = None
self.python_path = Path(sys.executable)
self.script_dir = Path(os.path.dirname(os.path.realpath(__file__)))
self.host = 'localhost'
self.webserver = None
self.webthread = None
paths = ['html_path', 'browser_path', 'chromedriver_path', 'tmp_dir']
for key, value in args.items():
if value is None:
continue
if key in paths:
value = Path(value)
value.resolve()
setattr(self, key, value)
if not self.html_path.exists():
raise FileNotFoundError(self.html_path)
self.webroot = self.html_path.parent
if hasattr(self, 'browser_path') and not self.browser_path.exists():
raise FileNotFoundError(self.browser_path)
atexit.register(self.cleanup)
def run(self):
# self.run_webserver()
self.run_threaded_webserver()
if self.use_browser:
self.run_wasm_browser()
else:
self.run_wasm_webdriver()
self.shutdown_threaded_webserver()
def run_webserver(self):
webroot = self.html_path.parent.resolve()
self.server_process =\
subprocess.Popen([
str(self.python_path),
'-m', 'http.server',
'--directory', str(webroot),
self.port
])
def run_threaded_webserver(self):
self.webserver = http.server.ThreadingHTTPServer(
(self.host, int(self.port)), self.get_http_handler_class())
self.webthread = threading.Thread(target=self.webserver.serve_forever)
self.webthread.start()
def shutdown_threaded_webserver(self):
if self.webserver is not None:
self.webserver.shutdown()
if self.webthread is not None:
self.webthread.join()
def run_wasm_webdriver(self):
url = f'http://localhost:{self.port}/{self.html_path.name}'
d = DesiredCapabilities.CHROME
d['goog:loggingPrefs'] = {'browser': 'ALL'}
ser = Service(executable_path=self.chromedriver_path)
driver = webdriver.Chrome(desired_capabilities=d, service=ser)
driver.get(url)
timeout = 15
test_done = False
while not test_done and timeout != 0:
# HACK : we don't know for sure how long each test takes
# so just sleep a bit here until we get desired result
# The test may never produce 'Finished testing' (eg. crash),
# so we need a timeout as well
time.sleep(1)
timeout = timeout - 1
for entry in driver.get_log('browser'):
regex = re.compile(r'[^"]*"(.*)".*')
match = regex.match(entry['message'])
if match is not None:
console_line = match.group(1)
print(console_line)
if 'Finished testing' in console_line:
test_done = True
def run_wasm_browser(self):
if not hasattr(self, 'browser_path'):
print('Error: browser path must be set to run with browser')
return
if not hasattr(self, 'tmp_dir'):
print('Error: tmp_dir must be set to run with browser')
return
self.create_tmp_dir()
self.browser_process =\
subprocess.Popen([
str(self.browser_path),
'--user-data-dir=' + str(self.tmp_dir),
'--enable-logging=stderr',
f'http://localhost:{self.port}/{self.html_path.name}'
],
stderr=subprocess.PIPE
)
# Only capture the console content
regex = re.compile(r'[^"]*CONSOLE[^"]*"(.*)"[.\w]*')
for line in self.browser_process.stderr:
str_line = line.decode('utf-8')
match = regex.match(str_line)
# Error condition, this should have matched
if 'CONSOLE' in str_line and match is None:
print('Error: did not match console line:')
print(str_line)
if match is not None:
console_line = match.group(1)
print(console_line)
if 'Finished testing' in str_line:
self.browser_process.kill()
break
def create_tmp_dir(self):
if not self.tmp_dir.exists():
self.tmp_dir.mkdir()
if not self.tmp_dir.is_dir():
raise NotADirectoryError(self.tmp_dir)
# Needed to bypass the "Welcome to Chrome" prompt
first_run = Path(self.tmp_dir, 'First Run')
first_run.touch()
def get_http_handler_class(self):
wtr = self
class OriginIsolationHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
def __init__(self, request, client_address, server):
super().__init__(request, client_address, server, directory=wtr.webroot)
# Headers required to enable SharedArrayBuffer
# See https://web.dev/cross-origin-isolation-guide/
def end_headers(self):
self.send_header("Cross-Origin-Opener-Policy", "same-origin")
self.send_header("Cross-Origin-Embedder-Policy", "require-corp")
self.send_header("Cross-Origin-Resource-Policy", "cross-origin")
http.server.SimpleHTTPRequestHandler.end_headers(self)
# We usually don't care that much about what the webserver is logging
def log_message(self, format_, *args):
return
return OriginIsolationHTTPRequestHandler
def cleanup(self):
if self.browser_process is not None:
self.browser_process.kill()
if self.server_process is not None:
self.server_process.kill()
self.shutdown_threaded_webserver()
def main():
parser = argparse.ArgumentParser(description='WASM testrunner')
parser.add_argument('html_path', help='Path to the HTML file to request')
parser.add_argument('--port', help='Port to run the webserver on', default='8000')
parser.add_argument('--use_browser', action='store_true')
parser.add_argument('--browser_path', help='Path to the browser to use')
parser.add_argument('--chromedriver_path', help='Absolute path to chromedriver',
default='chromedriver')
parser.add_argument('--tmp_dir', help='Path to the tmpdir to use when using a browser',
default='/tmp/wasm-testrunner')
args = vars(parser.parse_args())
test_runner = WasmTestRunner(args)
test_runner.run()
if __name__ == '__main__':
main()
|