Fix bug 652 - "Segfault when using QTextBlock::setUserData due to missing ownership transfer"

2 files changed, 31 insertions, 0 deletions

diff --git a/tests/QtGui/CMakeLists.txt b/tests/QtGui/CMakeLists.txt
index 1cc3696f5..9ed86d683 100644
--- a/tests/QtGui/CMakeLists.txt
+++ b/tests/QtGui/CMakeLists.txt
@@ -32,6 +32,7 @@ PYSIDE_TEST(bug_585.py)
 PYSIDE_TEST(bug_589.py)
 PYSIDE_TEST(bug_617.py)
 PYSIDE_TEST(bug_640.py)
+PYSIDE_TEST(bug_652.py)
 PYSIDE_TEST(customproxywidget_test.py)
 PYSIDE_TEST(deepcopy_test.py)
 PYSIDE_TEST(float_to_int_implicit_conversion_test.py)
diff --git a/tests/QtGui/bug_652.py b/tests/QtGui/bug_652.py
new file mode 100644
index 000000000..438fdbde4
--- /dev/null
+++ b/tests/QtGui/bug_652.py
@@ -0,0 +1,30 @@
+import unittest
+from PySide.QtGui import *
+
+class MyData(QTextBlockUserData):
+    def __init__(self, data):
+        QTextBlockUserData.__init__(self)
+        self.data = data
+
+    def getMyNiceData(self):
+        return self.data
+
+class TestBug652(unittest.TestCase):
+    """Segfault when using QTextBlock::setUserData due to missing ownership transfer"""
+    def testIt(self):
+        td = QTextDocument()
+        tc = QTextCursor(td)
+        tc.insertText("Hello world")
+        heyHo = "hey ho!"
+        tc.block().setUserData(MyData(heyHo))
+        self.assertEqual(type(tc.block().userData()), MyData)
+        self.assertEqual(tc.block().userData().getMyNiceData(), heyHo)
+
+        del tc
+        tc = QTextCursor(td)
+        blk = tc.block()
+        self.assertEqual(type(blk.userData()), MyData)
+        self.assertEqual(blk.userData().getMyNiceData(), heyHo)
+
+if __name__ == "__main__":
+    unittest.main()
\ No newline at end of file