commit 4447d87695e1aec1eef8f9f475cf1e8db15b749b
author Joshua Haberman <jhaberman@gmail.com> Sun Jan 09 13:13:59 2022 -0800
committer Joshua Haberman <jhaberman@gmail.com> Sun Jan 09 13:13:59 2022 -0800
tree 9daf2ae3c5605d5e71174af8a09adcf275729126
parent a2e280e86535a9c46184eccdb9e6295ad156ee69

Fixed tests that were missing enum type wrapper class.

python/message.c

diff --git a/python/message.c b/python/message.c
index 46be7bb..371348c 100644
--- a/python/message.c
+++ b/python/message.c
@@ -1636,7 +1636,6 @@
   if (PyModule_AddObject(m, "MessageMeta", message_meta_type)) return false;
 
   PyObject* mod = PyImport_ImportModule("google.protobuf.message");
-
   if (mod == NULL) return false;
 
   state->encode_error_class = PyObject_GetAttrString(mod, "EncodeError");
@@ -1644,8 +1643,16 @@
   state->message_class = PyObject_GetAttrString(mod, "Message");
   Py_DECREF(mod);
 
+  PyObject* enum_type_wrapper =
+      PyImport_ImportModule("google.protobuf.internal.enum_type_wrapper");
+  if (enum_type_wrapper == NULL) return false;
+
+  state->enum_type_wrapper_class =
+      PyObject_GetAttrString(enum_type_wrapper, "EnumTypeWrapper");
+  Py_DECREF(enum_type_wrapper);
+
   if (!state->encode_error_class || !state->decode_error_class ||
-      !state->message_class) {
+      !state->message_class || !state->enum_type_wrapper_class) {
     return false;
   }