From: Hai Shi <shihai1992@gmail.com>
Date: Tue, 17 Aug 2021 14:50:33 +0000 (+0800)
Subject: bpo-42035: Enhance test_get_type_name() of _testcapi (GH-27649)
X-Git-Tag: v3.11.0a1~396
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=fcd651d16fc5ac3d07dd3f57f1001a861a2e7d23;p=thirdparty%2FPython%2Fcpython.git

bpo-42035: Enhance test_get_type_name() of _testcapi (GH-27649)
---

diff --git a/Modules/_testcapimodule.c b/Modules/_testcapimodule.c
index 0a3c6e0b9b27..e5f1ad312b41 100644
--- a/Modules/_testcapimodule.c
+++ b/Modules/_testcapimodule.c
@@ -1154,6 +1154,20 @@ test_get_type_name(PyObject *self, PyObject *Py_UNUSED(ignored))
     assert(strcmp(PyUnicode_AsUTF8(tp_name), "HeapTypeNameType") == 0);
     Py_DECREF(tp_name);
 
+    PyObject *name = PyUnicode_FromString("test_name");
+    if (name == NULL) {
+        goto done;
+    }
+    if (PyObject_SetAttrString(HeapTypeNameType, "__name__", name) < 0) {
+        Py_DECREF(name);
+        goto done;
+    }
+    tp_name = PyType_GetName((PyTypeObject *)HeapTypeNameType);
+    assert(strcmp(PyUnicode_AsUTF8(tp_name), "test_name") == 0);
+    Py_DECREF(name);
+    Py_DECREF(tp_name);
+
+  done:
     Py_DECREF(HeapTypeNameType);
     Py_RETURN_NONE;
 }