1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-25 17:09:44 +00:00
trezor-firmware/tests/show_results.py

72 lines
2.2 KiB
Python
Executable File

#!/usr/bin/env python3
import http.server
import multiprocessing
import os
import posixpath
import time
import urllib
import webbrowser
from pathlib import Path
import click
ROOT = Path(__file__).parent.parent.resolve()
TEST_RESULT_PATH = ROOT / "tests" / "ui_tests" / "reporting" / "reports" / "test"
class NoCacheRequestHandler(http.server.SimpleHTTPRequestHandler):
def end_headers(self):
self.send_header("Cache-Control", "no-cache, no-store, must-revalidate")
self.send_header("Pragma", "no-cache")
self.send_header("Expires", "0")
return super().end_headers()
def log_message(self, format, *args):
pass
def translate_path(self, path):
# XXX
# Copy-pasted from Python 3.8 BaseHTTPRequestHandler so that we can inject
# the `directory` parameter.
# Otherwise, to keep compatible with 3.6, we'd need to mess with CWD. Which is
# unstable when we expect it to be erased and recreated under us.
path = path.split("?", 1)[0]
path = path.split("#", 1)[0]
# Don't forget explicit trailing slash when normalizing. Issue17324
trailing_slash = path.rstrip().endswith("/")
try:
path = urllib.parse.unquote(path, errors="surrogatepass")
except UnicodeDecodeError:
path = urllib.parse.unquote(path)
path = posixpath.normpath(path)
words = path.split("/")
words = filter(None, words)
path = str(TEST_RESULT_PATH) # XXX this is the only modified line
for word in words:
if os.path.dirname(word) or word in (os.curdir, os.pardir):
# Ignore components that are not a simple file/directory name
continue
path = os.path.join(path, word)
if trailing_slash:
path += "/"
return path
def launch_http_server(port):
http.server.test(HandlerClass=NoCacheRequestHandler, bind="localhost", port=port)
@click.command()
@click.option("-p", "--port", type=int, default=8000)
def main(port):
httpd = multiprocessing.Process(target=launch_http_server, args=(port,))
httpd.start()
time.sleep(0.5)
webbrowser.open(f"http://localhost:{port}/")
httpd.join()
if __name__ == "__main__":
main()