1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#!/usr/bin/env python
# Copyright (C) 2022 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
import os
import sys
import unittest
from pathlib import Path
sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
from shiboken_paths import init_paths
init_paths()
from sample import ByteArray
class ByteArrayBufferProtocolTest(unittest.TestCase):
'''Tests ByteArray implementation of Python buffer protocol.'''
def testByteArrayBufferProtocol(self):
# Tests ByteArray implementation of Python buffer protocol using the Path().is_dir
# function with a unicode object or other object implementing the Python buffer protocol.
Path(str(ByteArray('/tmp'))).is_dir()
class ByteArrayConcatenationOperatorTest(unittest.TestCase):
'''Test cases for ByteArray concatenation with '+' operator.'''
def testConcatByteArrayAndPythonString(self):
# Test concatenation of a ByteArray with a Python string, in this order.
ba = ByteArray('foo')
result = ba + '\x00bar'
self.assertEqual(type(result), ByteArray)
self.assertEqual(result, 'foo\x00bar')
def testConcatPythonStringAndByteArray(self):
# Test concatenation of a Python string with a ByteArray, in this order.
concat_python_string_add_qbytearray_worked = True # noqa: F841
ba = ByteArray('foo')
result = 'bar\x00' + ba
self.assertEqual(type(result), ByteArray)
self.assertEqual(result, 'bar\x00foo')
class ByteArrayOperatorEqual(unittest.TestCase):
'''TestCase for operator ByteArray == ByteArray.'''
def testDefault(self):
# ByteArray() == ByteArray()
obj1 = ByteArray()
obj2 = ByteArray()
self.assertEqual(obj1, obj2)
def testSimple(self):
# ByteArray(some_string) == ByteArray(some_string)
string = 'egg snakes'
self.assertEqual(ByteArray(string), ByteArray(string))
def testPyString(self):
# ByteArray(string) == string
string = 'my test string'
self.assertEqual(ByteArray(string), string)
def testQString(self):
# ByteArray(string) == string
string = 'another test string'
self.assertEqual(ByteArray(string), string)
class ByteArrayOperatorAt(unittest.TestCase):
'''TestCase for operator ByteArray[]'''
def testInRange(self):
# ByteArray[x] where x is a valid index.
string = 'abcdefgh'
obj = ByteArray(string)
for i in range(len(string)):
self.assertEqual(obj[i], bytes(string[i], "UTF8"))
def testInRangeReverse(self):
# ByteArray[x] where x is a valid index (reverse order).
string = 'abcdefgh'
obj = ByteArray(string)
for i in range(len(string) - 1, 0, -1):
self.assertEqual(obj[i], bytes(string[i], "UTF8"))
def testOutOfRange(self):
# ByteArray[x] where x is out of index.
string = '1234567'
obj = ByteArray(string)
self.assertRaises(IndexError, lambda: obj[len(string)])
def testNullStrings(self):
ba = ByteArray('\x00')
self.assertEqual(ba.at(0), '\x00')
self.assertEqual(ba[0], bytes('\x00', "UTF8"))
class ByteArrayOperatorLen(unittest.TestCase):
'''Test case for __len__ operator of ByteArray'''
def testBasic(self):
'''ByteArray __len__'''
self.assertEqual(len(ByteArray()), 0)
self.assertEqual(len(ByteArray('')), 0)
self.assertEqual(len(ByteArray(' ')), 1)
self.assertEqual(len(ByteArray('yabadaba')), 8)
class ByteArrayAndPythonStr(unittest.TestCase):
'''Test case for __str__ operator of ByteArray'''
def testStrOperator(self):
'''ByteArray __str__'''
self.assertEqual(ByteArray().__str__(), '')
self.assertEqual(ByteArray('').__str__(), '')
self.assertEqual(ByteArray('aaa').__str__(), 'aaa')
def testPythonStrAndNull(self):
s1 = bytes('123\000321', "UTF8")
ba = ByteArray(s1)
s2 = ba.data()
self.assertEqual(s1, s2)
self.assertEqual(type(s1), type(s2))
self.assertEqual(s1, ba)
self.assertNotEqual(type(s1), type(ba))
if __name__ == '__main__':
unittest.main()
|