Backport enterContext()

tests/compat.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+import sys
+from contextlib import AbstractContextManager
+from typing import Any
+from typing import Callable
+from typing import TypeVar
+
+from django import test
+
+_T = TypeVar("_T")
+
+if sys.version_info < (3, 11):
+
+    def _enter_context(cm: Any, addcleanup: Callable[..., None]) -> Any:
+        # We look up the special methods on the type to match the with
+        # statement.
+        cls = type(cm)
+        try:
+            enter = cls.__enter__
+            exit = cls.__exit__
+        except AttributeError:
+            raise TypeError(
+                f"'{cls.__module__}.{cls.__qualname__}' object does "
+                f"not support the context manager protocol"
+            ) from None
+        result = enter(cm)
+        addcleanup(exit, cm, None, None, None)
+        return result
+
+
+class SimpleTestCase(test.SimpleTestCase):
+    if sys.version_info < (3, 11):
+
+        def enterContext(self, cm: AbstractContextManager[_T]) -> _T:
+            result: _T = _enter_context(cm, self.addCleanup)
+            return result

tests/test_django_watchfiles.py

@@ -3,11 +3,11 @@
 import tempfile
 from pathlib import Path
 
-from django.test import SimpleTestCase
 from django.utils import autoreload
 
 from django_watchfiles import MutableWatcher
 from django_watchfiles import WatchfilesReloader
+from tests.compat import SimpleTestCase
 
 
 class MutableWatcherTests(SimpleTestCase):