mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-27 08:47:18 +00:00
Evgenii Alekseev
e63cb509f2
* initial auth implementation * add create user parser * add tests * update dependencies list * add login annd logout to index also improve auth * realworld fixes * add method set_option to Configuration and also use it everywhere * split CreateUser handler to additional read method * check user duplicate on auth mapping read * generate salt by using passlib instead of random.choice * case-insensetive usernames * update dependencies * update configuration reference * improve tests * fix codefactor errors * hide fields if authorization is enabled, but no auth supplied * add settings object for auth provider * readme update
145 lines
4.8 KiB
Python
145 lines
4.8 KiB
Python
import argparse
|
|
import pytest
|
|
|
|
from pathlib import Path
|
|
from pytest_mock import MockerFixture
|
|
from unittest import mock
|
|
|
|
from ahriman.application.handlers import CreateUser
|
|
from ahriman.core.configuration import Configuration
|
|
from ahriman.models.user import User
|
|
from ahriman.models.user_access import UserAccess
|
|
|
|
|
|
def _default_args(args: argparse.Namespace) -> argparse.Namespace:
|
|
"""
|
|
default arguments for these test cases
|
|
:param args: command line arguments fixture
|
|
:return: generated arguments for these test cases
|
|
"""
|
|
args.username = "user"
|
|
args.password = "pa55w0rd"
|
|
args.role = UserAccess.Status
|
|
return args
|
|
|
|
|
|
def test_run(args: argparse.Namespace, configuration: Configuration, mocker: MockerFixture) -> None:
|
|
"""
|
|
must run command
|
|
"""
|
|
args = _default_args(args)
|
|
get_auth_configuration_mock = mocker.patch("ahriman.application.handlers.CreateUser.get_auth_configuration")
|
|
create_configuration_mock = mocker.patch("ahriman.application.handlers.CreateUser.create_configuration")
|
|
create_user = mocker.patch("ahriman.application.handlers.CreateUser.create_user")
|
|
get_salt_mock = mocker.patch("ahriman.application.handlers.CreateUser.get_salt")
|
|
|
|
CreateUser.run(args, "x86_64", configuration)
|
|
get_auth_configuration_mock.assert_called_once()
|
|
create_configuration_mock.assert_called_once()
|
|
create_user.assert_called_once()
|
|
get_salt_mock.assert_called_once()
|
|
|
|
|
|
def test_create_configuration(configuration: Configuration, user: User, mocker: MockerFixture) -> None:
|
|
"""
|
|
must correctly create configuration file
|
|
"""
|
|
section = Configuration.section_name("auth", user.access.value)
|
|
mocker.patch("pathlib.Path.open")
|
|
set_mock = mocker.patch("ahriman.core.configuration.Configuration.set_option")
|
|
write_mock = mocker.patch("ahriman.core.configuration.Configuration.write")
|
|
|
|
CreateUser.create_configuration(configuration, user, "salt")
|
|
write_mock.assert_called_once()
|
|
set_mock.assert_has_calls([
|
|
mock.call("auth", "salt", pytest.helpers.anyvar(str)),
|
|
mock.call(section, user.username, user.password)
|
|
])
|
|
|
|
|
|
def test_create_configuration_not_loaded(configuration: Configuration, user: User, mocker: MockerFixture) -> None:
|
|
"""
|
|
must do nothing in case if configuration is not loaded
|
|
"""
|
|
configuration.path = None
|
|
mocker.patch("pathlib.Path.open")
|
|
write_mock = mocker.patch("ahriman.core.configuration.Configuration.write")
|
|
|
|
CreateUser.create_configuration(configuration, user, "salt")
|
|
write_mock.assert_not_called()
|
|
|
|
|
|
def test_create_configuration_user_exists(configuration: Configuration, user: User, mocker: MockerFixture) -> None:
|
|
"""
|
|
must correctly update configuration file if user already exists
|
|
"""
|
|
section = Configuration.section_name("auth", user.access.value)
|
|
configuration.set_option(section, user.username, "")
|
|
mocker.patch("pathlib.Path.open")
|
|
mocker.patch("ahriman.core.configuration.Configuration.write")
|
|
|
|
CreateUser.create_configuration(configuration, user, "salt")
|
|
assert configuration.get(section, user.username) == user.password
|
|
|
|
|
|
def test_create_user(args: argparse.Namespace, user: User) -> None:
|
|
"""
|
|
must create user
|
|
"""
|
|
args = _default_args(args)
|
|
generated = CreateUser.create_user(args, "salt")
|
|
assert generated.username == user.username
|
|
assert generated.check_credentials(user.password, "salt")
|
|
assert generated.access == user.access
|
|
|
|
|
|
def test_create_user_getpass(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must create user and get password from command line
|
|
"""
|
|
args = _default_args(args)
|
|
args.password = None
|
|
|
|
getpass_mock = mocker.patch("getpass.getpass", return_value="password")
|
|
generated = CreateUser.create_user(args, "salt")
|
|
|
|
getpass_mock.assert_called_once()
|
|
assert generated.check_credentials("password", "salt")
|
|
|
|
|
|
def test_get_salt_read(configuration: Configuration) -> None:
|
|
"""
|
|
must read salt from configuration
|
|
"""
|
|
assert CreateUser.get_salt(configuration) == "salt"
|
|
|
|
|
|
def test_get_salt_generate(configuration: Configuration) -> None:
|
|
"""
|
|
must generate salt if it does not exist
|
|
"""
|
|
configuration.remove_option("auth", "salt")
|
|
|
|
salt = CreateUser.get_salt(configuration, 16)
|
|
assert salt
|
|
assert len(salt) == 16
|
|
|
|
|
|
def test_get_auth_configuration() -> None:
|
|
"""
|
|
must load empty configuration
|
|
"""
|
|
assert CreateUser.get_auth_configuration(Path("path"))
|
|
|
|
|
|
def test_get_auth_configuration_exists(mocker: MockerFixture) -> None:
|
|
"""
|
|
must load configuration from filesystem
|
|
"""
|
|
mocker.patch("pathlib.Path.open")
|
|
mocker.patch("pathlib.Path.is_file", return_value=True)
|
|
read_mock = mocker.patch("ahriman.core.configuration.Configuration.read")
|
|
|
|
CreateUser.get_auth_configuration(Path("path"))
|
|
read_mock.assert_called_once()
|