|
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