blob: 6425a4b86990bceee11d76f224af3d8757300e0e (
plain)
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
|
# Copyright (C) 2022 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
import os
import sys
import unittest
from pathlib import Path
sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
from init_paths import init_test_paths
init_test_paths(False)
from PySide6.QtCore import (QMutex, QMutexLocker, QReadLocker, QReadWriteLock,
QThread, QWriteLocker)
class MyWriteThread(QThread):
def __init__(self, lock):
super().__init__()
self.lock = lock
self.started = False
self.canQuit = False
def run(self):
self.started = True
while not self.lock.tryLockForWrite():
pass
self.canQuit = True
self.lock.unlock()
class MyReadThread(QThread):
def __init__(self, lock):
super().__init__()
self.lock = lock
self.started = False
self.canQuit = False
def run(self):
self.started = True
while not self.lock.tryLockForRead():
pass
self.canQuit = True
self.lock.unlock()
class MyMutexedThread(QThread):
def __init__(self, mutex):
super().__init__()
self.mutex = mutex
self.started = False
self.canQuit = False
def run(self):
self.started = True
while not self.mutex.tryLock():
pass
self.mutex.unlock()
self.canQuit = True
class TestQMutex (unittest.TestCase):
def testReadLocker(self):
lock = QReadWriteLock()
thread = MyWriteThread(lock)
with QReadLocker(lock):
thread.start()
while not thread.started:
QThread.msleep(10)
self.assertFalse(thread.canQuit)
self.assertTrue(thread.wait(2000))
self.assertTrue(thread.canQuit)
def testWriteLocker(self):
lock = QReadWriteLock()
thread = MyReadThread(lock)
with QWriteLocker(lock):
thread.start()
while not thread.started:
QThread.msleep(10)
self.assertFalse(thread.canQuit)
self.assertTrue(thread.wait(2000))
self.assertTrue(thread.canQuit)
def testMutexLocker(self):
mutex = QMutex()
thread = MyMutexedThread(mutex)
with QMutexLocker(mutex):
thread.start()
while not thread.started:
QThread.msleep(10)
self.assertFalse(thread.canQuit)
self.assertTrue(thread.wait(2000))
self.assertTrue(thread.canQuit)
def testWithAsLocker(self):
lock = QReadWriteLock()
with QReadLocker(lock) as locker:
self.assertTrue(isinstance(locker, QReadLocker))
with QWriteLocker(lock) as locker:
self.assertTrue(isinstance(locker, QWriteLocker))
mutex = QMutex()
with QMutexLocker(mutex) as locker:
self.assertTrue(isinstance(locker, QMutexLocker))
with self.assertRaises(TypeError):
with QMutexLocker(lock) as locker:
pass
if __name__ == '__main__':
unittest.main()
|