Skip to content

Commit 724837c

Browse files
committed
Fix type annotations in conftest
1 parent 5274011 commit 724837c

File tree

1 file changed

+19
-17
lines changed

1 file changed

+19
-17
lines changed

tests/conftest.py

Lines changed: 19 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
1-
import typing
1+
from typing import Any, Dict, Generator, List
22

33
import attr
44
import pytest
55
import responses
66

77
from labels.github import Label
88

9-
ResponseLabel = typing.Dict[str, typing.Any]
10-
ResponseLabels = typing.List[ResponseLabel]
9+
ResponseLabel = Dict[str, Any]
10+
ResponseLabels = List[ResponseLabel]
1111

1212

1313
@pytest.fixture(name="username", scope="session")
@@ -49,7 +49,7 @@ class FakeProc:
4949

5050

5151
@pytest.fixture(name="mock_repo_info")
52-
def fixture_mock_repo_info(mocker: typing.Any, remote_url: str) -> typing.Any:
52+
def fixture_mock_repo_info(mocker: Any, remote_url: str) -> Any:
5353
"""Patch the subprocess call to git remote get-url."""
5454

5555
return mocker.patch(
@@ -60,7 +60,7 @@ def fixture_mock_repo_info(mocker: typing.Any, remote_url: str) -> typing.Any:
6060

6161

6262
@pytest.fixture(name="mock_repo_info_error")
63-
def fixture_mock_repo_info_error(mocker: typing.Any) -> typing.Any:
63+
def fixture_mock_repo_info_error(mocker: Any) -> Any:
6464
"""Patch the subprocess call to git remote get-url with an error."""
6565

6666
return mocker.patch(
@@ -71,7 +71,7 @@ def fixture_mock_repo_info_error(mocker: typing.Any) -> typing.Any:
7171

7272

7373
@pytest.fixture(name="mock_repo_info_bad_url")
74-
def fixture_mock_repo_info_bad_url(mocker: typing.Any) -> typing.Any:
74+
def fixture_mock_repo_info_bad_url(mocker: Any) -> Any:
7575
"""Patch the subprocess call to git remote get-url with a bad URL."""
7676

7777
return mocker.patch(
@@ -154,7 +154,7 @@ def fixture_response_list_labels(
154154
@pytest.fixture(name="mock_list_labels")
155155
def fixture_mock_list_labels(
156156
base_url: str, repo_owner: str, repo_name: str, response_list_labels: ResponseLabels
157-
) -> None:
157+
) -> Generator:
158158
"""Mock requests for list labels."""
159159
with responses.RequestsMock() as rsps:
160160
rsps.add(
@@ -176,7 +176,7 @@ def fixture_mock_list_labels_paginated(
176176
response_get_infra: ResponseLabel,
177177
response_get_docs: ResponseLabel,
178178
response_get_bug: ResponseLabel,
179-
):
179+
) -> Generator:
180180
"""Mock requests for list labels with pagination."""
181181

182182
with responses.RequestsMock() as rsps:
@@ -215,7 +215,7 @@ def fixture_mock_list_labels_paginated(
215215
@pytest.fixture(name="mock_get_label")
216216
def fixture_mock_get_label(
217217
base_url: str, repo_owner: str, repo_name: str, response_get_bug: ResponseLabel
218-
) -> None:
218+
) -> Generator:
219219
"""Mock requests for get label."""
220220
with responses.RequestsMock() as rsps:
221221
rsps.add(
@@ -231,7 +231,7 @@ def fixture_mock_get_label(
231231
@pytest.fixture(name="mock_edit_label")
232232
def fixture_mock_edit_label(
233233
base_url: str, repo_owner: str, repo_name: str, response_get_bug: ResponseLabel
234-
) -> None:
234+
) -> Generator:
235235
"""Mock requests for edit label."""
236236
with responses.RequestsMock() as rsps:
237237
rsps.add(
@@ -259,7 +259,7 @@ def fixture_mock_create_label(
259259
repo_name: str,
260260
label: Label,
261261
response_get_bug: ResponseLabel,
262-
) -> None:
262+
) -> Generator:
263263
"""Mock requests for create label."""
264264
with responses.RequestsMock() as rsps:
265265
rsps.add(
@@ -273,7 +273,9 @@ def fixture_mock_create_label(
273273

274274

275275
@pytest.fixture(name="mock_delete_label")
276-
def fixture_mock_delete_label(base_url: str, repo_owner: str, repo_name: str) -> None:
276+
def fixture_mock_delete_label(
277+
base_url: str, repo_owner: str, repo_name: str
278+
) -> Generator:
277279
"""Mock requests for delete label."""
278280
with responses.RequestsMock() as rsps:
279281
rsps.add(
@@ -287,7 +289,7 @@ def fixture_mock_delete_label(base_url: str, repo_owner: str, repo_name: str) ->
287289
@pytest.fixture(name="mock_sync")
288290
def fixture_mock_sync(
289291
base_url: str, repo_owner: str, repo_name: str, response_list_labels: ResponseLabels
290-
) -> None:
292+
) -> Generator:
291293
with responses.RequestsMock() as rsps:
292294
# Response mock for when sync requests the existing remote labels
293295
rsps.add(
@@ -343,7 +345,7 @@ def fixture_mock_sync(
343345

344346

345347
@pytest.fixture(name="labels")
346-
def fixture_labels() -> typing.List[Label]:
348+
def fixture_labels() -> List[Label]:
347349
"""Return a list of Label instances."""
348350
return [
349351
Label(
@@ -385,7 +387,7 @@ def fixture_labels() -> typing.List[Label]:
385387

386388

387389
@pytest.fixture(name="labels_file_dict")
388-
def fixture_labels_file_content() -> typing.Dict[str, typing.Any]:
390+
def fixture_labels_file_content() -> Dict[str, Any]:
389391
"""Return a mapping from label names to dicts representing Labels."""
390392
return {
391393
"bug": {
@@ -437,7 +439,7 @@ def fixture_labels_file_content() -> typing.Dict[str, typing.Any]:
437439

438440

439441
@pytest.fixture(name="labels_file_write")
440-
def fixture_labels_file_write(tmpdir: typing.Any) -> str:
442+
def fixture_labels_file_write(tmpdir: Any) -> str:
441443
"""Return a filepath to a temporary file."""
442444
labels_file = tmpdir.join("labels.toml")
443445
return str(labels_file)
@@ -450,6 +452,6 @@ def fixture_labels_file_load() -> str:
450452

451453

452454
@pytest.fixture(name="labels_file_sync")
453-
def fixture_labels_file_sync(tmpdir: typing.Any) -> str:
455+
def fixture_labels_file_sync(tmpdir: Any) -> str:
454456
"""Return a filepath to an existing labels file for the sync test."""
455457
return "tests/sync.toml"

0 commit comments

Comments
 (0)