1 files changed, 154 insertions, 0 deletions
diff --git a/sources/pyside6/tests/QtCore/blocking_signals_test.py b/sources/pyside6/tests/QtCore/blocking_signals_test.py
new file mode 100644
index 000000000..493abb071
--- /dev/null
+++ b/sources/pyside6/tests/QtCore/blocking_signals_test.py
@@ -0,0 +1,154 @@
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+''' Test case for QObject.signalsBlocked() and blockSignal()'''
+
+import gc
+import os
+import sys
+from tempfile import mkstemp
+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 QObject, Signal, QFile, QSignalBlocker
+
+
+class Sender(QObject):
+    mysignal = Signal()
+    mysignal_int_int = Signal(int, int)
+
+
+class TestSignalsBlockedBasic(unittest.TestCase):
+    '''Basic test case for signalsBlocked'''
+
+    def testBasic(self):
+        '''QObject.signalsBlocked() and blockSignals()
+        The signals aren't blocked by default.
+        blockSignals returns the previous value'''
+        obj = QObject()
+        self.assertTrue(not obj.signalsBlocked())
+        self.assertTrue(not obj.blockSignals(True))
+        self.assertTrue(obj.signalsBlocked())
+        self.assertTrue(obj.blockSignals(False))
+        blocker = QSignalBlocker(obj)
+        self.assertTrue(obj.signalsBlocked())
+        blocker.unblock()
+        self.assertTrue(not obj.signalsBlocked())
+        blocker.reblock()
+        self.assertTrue(obj.signalsBlocked())
+        del blocker
+        # PYSIDE-535: Need to collect garbage in PyPy to trigger deletion
+        gc.collect()
+        self.assertTrue(not obj.signalsBlocked())
+
+    def testContext(self):
+        obj = QObject()
+        self.assertTrue(not obj.signalsBlocked())
+        with QSignalBlocker(obj):
+            self.assertTrue(obj.signalsBlocked())
+        self.assertTrue(not obj.signalsBlocked())
+
+    def testContextWithAs(self):
+        obj = QObject()
+        self.assertTrue(not obj.signalsBlocked())
+        with QSignalBlocker(obj) as blocker:
+            self.assertTrue(obj.signalsBlocked())
+            self.assertEqual(type(blocker), QSignalBlocker)
+        self.assertTrue(not obj.signalsBlocked())
+
+
+class TestSignalsBlocked(unittest.TestCase):
+    '''Test case to check if the signals are really blocked'''
+
+    def setUp(self):
+        # Set up the basic resources needed
+        self.obj = Sender()
+        self.args = tuple()
+        self.called = False
+
+    def tearDown(self):
+        # Delete used resources
+        del self.obj
+        del self.args
+        # PYSIDE-535: Need to collect garbage in PyPy to trigger deletion
+        gc.collect()
+
+    def callback(self, *args):
+        # Default callback
+        if args == self.args:
+            self.called = True
+        else:
+            raise TypeError("Invalid arguments")
+
+    def testShortCircuitSignals(self):
+        # Blocking of Python short-circuit signals
+        self.obj.mysignal.connect(self.callback)
+
+        self.obj.mysignal.emit()
+        self.assertTrue(self.called)
+
+        self.called = False
+        self.obj.blockSignals(True)
+        self.obj.mysignal.emit()
+        self.assertTrue(not self.called)
+
+    def testPythonSignals(self):
+        # Blocking of Python typed signals
+
+        self.obj.mysignal_int_int.connect(self.callback)
+        self.args = (1, 3)
+
+        self.obj.mysignal_int_int.emit(*self.args)
+        self.assertTrue(self.called)
+
+        self.called = False
+        self.obj.blockSignals(True)
+        self.obj.mysignal_int_int.emit(*self.args)
+        self.assertTrue(not self.called)
+
+
+class TestQFileSignalBlocking(unittest.TestCase):
+    '''Test case for blocking the signal QIODevice.aboutToClose()'''
+
+    def setUp(self):
+        # Set up the needed resources - A temp file and a QFile
+        self.called = False
+        handle, self.filename = mkstemp()
+        os.close(handle)
+
+        self.qfile = QFile(self.filename)
+
+    def tearDown(self):
+        # Release acquired resources
+        os.remove(self.filename)
+        del self.qfile
+        # PYSIDE-535: Need to collect garbage in PyPy to trigger deletion
+        gc.collect()
+
+    def callback(self):
+        # Default callback
+        self.called = True
+
+    def testAboutToCloseBlocking(self):
+        # QIODevice.aboutToClose() blocking
+
+        self.qfile.aboutToClose.connect(self.callback)
+
+        self.assertTrue(self.qfile.open(QFile.ReadOnly))
+        self.qfile.close()
+        self.assertTrue(self.called)
+
+        self.called = False
+        self.qfile.blockSignals(True)
+
+        self.assertTrue(self.qfile.open(QFile.ReadOnly))
+        self.qfile.close()
+        self.assertTrue(not self.called)
+
+
+if __name__ == '__main__':
+    unittest.main()