|
13 | 13 | # limitations under the License. |
14 | 14 |
|
15 | 15 | import os |
16 | | -from typing import Any, List, NoReturn |
| 16 | +from typing import List, NoReturn, TypeVar |
17 | 17 |
|
18 | 18 | from click.testing import CliRunner |
19 | 19 | from mock import MagicMock, patch |
20 | 20 | import pytest |
21 | 21 |
|
22 | 22 | from check_status import cli |
23 | 23 |
|
| 24 | +a = TypeVar("a") |
| 25 | + |
24 | 26 | MOCK_SERVICE_NAME = "myservice" |
25 | 27 | MOCK_GH_TOKEN = "aaaaa" |
26 | 28 | MOCK_REPO_NAME = "foocorp/bar" |
@@ -70,7 +72,7 @@ def service_data(name: str, tags: List[str]) -> dict: |
70 | 72 |
|
71 | 73 |
|
72 | 74 | @patch("check_status.discovery") |
73 | | -def test_set_wrongtag(discovery_mock: Any) -> NoReturn: |
| 75 | +def test_set_wrongtag(discovery_mock: a) -> NoReturn: |
74 | 76 | service_mock = MagicMock() |
75 | 77 | service_mock.projects = MagicMock(return_value=service_mock) |
76 | 78 | service_mock.locations = MagicMock(return_value=service_mock) |
@@ -110,7 +112,7 @@ def test_set_wrongtag(discovery_mock: Any) -> NoReturn: |
110 | 112 |
|
111 | 113 | @patch("check_status.discovery") |
112 | 114 | @patch("check_status.github") |
113 | | -def test_set_check_calls(github_mock: Any, discovery_mock: Any) -> NoReturn: |
| 115 | +def test_set_check_calls(github_mock: a, discovery_mock: a) -> NoReturn: |
114 | 116 | service_mock = MagicMock() |
115 | 117 | service_mock.projects = MagicMock(return_value=service_mock) |
116 | 118 | service_mock.locations = MagicMock(return_value=service_mock) |
|
0 commit comments