mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-27 08:47:18 +00:00
Evgeniy Alekseev
33cfa8fef8
* add init subcommand * add also init command to repository object * add ability to generate list of architectures * check if architecture list is not empty
98 lines
3.1 KiB
Python
98 lines
3.1 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
|
|
|
|
|
|
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_extract_architectures(args: argparse.Namespace, mocker: MockerFixture) -> None:
|
|
"""
|
|
must generate list of available architectures
|
|
"""
|
|
args.architecture = []
|
|
args.configuration = Path("")
|
|
mocker.patch("ahriman.core.configuration.Configuration.getpath")
|
|
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, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception if no available architectures found
|
|
"""
|
|
args.architecture = []
|
|
args.command = "config"
|
|
args.configuration = Path("")
|
|
mocker.patch("ahriman.core.configuration.Configuration.getpath")
|
|
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) -> None:
|
|
"""
|
|
must raise exception on missing architectures
|
|
"""
|
|
args.command = "config"
|
|
args.architecture = None
|
|
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)
|