Skip to content

Commit 5f497ee

Browse files
Fix type of data parameter in get test requests
Lists of two-tuples of strings are allowed Co-authored-by: Marti Raudsepp <[email protected]>
1 parent fc30395 commit 5f497ee

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

rest_framework-stubs/test.pyi

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1+
from collections.abc import Iterable, Mapping
12
from typing import Any
3+
from typing_extensions import TypeAlias
24

35
import coreapi
46
import requests
@@ -14,6 +16,10 @@ from rest_framework.authtoken.models import Token
1416
from rest_framework.request import Request
1517
from rest_framework.response import _MonkeyPatchedResponse
1618

19+
_GetDataType: TypeAlias = (
20+
Mapping[str, str | bytes | Iterable[str | bytes]] | Iterable[tuple[str, str | bytes | Iterable[str | bytes]]] | None
21+
)
22+
1723
def force_authenticate(
1824
request: HttpRequest, user: AnonymousUser | AbstractBaseUser | None = ..., token: Token | None = ...
1925
) -> None: ...
@@ -50,7 +56,7 @@ class APIRequestFactory(DjangoRequestFactory):
5056
renderer_classes: Any
5157
def __init__(self, enforce_csrf_checks: bool = ..., **defaults: Any) -> None: ...
5258
def request(self, **kwargs: Any) -> Request: ... # type: ignore[override]
53-
def get(self, path: str, data: dict[str, Any] | str | None = ..., follow: bool = ..., **extra: Any): ... # type: ignore[override]
59+
def get(self, path: str, data: _GetDataType = ..., follow: bool = ..., **extra: Any): ... # type: ignore[override]
5460
def post(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> Request: ... # type: ignore[override]
5561
def put(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> Request: ... # type: ignore[override]
5662
def patch(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> Request: ... # type: ignore[override]
@@ -68,7 +74,7 @@ class APIClient(APIRequestFactory, DjangoClient):
6874
def credentials(self, **kwargs: Any): ...
6975
def force_authenticate(self, user: AnonymousUser | AbstractBaseUser = ..., token: Token | None = ...) -> None: ...
7076
def request(self, **kwargs: Any) -> _MonkeyPatchedResponse: ... # type: ignore[override]
71-
def get(self, path: str, data: dict[str, Any] | str | None = ..., follow: bool = ..., **extra: Any): ... # type: ignore[override]
77+
def get(self, path: str, data: _GetDataType = ..., follow: bool = ..., **extra: Any): ... # type: ignore[override]
7278
def post(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> _MonkeyPatchedResponse: ... # type: ignore[override]
7379
def put(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> _MonkeyPatchedResponse: ... # type: ignore[override]
7480
def patch(self, path: str, data: Any | None = ..., format: str | None = ..., content_type: str | None = ..., follow: bool = ..., **extra: Any) -> _MonkeyPatchedResponse: ... # type: ignore[override]

0 commit comments

Comments
 (0)