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
|
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Various helper functions related to working with websockets."""
import logging
import time
from gpu_tests import common_typing as ct
from gpu_tests.util import websocket_server as wss
def HandleWebsocketReceiveTimeoutError(tab: ct.Tab, start_time: float) -> None:
"""Helper function for when a message is not received on time."""
logging.error(
'Timed out waiting for websocket message (%.3f seconds since test '
'start), checking for hung renderer',
time.time() - start_time)
# Telemetry has some code to automatically crash the renderer and GPU
# processes if it thinks that the renderer is hung. So, execute some
# trivial JavaScript now to hit that code if we got the timeout because of
# a hung renderer. If we do detect a hung renderer, this will raise
# another exception and prevent the following line about the renderer not
# being hung from running.
tab.action_runner.EvaluateJavaScript('let somevar = undefined;', timeout=5)
logging.error('Timeout does *not* appear to be due to a hung renderer')
def HandlePrematureSocketClose(original_error: wss.ClientClosedConnectionError,
start_time: float) -> None:
elapsed = time.time() - start_time
raise RuntimeError(
f'Detected closed websocket ({elapsed:.3f} seconds since test start) - '
f'likely caused by a renderer crash') from original_error
def GetScaledConnectionTimeout(num_jobs: int) -> float:
"""Gets a scaled Websocket setup timeout based on number of test jobs.
Args:
num_jobs: How many parallel test jobs are being run.
Returns:
The scaled timeout to pass to WaitForConnection that accounts for slowness
caused by multiple test jobs.
"""
# Target a 2x multiplier when running 4 jobs.
multiplier = 1 + (num_jobs - 1) / 3.0
return multiplier * wss.WEBSOCKET_SETUP_TIMEOUT_SECONDS
|
