backportUnittest: backport unittest.TestCase.enterContext

product/ERP5Type/tests/backportUnittest.py

@@ -12,6 +12,7 @@ def patch():
     import six
 
     import contextlib
+    import sys
     import traceback
     from unittest import TestCase, TextTestResult, TextTestRunner
 
@@ -27,6 +28,16 @@ def patch():
         yield
       TestCase.subTest = subTest
 
+    if sys.version_info < (3, 11):
+        def enterContext(self, cm):
+            cls = type(cm)
+            enter = cls.__enter__
+            exit = cls.__exit__
+            result = enter(cm)
+            self.addCleanup(exit, cm, None, None, None)
+            return result
+        TestCase.enterContext = enterContext
+
     TextTestResult_addError = six.get_unbound_function(TextTestResult.addError)
     def addError(self, test, err):
         if isinstance(err[1], SetupSiteError):