diff --git a/Lib/test/test_builtin.py b/Lib/test/test_builtin.py index c3f04ec7447c6e..e0d2a698dcffea 100644 --- a/Lib/test/test_builtin.py +++ b/Lib/test/test_builtin.py @@ -1511,6 +1511,11 @@ def test_bytearray_translate(self): self.assertRaises(ValueError, x.translate, b"1", 1) self.assertRaises(TypeError, x.translate, b"1"*256, 1) + def test_bytearray_extend_error(self): + array = bytearray() + bad_iter = map(int, "X") + self.assertRaises(ValueError, array.extend, bad_iter) + def test_construct_singletons(self): for const in None, Ellipsis, NotImplemented: tp = type(const) diff --git a/Misc/NEWS.d/next/Core and Builtins/2019-06-26-18-41-00.bpo-37417.VsZeHL.rst b/Misc/NEWS.d/next/Core and Builtins/2019-06-26-18-41-00.bpo-37417.VsZeHL.rst new file mode 100644 index 00000000000000..f004631e236103 --- /dev/null +++ b/Misc/NEWS.d/next/Core and Builtins/2019-06-26-18-41-00.bpo-37417.VsZeHL.rst @@ -0,0 +1,2 @@ +:meth:`bytearray.extend` now correctly handles errors that arise during iteration. +Patch by Brandt Bucher. \ No newline at end of file diff --git a/Objects/bytearrayobject.c b/Objects/bytearrayobject.c index 8a0994ffa4ab09..41de28d58b1a74 100644 --- a/Objects/bytearrayobject.c +++ b/Objects/bytearrayobject.c @@ -1689,6 +1689,10 @@ bytearray_extend(PyByteArrayObject *self, PyObject *iterable_of_ints) } Py_DECREF(bytearray_obj); + if (PyErr_Occurred()) { + return NULL; + } + Py_RETURN_NONE; }