mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-24 07:17:17 +00:00
Evgeniy Alekseev
1192f12f91
according to the source code defaults always updates the values dictionary. This in this specific case it is impossible to override the value it will be always empty list. In order to handle it we are adding another property to the Handler class which allows to run with None architecture list. This particular set_defaults behaviour is still useful for other cases when we have to run command without any specific architecture
118 lines
3.9 KiB
Python
118 lines
3.9 KiB
Python
import argparse
|
|
import pytest
|
|
|
|
from pathlib import Path
|
|
from pytest_mock import MockerFixture
|
|
|
|
from ahriman.application.handlers import Handler
|
|
from ahriman.core.configuration import Configuration
|
|
from ahriman.core.exceptions import MissingArchitecture, MultipleArchitecture
|
|
|
|
|
|
def test_call(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must call inside lock
|
|
"""
|
|
args.configuration = Path("")
|
|
args.no_log = False
|
|
mocker.patch("ahriman.application.handlers.Handler.run")
|
|
mocker.patch("ahriman.core.configuration.Configuration.from_path")
|
|
enter_mock = mocker.patch("ahriman.application.lock.Lock.__enter__")
|
|
exit_mock = mocker.patch("ahriman.application.lock.Lock.__exit__")
|
|
|
|
assert Handler.call(args, "x86_64")
|
|
enter_mock.assert_called_once()
|
|
exit_mock.assert_called_once()
|
|
|
|
|
|
def test_call_exception(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must process exception
|
|
"""
|
|
mocker.patch("ahriman.application.lock.Lock.__enter__", side_effect=Exception())
|
|
assert not Handler.call(args, "x86_64")
|
|
|
|
|
|
def test_execute(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must run execution in multiple processes
|
|
"""
|
|
args.architecture = ["i686", "x86_64"]
|
|
starmap_mock = mocker.patch("multiprocessing.pool.Pool.starmap")
|
|
|
|
Handler.execute(args)
|
|
starmap_mock.assert_called_once()
|
|
|
|
|
|
def test_execute_multiple_not_supported(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise an exception if multiple architectures are not supported by the handler
|
|
"""
|
|
args.architecture = ["i686", "x86_64"]
|
|
args.command = "web"
|
|
mocker.patch.object(Handler, "ALLOW_MULTI_ARCHITECTURE_RUN", False)
|
|
|
|
with pytest.raises(MultipleArchitecture):
|
|
Handler.execute(args)
|
|
|
|
|
|
def test_execute_single(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must run execution in current process if only one architecture supplied
|
|
"""
|
|
args.architecture = ["x86_64"]
|
|
starmap_mock = mocker.patch("multiprocessing.pool.Pool.starmap")
|
|
|
|
Handler.execute(args)
|
|
starmap_mock.assert_not_called()
|
|
|
|
|
|
def test_extract_architectures(args: argparse.Namespace, configuration: Configuration, mocker: MockerFixture) -> None:
|
|
"""
|
|
must generate list of available architectures
|
|
"""
|
|
args.configuration = configuration.path
|
|
known_architectures_mock = mocker.patch("ahriman.models.repository_paths.RepositoryPaths.known_architectures")
|
|
|
|
Handler.extract_architectures(args)
|
|
known_architectures_mock.assert_called_once()
|
|
|
|
|
|
def test_extract_architectures_empty(args: argparse.Namespace, configuration: Configuration,
|
|
mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception if no available architectures found
|
|
"""
|
|
args.command = "config"
|
|
args.configuration = configuration.path
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.known_architectures", return_value=set())
|
|
|
|
with pytest.raises(MissingArchitecture):
|
|
Handler.extract_architectures(args)
|
|
|
|
|
|
def test_extract_architectures_exception(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception on missing architectures
|
|
"""
|
|
args.command = "config"
|
|
mocker.patch.object(Handler, "ALLOW_AUTO_ARCHITECTURE_RUN", False)
|
|
with pytest.raises(MissingArchitecture):
|
|
Handler.extract_architectures(args)
|
|
|
|
|
|
def test_extract_architectures_specified(args: argparse.Namespace) -> None:
|
|
"""
|
|
must return architecture list if it has been specified
|
|
"""
|
|
architectures = args.architecture = ["i686", "x86_64"]
|
|
assert Handler.extract_architectures(args) == set(architectures)
|
|
|
|
|
|
def test_run(args: argparse.Namespace, configuration: Configuration) -> None:
|
|
"""
|
|
must raise NotImplemented for missing method
|
|
"""
|
|
with pytest.raises(NotImplementedError):
|
|
Handler.run(args, "x86_64", configuration, True)
|