check if testCreatePrototypeOverride exist

PiperOrigin-RevId: 501229501
2 years ago · 9b5a084118
parent d86f531b54
commit 9b5a084118
1 changed files with 2 additions and 1 deletions
--- a/python/pb_unit_tests/message_factory_test_wrapper.py
+++ b/python/pb_unit_tests/message_factory_test_wrapper.py
@ -26,7 +26,8 @@
 from google.protobuf.internal.message_factory_test import *
 import unittest

-MessageFactoryTest.testCreatePrototypeOverride.__unittest_expecting_failure__ = True
+if hasattr(MessageFactoryTest, 'testCreatePrototypeOverride'):
+  MessageFactoryTest.testCreatePrototypeOverride.__unittest_expecting_failure__ = True
 MessageFactoryTest.testDuplicateExtensionNumber.__unittest_expecting_failure__ = True
 MessageFactoryTest.testGetMessages.__unittest_expecting_failure__ = True
 MessageFactoryTest.testGetPrototype.__unittest_expecting_failure__ = True