new typing style

This commit is contained in:
Maxim Devaev
2022-09-04 18:08:40 +03:00
parent 4b75221e94
commit ee3e224e39
129 changed files with 593 additions and 941 deletions

View File

@@ -26,7 +26,6 @@ import tempfile
import builtins
import getpass
from typing import List
from typing import Generator
from typing import Any
@@ -54,7 +53,7 @@ def _htpasswd_fixture(request) -> Generator[passlib.apache.HtpasswdFile, None, N
os.remove(path)
def _run_htpasswd(cmd: List[str], htpasswd_path: str, internal_type: str="htpasswd") -> None:
def _run_htpasswd(cmd: list[str], htpasswd_path: str, internal_type: str="htpasswd") -> None:
cmd = ["kvmd-htpasswd", *cmd, "--set-options"]
if internal_type != "htpasswd": # By default
cmd.append("kvmd/auth/internal/type=" + internal_type)

View File

@@ -23,10 +23,7 @@
import os
import contextlib
from typing import List
from typing import Dict
from typing import AsyncGenerator
from typing import Optional
import passlib.apache
@@ -40,7 +37,7 @@ from kvmd.plugins.auth import get_auth_service_class
# =====
def _make_service_kwargs(path: str) -> Dict:
def _make_service_kwargs(path: str) -> dict:
cls = get_auth_service_class("htpasswd")
scheme = cls.get_plugin_options()
return make_config({"file": path}, scheme)._unpack()
@@ -50,7 +47,7 @@ def _make_service_kwargs(path: str) -> Dict:
async def _get_configured_manager(
internal_path: str,
external_path: str="",
force_internal_users: Optional[List[str]]=None,
force_internal_users: (list[str] | None)=None,
) -> AsyncGenerator[AuthManager, None]:
manager = AuthManager(

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
from typing import Dict
from typing import AsyncGenerator
import aiohttp.web
@@ -68,7 +67,7 @@ async def _auth_server_port_fixture(aiohttp_server) -> AsyncGenerator[int, None]
{"verify": False},
{"user": "server-admin", "passwd": "server-pass"},
])
async def test_ok(auth_server_port: int, kwargs: Dict) -> None:
async def test_ok(auth_server_port: int, kwargs: dict) -> None:
url = "http://localhost:%d/%s" % (
auth_server_port,
("auth_plus_basic" if kwargs.get("user") else "auth"),

View File

@@ -24,9 +24,7 @@ import os
import asyncio
import pwd
from typing import Dict
from typing import AsyncGenerator
from typing import Optional
import pytest
import pytest_asyncio
@@ -41,7 +39,7 @@ _PASSWD = "query"
# =====
async def _run_process(cmd: str, input: Optional[str]=None) -> None: # pylint: disable=redefined-builtin
async def _run_process(cmd: str, input: (str | None)=None) -> None: # pylint: disable=redefined-builtin
proc = await asyncio.create_subprocess_exec(
*cmd.split(" "),
stdin=(asyncio.subprocess.PIPE if input is not None else None),
@@ -75,7 +73,7 @@ async def _test_user() -> AsyncGenerator[None, None]:
{"allow_users": [_USER]},
{"allow_uids_at": _UID},
])
async def test_ok(test_user, kwargs: Dict) -> None: # type: ignore
async def test_ok(test_user, kwargs: dict) -> None: # type: ignore
_ = test_user
async with get_configured_auth_service("pam", **kwargs) as service:
assert not (await service.authorize(_USER, "invalid_password"))
@@ -88,7 +86,7 @@ async def test_ok(test_user, kwargs: Dict) -> None: # type: ignore
{"deny_users": [_USER]},
{"allow_uids_at": _UID + 1},
])
async def test_fail(test_user, kwargs: Dict) -> None: # type: ignore
async def test_fail(test_user, kwargs: dict) -> None: # type: ignore
_ = test_user
async with get_configured_auth_service("pam", **kwargs) as service:
assert not (await service.authorize(_USER, "invalid_password"))

View File

@@ -22,8 +22,6 @@
import asyncio
from typing import List
import pytest
from kvmd.aiotools import AioExclusiveRegion
@@ -123,7 +121,7 @@ async def test_fail__region__access_two() -> None:
# =====
@pytest.mark.asyncio
async def test_ok__shield_fg() -> None:
ops: List[str] = []
ops: list[str] = []
async def foo(op: str, delay: float) -> None: # pylint: disable=disallowed-name
await asyncio.sleep(delay)

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
from typing import List
from typing import Any
import pytest
@@ -72,7 +71,7 @@ def test_fail__valid_user(arg: Any) -> None:
(", foo, ", ["foo"]),
([], []),
])
def test_ok__valid_users_list(arg: Any, retval: List) -> None:
def test_ok__valid_users_list(arg: Any, retval: list) -> None:
assert valid_users_list(arg) == retval

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
from typing import List
from typing import Any
import pytest
@@ -151,7 +150,7 @@ def test_fail__valid_float_f01(arg: Any) -> None:
(", a, ", ["a"]),
([], []),
])
def test_ok__valid_string_list(arg: Any, retval: List) -> None:
def test_ok__valid_string_list(arg: Any, retval: list) -> None:
assert valid_string_list(arg) == retval
@@ -165,7 +164,7 @@ def test_ok__valid_string_list(arg: Any, retval: List) -> None:
(", 1, ", [1]),
([], []),
])
def test_ok__valid_string_list__subval(arg: Any, retval: List) -> None: # pylint: disable=invalid-name
def test_ok__valid_string_list__subval(arg: Any, retval: list) -> None: # pylint: disable=invalid-name
assert valid_string_list(arg, subval=int) == retval

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
from typing import List
from typing import Any
import pytest
@@ -168,7 +167,7 @@ def test_fail__valid_port(arg: Any) -> None:
("80,443,443,", [80, 443, 443]),
(65535, [65535]),
])
def test_ok__valid_ports_list(arg: Any, retval: List[int]) -> None:
def test_ok__valid_ports_list(arg: Any, retval: list[int]) -> None:
assert valid_ports_list(arg) == retval

View File

@@ -22,7 +22,6 @@
import os
from typing import List
from typing import Any
import pytest
@@ -148,7 +147,7 @@ def test_fail__valid_unix_mode(arg: Any) -> None:
("/bin/true, 1, 2, 3,", ["/bin/true", "1", "2", "3"]),
("/bin/true", ["/bin/true"]),
])
def test_ok__valid_command(arg: Any, retval: List[str]) -> None:
def test_ok__valid_command(arg: Any, retval: list[str]) -> None:
assert valid_command(arg) == retval