add tests for _PyErr_SetLocaleString

Author: picnixz

Lib/test/test_capi/test_exceptions.py

@@ -256,6 +256,23 @@ def test_set_string(self):
         # CRASHES setstring(ZeroDivisionError, NULL)
         # CRASHES setstring(NULL, b'error')
 
+    def test_set_locale_string(self):
+        # test _PyErr_SetLocaleString()
+        setlocalestring = _testcapi.err_setlocalestring
+        with self.assertRaises(ZeroDivisionError) as e:
+            setlocalestring(ZeroDivisionError, b'error')
+        self.assertEqual(e.exception.args, ('error',))
+        with self.assertRaises(ZeroDivisionError) as e:
+            setlocalestring(ZeroDivisionError, 'помилка'.encode())
+        self.assertEqual(e.exception.args, ('помилка',))
+
+        with self.assertRaises(ZeroDivisionError):
+            setlocalestring(ZeroDivisionError, b'\xff')
+        self.assertRaises(SystemError, setlocalestring, list, b'error')
+
+        # CRASHES setlocalestring(ZeroDivisionError, NULL)
+        # CRASHES setlocalestring(NULL, b'error')
+
     def test_format(self):
         """Test PyErr_Format()"""
         import_helper.import_module('ctypes')

Modules/_testcapi/clinic/exceptions.c.h

@@ -3,6 +3,9 @@
 
 #include "parts.h"
 #include "util.h"
+
+#include "pycore_pyerrors.h"
+
 #include "clinic/exceptions.c.h"
 
 
@@ -154,6 +157,25 @@ _testcapi_err_setstring_impl(PyObject *module, PyObject *exc,
     return NULL;
 }
 
+/*[clinic input]
+_testcapi.err_setlocalestring
+    exc: object
+    value: str(zeroes=True, accept={robuffer, str, NoneType})
+    /
+[clinic start generated code]*/
+
+static PyObject *
+_testcapi_err_setlocalestring_impl(PyObject *module, PyObject *exc,
+                                   const char *value,
+                                   Py_ssize_t value_length)
+/*[clinic end generated code: output=5c02a10d4eb573d7 input=628c26bdc2497a92]*/
+{
+    NULLABLE(exc)
+    _PyErr_SetLocaleString(exc, value);
+    return NULL;
+}
+
+
 /*[clinic input]
 _testcapi.err_setfromerrnowithfilename
     error: int
@@ -396,6 +418,7 @@ static PyMethodDef test_methods[] = {
     _TESTCAPI_EXC_SET_OBJECT_METHODDEF
     _TESTCAPI_EXC_SET_OBJECT_FETCH_METHODDEF
     _TESTCAPI_ERR_SETSTRING_METHODDEF
+    _TESTCAPI_ERR_SETLOCALESTRING_METHODDEF
     _TESTCAPI_ERR_SETFROMERRNOWITHFILENAME_METHODDEF
     _TESTCAPI_RAISE_EXCEPTION_METHODDEF
     _TESTCAPI_RAISE_MEMORYERROR_METHODDEF