]> git.ipfire.org Git - thirdparty/starlette.git/commitdiff
Use `Self` in `TestClient.__enter__` (#2951)
authorMarcelo Trylesinski <marcelotryle@gmail.com>
Sat, 21 Jun 2025 01:54:43 +0000 (09:54 +0800)
committerGitHub <noreply@github.com>
Sat, 21 Jun 2025 01:54:43 +0000 (09:54 +0800)
starlette/testclient.py

index df8e113807bf155ef54fc559bb83f7222e4e5fff..575b1a71f22bea0976f2fa22b62e041e934ec8ba 100644 (file)
@@ -35,6 +35,11 @@ if sys.version_info >= (3, 10):  # pragma: no cover
 else:  # pragma: no cover
     from typing_extensions import TypeGuard
 
+if sys.version_info >= (3, 11):  # pragma: no cover
+    from typing import Self
+else:  # pragma: no cover
+    from typing_extensions import Self
+
 try:
     import httpx
 except ModuleNotFoundError:  # pragma: no cover
@@ -667,7 +672,7 @@ class TestClient(httpx.Client):
 
         return session
 
-    def __enter__(self) -> TestClient:
+    def __enter__(self) -> Self:
         with contextlib.ExitStack() as stack:
             self.portal = portal = stack.enter_context(anyio.from_thread.start_blocking_portal(**self.async_backend))