diff options
author | Devaev Maxim <[email protected]> | 2019-05-04 09:47:33 +0300 |
---|---|---|
committer | Devaev Maxim <[email protected]> | 2019-05-04 09:47:33 +0300 |
commit | 4901cf78ad11ee22ace63f94da3bf8c700978964 (patch) | |
tree | 978640cd707d6198745a3174398dd89b29c7def5 /testenv/tests/auth/test_manager.py | |
parent | 5f5a6a9a81af0fcf6566575e400c2baf20b50eb9 (diff) |
tests reorganized
Diffstat (limited to 'testenv/tests/auth/test_manager.py')
-rw-r--r-- | testenv/tests/auth/test_manager.py | 133 |
1 files changed, 0 insertions, 133 deletions
diff --git a/testenv/tests/auth/test_manager.py b/testenv/tests/auth/test_manager.py deleted file mode 100644 index be6b6455..00000000 --- a/testenv/tests/auth/test_manager.py +++ /dev/null @@ -1,133 +0,0 @@ -# ========================================================================== # -# # -# KVMD - The main Pi-KVM daemon. # -# # -# Copyright (C) 2018 Maxim Devaev <[email protected]> # -# # -# This program is free software: you can redistribute it and/or modify # -# it under the terms of the GNU General Public License as published by # -# the Free Software Foundation, either version 3 of the License, or # -# (at your option) any later version. # -# # -# This program is distributed in the hope that it will be useful, # -# but WITHOUT ANY WARRANTY; without even the implied warranty of # -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # -# GNU General Public License for more details. # -# # -# You should have received a copy of the GNU General Public License # -# along with this program. If not, see <https://www.gnu.org/licenses/>. # -# # -# ========================================================================== # - - -import os -import contextlib - -from typing import List -from typing import Dict -from typing import AsyncGenerator -from typing import Optional - -import passlib.apache - -import pytest - -from kvmd.yamlconf import make_config - -from kvmd.apps.kvmd.auth import AuthManager - -from kvmd.plugins.auth import get_auth_service_class - - -# ===== -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() # pylint: disable=protected-access - - -async def _get_configured_manager( - internal_path: str, - external_path: str="", - internal_users: Optional[List[str]]=None, -) -> AsyncGenerator[AuthManager, None]: - - manager = AuthManager( - internal_type="htpasswd", - internal_kwargs=_make_service_kwargs(internal_path), - external_type=("htpasswd" if external_path else ""), - external_kwargs=(_make_service_kwargs(external_path) if external_path else {}), - internal_users=(internal_users or []), - ) - - try: - yield manager - finally: - await manager.cleanup() - - -# ===== -async def test_ok__internal(tmpdir) -> None: # type: ignore - path = os.path.abspath(str(tmpdir.join("htpasswd"))) - - htpasswd = passlib.apache.HtpasswdFile(path, new=True) - htpasswd.set_password("admin", "pass") - htpasswd.save() - - async with _get_configured_manager(path) as manager: - assert manager.check("xxx") is None - manager.logout("xxx") - - assert (await manager.login("user", "foo")) is None - assert (await manager.login("admin", "foo")) is None - assert (await manager.login("user", "pass")) is None - - token = await manager.login("admin", "pass") - assert isinstance(token, str) - assert len(token) == 64 - - again = await manager.login("admin", "pass") - assert token == again - - assert manager.check(token) == "admin" - manager.logout(token) - assert manager.check(token) is None - - again = await manager.login("admin", "pass") - assert token != again - - -async def test_ok__external(tmpdir) -> None: # type: ignore - path1 = os.path.abspath(str(tmpdir.join("htpasswd1"))) - path2 = os.path.abspath(str(tmpdir.join("htpasswd2"))) - - htpasswd1 = passlib.apache.HtpasswdFile(path1, new=True) - htpasswd1.set_password("admin", "pass1") - htpasswd1.set_password("local", "foobar") - htpasswd1.save() - - htpasswd2 = passlib.apache.HtpasswdFile(path2, new=True) - htpasswd2.set_password("admin", "pass2") - htpasswd2.set_password("user", "foobar") - htpasswd2.save() - - async with _get_configured_manager(path1, path2, ["admin"]) as manager: - assert (await manager.login("local", "foobar")) is None - assert (await manager.login("admin", "pass2")) is None - - token = await manager.login("admin", "pass1") - assert token is not None - - assert manager.check(token) == "admin" - manager.logout(token) - assert manager.check(token) is None - - token = await manager.login("user", "foobar") - assert token is not None - - assert manager.check(token) == "user" - manager.logout(token) - assert manager.check(token) is None |