Skip to content

Commit c480449

Browse files
committed
Add full typing for _responses and record
1 parent 9005f98 commit c480449

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

src/trio/_tests/test_socket.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,18 @@
3333
Union[tuple[str, int], tuple[str, int, int, int]],
3434
]
3535
GetAddrInfoResponse: TypeAlias = list[GaiTuple]
36+
GetAddrInfoArgs: TypeAlias = tuple[
37+
str | bytes | None,
38+
str | bytes | int | None,
39+
int,
40+
int,
41+
int,
42+
int,
43+
]
3644
else:
3745
GaiTuple: object
3846
GetAddrInfoResponse = object
47+
GetAddrInfoArgs = object
3948

4049
################################################################
4150
# utils
@@ -54,10 +63,10 @@ def __init__(
5463
) -> None:
5564
self._orig_getaddrinfo = orig_getaddrinfo
5665
self._responses: dict[
57-
tuple[str | int | bytes | None, ...],
66+
GetAddrInfoArgs,
5867
GetAddrInfoResponse | str,
5968
] = {}
60-
self.record: list[tuple[str | int | bytes | None, ...]] = []
69+
self.record: list[GetAddrInfoArgs] = []
6170

6271
# get a normalized getaddrinfo argument tuple
6372
def _frozenbind(

0 commit comments

Comments
 (0)