1 files changed, 73 insertions, 0 deletions
diff --git a/tests/QtCore/qbytearray_operator_iadd_test.py b/tests/QtCore/qbytearray_operator_iadd_test.py
new file mode 100644
index 000000000..168f22de9
--- /dev/null
+++ b/tests/QtCore/qbytearray_operator_iadd_test.py
@@ -0,0 +1,73 @@
+
+import unittest
+import new
+
+from PySide.QtCore import QByteArray, QString
+from helper.docmodifier import DocModifier
+
+class BaseQByteArrayOperatorIAdd(object):
+    '''Base class for QByteArray += operator tests.
+    
+    Implementing classes should inherit from unittest.TestCase and implement
+    setUp, setting self.obj and self.orig_obj to the target QByteArray and original
+    one, respectively'''
+
+    __metaclass__ = DocModifier
+
+    def testSingleString(self):
+        '''QByteArray += string of size 1'''
+        s = '0'
+        self.obj += s
+        self.assertEqual(self.obj, self.orig_obj + s)
+        self.assertEqual(self.obj.size(), self.orig_obj.size() + len(s))
+
+    def testString(self):
+        '''QByteArray += string of size > 1'''
+        s = 'dummy'
+        self.obj += s
+        self.assertEqual(self.obj, self.orig_obj + s)
+        self.assertEqual(self.obj.size(), self.orig_obj.size() + len(s))
+
+    def testQByteArray(self):
+        '''QByteArray += QByteArray'''
+        s = QByteArray('array')
+        self.obj += s
+        self.assertEqual(self.obj, self.orig_obj + s)
+
+    def testChar(self):
+        '''QByteArray += char (number < 256)'''
+        s = ord('a')
+        self.obj += s
+        self.assertEqual(self.obj, self.orig_obj + s)
+        self.assertEqual(self.obj.size(), self.orig_obj.size() + 1)
+
+    def testQString(self):
+        '''QByteArray += QString'''
+        s = QString('dummy')
+        self.obj += s
+        self.assertEqual(self.obj, self.orig_obj + s)
+        self.assertEqual(self.obj.size(), self.orig_obj.size() + s.size())
+
+class NullQByteArrayOperatorIAdd(unittest.TestCase, BaseQByteArrayOperatorIAdd):
+    '''Test case for operator QByteArray += on null QByteArrays'''
+
+    doc_prefix = 'Null object'
+    doc_filter = lambda x: x.startswith('test')
+
+    def setUp(self):
+        self.obj = QByteArray()
+        self.orig_obj = QByteArray()
+
+
+class ValidQByteArrayOperatorIAdd(unittest.TestCase, BaseQByteArrayOperatorIAdd):
+    '''Test case for operator QByteArray += on valid QByteArrays'''
+
+    doc_prefix = 'Valid object'
+    doc_filter = lambda x: x.startswith('test')
+
+    def setUp(self):
+        self.obj = QByteArray('some byte array')
+        self.orig_obj = QByteArray('some byte array')
+
+if __name__ == '__main__':
+    unittest.main()