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
|
import functools
import time
import pytest
from pytestqt.qt_compat import qt_api
pytest_plugins = "pytester"
@pytest.fixture
def stop_watch():
"""
Fixture that makes it easier for tests to ensure signals emitted and
timeouts are being respected.
"""
class StopWatch:
def __init__(self):
self._start_time = None
self.elapsed = None
def start(self):
self._start_time = time.monotonic()
def stop(self):
self.elapsed = (time.monotonic() - self._start_time) * 1000.0
def check(self, timeout, *delays):
"""
Make sure either timeout (if given) or at most of the given
delays used to trigger a signal has passed.
"""
self.stop()
if timeout is None:
timeout = max(delays) * 1.35 # 35% tolerance
max_wait_ms = max(delays + (timeout,))
assert self.elapsed < max_wait_ms
return StopWatch()
@pytest.fixture
def timer():
"""
Returns a Timer-like object which can be used to trigger signals and callbacks
after some time.
It is recommended to use this instead of ``QTimer.singleShot`` uses a static timer which may
trigger after a test finishes, possibly causing havoc.
"""
class Timer(qt_api.QtCore.QObject):
def __init__(self):
qt_api.QtCore.QObject.__init__(self)
self.timers_and_slots = []
def shutdown(self):
while self.timers_and_slots:
t, slot = self.timers_and_slots.pop(-1)
t.stop()
t.timeout.disconnect(slot)
def single_shot(self, signal, delay):
t = qt_api.QtCore.QTimer(self)
t.setSingleShot(True)
slot = functools.partial(self._emit, signal)
t.timeout.connect(slot)
t.start(delay)
self.timers_and_slots.append((t, slot))
def single_shot_callback(self, callback, delay):
t = qt_api.QtCore.QTimer(self)
t.setSingleShot(True)
t.timeout.connect(callback)
t.start(delay)
self.timers_and_slots.append((t, callback))
def _emit(self, signal):
signal.emit()
timer = Timer()
yield timer
timer.shutdown()
|
