_struct

Signed-off-by: Matthew A Johnson <matjoh@microsoft.com>

Author: matajoh

Lib/test/test_freeze/test_datetime.py

@@ -1,4 +1,3 @@
-
 import unittest
 from datetime import datetime, timedelta
 

Lib/test/test_freeze/test_struct.py

@@ -0,0 +1,13 @@
+import unittest
+from struct import Struct
+
+from .test_common import BaseObjectTest
+
+
+class TestStruct(BaseObjectTest):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, obj=Struct("i"), **kwargs)
+
+
+if __name__ == "__main__":
+    unittest.main()

Modules/_elementtree.c

@@ -4386,11 +4386,10 @@ module_exec(PyObject *m)
     if(register_freezable != NULL)
     {
         PyObject* result = PyObject_CallOneArg(register_freezable, (PyObject *)st->Element_Type);
+        Py_DECREF(register_freezable);
         if(result == NULL){
             goto error;
         }
-
-        Py_DECREF(register_freezable);
     }
 
     st->deepcopy_obj = _PyImport_GetModuleAttrString("copy", "deepcopy");

Modules/_struct.c

@@ -2503,6 +2503,9 @@ _structmodule_exec(PyObject *m)
     if (PyModule_AddType(m, (PyTypeObject *)state->PyStructType) < 0) {
         return -1;
     }
+    if (_PyImmutability_RegisterFreezable((PyTypeObject *)state->PyStructType) < 0){
+        return -1;
+    }
 
     state->unpackiter_type = PyType_FromModuleAndSpec(
         m, &unpackiter_type_spec, NULL);