1 files changed, 95 insertions, 0 deletions
diff --git a/tests/QtCore/qtextstream_test.py b/tests/QtCore/qtextstream_test.py
new file mode 100644
index 000000000..13dc6516c
--- /dev/null
+++ b/tests/QtCore/qtextstream_test.py
@@ -0,0 +1,95 @@
+
+'''Unit tests for QTextStream'''
+
+import unittest
+
+from PySide.QtCore import QTextStream, QIODevice, QString, QByteArray
+from PySide.QtCore import QTextCodec, QFile
+
+class QTextStreamShiftTest(unittest.TestCase):
+
+    def setUp(self):
+        self.ba = QByteArray()
+        self.read = QTextStream(self.ba, QIODevice.ReadOnly)
+        self.write = QTextStream(self.ba, QIODevice.WriteOnly)
+
+    def testNumber(self):
+        '''QTextStream << number'''
+
+        self.write << QString('4')
+        self.write.flush()
+        res = self.read.readLine()
+        self.assert_(isinstance(res, QString))
+        self.assertEqual(res, QString('4'))
+
+    def testString(self):
+        self.write << QString('Test_it!')
+        self.write.flush()
+        res = QString()
+        self.read >> res
+        self.assertEqual(res, QString('Test_it!'))
+
+class QTextStreamGetSet(unittest.TestCase):
+
+    def setUp(self):
+        self.obj = QTextStream()
+
+
+    def testCodec(self):
+        '''QTextStream set/get Codec'''
+
+        codec = QTextCodec.codecForName('ISO8859-1')
+        self.obj.setCodec(codec)
+        self.assertEqual(codec, self.obj.codec())
+
+    def testDevice(self):
+        '''QTextStream get/set Device'''
+        device = QFile()
+        self.obj.setDevice(device)
+        self.assertEqual(device, self.obj.device())
+        self.obj.setDevice(None)
+        self.assertEqual(None, self.obj.device())
+
+class QTextStreamInitialization(unittest.TestCase):
+
+    def testConstruction(self):
+        '''QTextStream construction'''
+        obj = QTextStream()
+
+        self.assertEqual(obj.codec(), QTextCodec.codecForLocale())
+        self.assertEqual(obj.device(), None)
+        self.assertEqual(obj.string(), None)
+
+        self.assert_(obj.atEnd())
+        self.assertEqual(obj.readAll(), '')
+
+class QTextStreamReadLinesFromDevice(unittest.TestCase):
+
+    def _check_data(self, data_set):
+
+        for data, lines in data_set:
+            stream = QTextStream(data)
+
+            res = []
+            while not stream.atEnd():
+                res.append(stream.readLine())
+
+            self.assertEqual(res, lines)
+
+    def testLatin1(self):
+        '''QTextStream readLine for simple Latin1 strings'''
+
+        data = []
+
+        data.append((QByteArray(), []))
+        data.append((QByteArray('\n'), ['']))
+        data.append((QByteArray('\r\n'), ['']))
+        data.append((QByteArray('ole'), ['ole']))
+        data.append((QByteArray('ole\n'), ['ole']))
+        data.append((QByteArray('ole\r\n'), ['ole']))
+        data.append((QByteArray('ole\r\ndole\r\ndoffen'), ['ole', 'dole', 'doffen']))
+
+        self._check_data(data)
+
+if __name__ == '__main__':
+    unittest.main()