diff options
Diffstat (limited to 'sources/pyside6/tests/QtCore/qdatastream_test.py')
| -rw-r--r-- | sources/pyside6/tests/QtCore/qdatastream_test.py | 61 |
1 files changed, 30 insertions, 31 deletions
diff --git a/sources/pyside6/tests/QtCore/qdatastream_test.py b/sources/pyside6/tests/QtCore/qdatastream_test.py index cd0e3e0cb..517f466aa 100644 --- a/sources/pyside6/tests/QtCore/qdatastream_test.py +++ b/sources/pyside6/tests/QtCore/qdatastream_test.py @@ -1,32 +1,6 @@ #!/usr/bin/python - -############################################################################# -## -## Copyright (C) 2016 The Qt Company Ltd. -## Contact: https://www.qt.io/licensing/ -## -## This file is part of the test suite of Qt for Python. -## -## $QT_BEGIN_LICENSE:GPL-EXCEPT$ -## Commercial License Usage -## Licensees holding valid commercial Qt licenses may use this file in -## accordance with the commercial license agreement provided with the -## Software or, alternatively, in accordance with the terms contained in -## a written agreement between you and The Qt Company. For licensing terms -## and conditions see https://www.qt.io/terms-conditions. For further -## information use the contact form at https://www.qt.io/contact-us. -## -## GNU General Public License Usage -## Alternatively, this file may be used under the terms of the GNU -## General Public License version 3 as published by the Free Software -## Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT -## included in the packaging of this file. Please review the following -## information to ensure the GNU General Public License requirements will -## be met: https://www.gnu.org/licenses/gpl-3.0.html. -## -## $QT_END_LICENSE$ -## -############################################################################# +# Copyright (C) 2022 The Qt Company Ltd. +# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 '''Unit tests for QDataStream''' @@ -35,8 +9,10 @@ import sys 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 QBitArray, QByteArray, QIODevice, QDataStream, QDate, QTime, QDateTime @@ -275,7 +251,6 @@ class QDataStreamShift(unittest.TestCase): class QDataStreamShiftBitArray(unittest.TestCase): - def _check_bitarray(self, data_set): '''Check the >> operator for the given data set''' @@ -308,7 +283,7 @@ class QDataStreamShiftBitArray(unittest.TestCase): self._check_bitarray([(serialized, QDataStream.ReadPastEnd, QBitArray())]) -class QDataStreamRawData(unittest.TestCase): +class QDataStreamBuffer(unittest.TestCase): def testRawData(self): data = QDataStream() self.assertEqual(data.readRawData(4), None) @@ -321,7 +296,31 @@ class QDataStreamRawData(unittest.TestCase): data = QDataStream(ba) self.assertEqual(data.readRawData(4), bytes('AB\x00C', "UTF-8")) + def testRawDataBytes(self): + test_data = b'AB\0' + data = QDataStream() + ba = QByteArray() + data = QDataStream(ba, QIODevice.WriteOnly) + data.writeRawData(test_data) + self.assertEqual(ba.data(), test_data) + data = QDataStream(ba) + self.assertEqual(data.readRawData(3), test_data) + + def testBytes(self): + dataOne = QDataStream() + self.assertEqual(dataOne.readBytes(4), None) + + ba = QByteArray() + data = QDataStream(ba, QIODevice.WriteOnly) + # writeBytes() writes a quint32 containing the length of the data, + # followed by the data. + data.writeBytes(bytes('AB\x00C', 'UTF-8')) + self.assertEqual(ba.data(), bytes('\x00\x00\x00\x04AB\x00C', 'UTF-8')) + + data = QDataStream(ba) + buffer = data.readBytes(4) + self.assertEqual(buffer, bytes('AB\x00C', 'UTF-8')) + if __name__ == '__main__': unittest.main() - |