mirror of
https://github.com/python/cpython.git
synced 2025-08-04 00:48:58 +00:00
gh-98783: Fix crashes when str
subclasses are used in _PyUnicode_Equal
(#98806)
This commit is contained in:
parent
3ac8c0ab6e
commit
76f989dc3e
5 changed files with 36 additions and 3 deletions
|
@ -1317,6 +1317,15 @@ order (MRO) for bases """
|
|||
with self.assertRaisesRegex(AttributeError, "'X' object has no attribute 'a'"):
|
||||
X().a
|
||||
|
||||
# Test string subclass in `__slots__`, see gh-98783
|
||||
class SubStr(str):
|
||||
pass
|
||||
class X(object):
|
||||
__slots__ = (SubStr('x'),)
|
||||
X().x = 1
|
||||
with self.assertRaisesRegex(AttributeError, "'X' object has no attribute 'a'"):
|
||||
X().a
|
||||
|
||||
def test_slots_special(self):
|
||||
# Testing __dict__ and __weakref__ in __slots__...
|
||||
class D(object):
|
||||
|
@ -3589,6 +3598,16 @@ order (MRO) for bases """
|
|||
self.assertEqual(o.__str__(), '41')
|
||||
self.assertEqual(o.__repr__(), 'A repr')
|
||||
|
||||
def test_repr_with_module_str_subclass(self):
|
||||
# gh-98783
|
||||
class StrSub(str):
|
||||
pass
|
||||
class Some:
|
||||
pass
|
||||
Some.__module__ = StrSub('example')
|
||||
self.assertIsInstance(repr(Some), str) # should not crash
|
||||
self.assertIsInstance(repr(Some()), str) # should not crash
|
||||
|
||||
def test_keyword_arguments(self):
|
||||
# Testing keyword arguments to __init__, __call__...
|
||||
def f(a): return a
|
||||
|
|
|
@ -1334,6 +1334,12 @@ class LongTest(unittest.TestCase):
|
|||
b'\xff\xff\xff\xff\xff')
|
||||
self.assertRaises(OverflowError, (1).to_bytes, 0, 'big')
|
||||
|
||||
# gh-98783
|
||||
class SubStr(str):
|
||||
pass
|
||||
self.assertEqual((0).to_bytes(1, SubStr('big')), b'\x00')
|
||||
self.assertEqual((0).to_bytes(0, SubStr('little')), b'')
|
||||
|
||||
def test_from_bytes(self):
|
||||
def check(tests, byteorder, signed=False):
|
||||
def equivalent_python(byte_array, byteorder, signed=False):
|
||||
|
@ -1534,6 +1540,12 @@ class LongTest(unittest.TestCase):
|
|||
self.assertRaises(TypeError, int.from_bytes, MissingBytes())
|
||||
self.assertRaises(ZeroDivisionError, int.from_bytes, RaisingBytes())
|
||||
|
||||
# gh-98783
|
||||
class SubStr(str):
|
||||
pass
|
||||
self.assertEqual(int.from_bytes(b'', SubStr('big')), 0)
|
||||
self.assertEqual(int.from_bytes(b'\x00', SubStr('little')), 0)
|
||||
|
||||
@support.cpython_only
|
||||
def test_from_bytes_small(self):
|
||||
# bpo-46361
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue