summaryrefslogtreecommitdiff
path: root/testenv/tests/apps/test_htpasswd.py
diff options
context:
space:
mode:
authorDevaev Maxim <[email protected]>2019-05-04 09:47:33 +0300
committerDevaev Maxim <[email protected]>2019-05-04 09:47:33 +0300
commit4901cf78ad11ee22ace63f94da3bf8c700978964 (patch)
tree978640cd707d6198745a3174398dd89b29c7def5 /testenv/tests/apps/test_htpasswd.py
parent5f5a6a9a81af0fcf6566575e400c2baf20b50eb9 (diff)
tests reorganized
Diffstat (limited to 'testenv/tests/apps/test_htpasswd.py')
-rw-r--r--testenv/tests/apps/test_htpasswd.py170
1 files changed, 0 insertions, 170 deletions
diff --git a/testenv/tests/apps/test_htpasswd.py b/testenv/tests/apps/test_htpasswd.py
deleted file mode 100644
index 52a38640..00000000
--- a/testenv/tests/apps/test_htpasswd.py
+++ /dev/null
@@ -1,170 +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 hashlib
-import tempfile
-import builtins
-import getpass
-
-from typing import List
-from typing import Generator
-from typing import Any
-
-import passlib.apache
-
-import pytest
-
-from kvmd.apps.htpasswd import main
-
-
-# =====
-def _make_passwd(user: str) -> str:
- return hashlib.md5(user.encode()).hexdigest()
-
-
[email protected](name="htpasswd", params=[[], ["admin"], ["admin", "user"]])
-def _htpasswd_fixture(request) -> Generator[passlib.apache.HtpasswdFile, None, None]: # type: ignore
- (fd, path) = tempfile.mkstemp()
- os.close(fd)
- htpasswd = passlib.apache.HtpasswdFile(path)
- for user in request.param:
- htpasswd.set_password(user, _make_passwd(user))
- htpasswd.save()
- yield htpasswd
- os.remove(path)
-
-
-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)
- if htpasswd_path:
- cmd.append("kvmd/auth/internal/file=" + htpasswd_path)
- main(cmd)
-
-
-# =====
-def test_ok__list(htpasswd: passlib.apache.HtpasswdFile, capsys) -> None: # type: ignore
- _run_htpasswd(["list"], htpasswd.path)
- (out, err) = capsys.readouterr()
- assert len(err) == 0
- assert sorted(filter(None, out.split("\n"))) == sorted(htpasswd.users()) == sorted(set(htpasswd.users()))
-
-
-# =====
-def test_ok__set_change_stdin(htpasswd: passlib.apache.HtpasswdFile, mocker) -> None: # type: ignore
- old_users = set(htpasswd.users())
- if old_users:
- assert htpasswd.check_password("admin", _make_passwd("admin"))
-
- mocker.patch.object(builtins, "input", (lambda: " test "))
- _run_htpasswd(["set", "admin", "--read-stdin"], htpasswd.path)
-
- htpasswd.load(force=True)
- assert htpasswd.check_password("admin", " test ")
- assert old_users == set(htpasswd.users())
-
-
-def test_ok__set_add_stdin(htpasswd: passlib.apache.HtpasswdFile, mocker) -> None: # type: ignore
- old_users = set(htpasswd.users())
- if old_users:
- mocker.patch.object(builtins, "input", (lambda: " test "))
- _run_htpasswd(["set", "new", "--read-stdin"], htpasswd.path)
-
- htpasswd.load(force=True)
- assert htpasswd.check_password("new", " test ")
- assert old_users.union(["new"]) == set(htpasswd.users())
-
-
-# =====
-def test_ok__set_change_getpass(htpasswd: passlib.apache.HtpasswdFile, mocker) -> None: # type: ignore
- old_users = set(htpasswd.users())
- if old_users:
- assert htpasswd.check_password("admin", _make_passwd("admin"))
-
- mocker.patch.object(getpass, "getpass", (lambda *_, **__: " test "))
- _run_htpasswd(["set", "admin"], htpasswd.path)
-
- htpasswd.load(force=True)
- assert htpasswd.check_password("admin", " test ")
- assert old_users == set(htpasswd.users())
-
-
-def test_fail__set_change_getpass(htpasswd: passlib.apache.HtpasswdFile, mocker) -> None: # type: ignore
- old_users = set(htpasswd.users())
- if old_users:
- assert htpasswd.check_password("admin", _make_passwd("admin"))
-
- count = 0
-
- def fake_getpass(*_: Any, **__: Any) -> str:
- nonlocal count
- assert count <= 1
- if count == 0:
- passwd = " test "
- else:
- passwd = "test "
- count += 1
- return passwd
-
- mocker.patch.object(getpass, "getpass", fake_getpass)
- with pytest.raises(SystemExit, match="Sorry, passwords do not match"):
- _run_htpasswd(["set", "admin"], htpasswd.path)
- assert count == 2
-
- htpasswd.load(force=True)
- assert htpasswd.check_password("admin", _make_passwd("admin"))
- assert old_users == set(htpasswd.users())
-
-
-# =====
-def test_ok__del(htpasswd: passlib.apache.HtpasswdFile) -> None:
- old_users = set(htpasswd.users())
-
- if old_users:
- assert htpasswd.check_password("admin", _make_passwd("admin"))
-
- _run_htpasswd(["del", "admin"], htpasswd.path)
-
- htpasswd.load(force=True)
- assert not htpasswd.check_password("admin", _make_passwd("admin"))
- assert old_users.difference(["admin"]) == set(htpasswd.users())
-
-
-# =====
-def test_fail__not_htpasswd() -> None:
- with pytest.raises(SystemExit, match="Error: KVMD internal auth not using 'htpasswd'"):
- _run_htpasswd(["list"], "", internal_type="http")
-
-
-def test_fail__unknown_plugin() -> None:
- with pytest.raises(SystemExit, match="Config error: Unknown plugin 'auth/foobar'"):
- _run_htpasswd(["list"], "", internal_type="foobar")
-
-
-def test_fail__invalid_passwd(mocker, tmpdir) -> None: # type: ignore
- path = os.path.abspath(str(tmpdir.join("htpasswd")))
- open(path, "w").close()
- mocker.patch.object(builtins, "input", (lambda: "\n"))
- with pytest.raises(SystemExit, match="The argument is not a valid passwd characters"):
- _run_htpasswd(["set", "admin", "--read-stdin"], path)