1 files changed, 49 insertions, 0 deletions
diff --git a/tests/QtGui/qpainter_test.py b/tests/QtGui/qpainter_test.py
new file mode 100644
index 000000000..ad1d1e471
--- /dev/null
+++ b/tests/QtGui/qpainter_test.py
@@ -0,0 +1,49 @@
+import unittest
+
+from PySide.QtGui import QPainter, QBrush, QLinearGradient
+from PySide.QtCore import QRect, QRectF, Qt
+
+class QPainterDrawText(unittest.TestCase):
+
+    def setUp(self):
+        self.painter = QPainter()
+        self.text = 'teste!'
+
+    def tearDown(self):
+        del self.text
+        del self.painter
+
+    def testDrawText(self):
+        # bug #254
+        rect = self.painter.drawText(100, 100, 100, 100,
+                                    Qt.AlignCenter | Qt.TextWordWrap,
+                                    self.text)
+        self.assert_(isinstance(rect, QRect))
+
+    def testDrawTextWithRect(self):
+        # bug #225
+        rect = QRect(100, 100, 100, 100)
+        newRect = self.painter.drawText(rect, Qt.AlignCenter | Qt.TextWordWrap,
+                              self.text)
+
+        self.assert_(isinstance(newRect, QRect))
+
+    def testDrawTextWithRectF(self):
+        '''QPainter.drawText(QRectF, ... ,QRectF*) inject code'''
+        rect = QRectF(100, 52.3, 100, 100)
+        newRect = self.painter.drawText(rect, Qt.AlignCenter | Qt.TextWordWrap,
+                              self.text)
+
+        self.assert_(isinstance(newRect, QRectF))
+
+class SetBrushWithOtherArgs(unittest.TestCase):
+    '''Using qpainter.setBrush with args other than QBrush'''
+
+    def testSetBrushGradient(self):
+        painter = QPainter()
+        gradient = QLinearGradient(0, 0, 0, 0)
+        painter.setBrush(gradient)
+
+if __name__ == '__main__':
+    unittest.main()
+