mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-27 08:47:18 +00:00
Evgeniy Alekseev
09b0f2914d
This feature also introduces the followiing changes * aur-search command now works as expected with multiterms * printer classes for managing of data print * --sort-by argument for aur-search subcommand instead of using package name * --quiet argument now has also --no-quite option * if --quite is supplied, the log level will be set to warn instead of critical to be able to see error messages * pretty_datetime function now also supports datetime objects * BuildStatus is now pure dataclass
109 lines
4.4 KiB
Python
109 lines
4.4 KiB
Python
import argparse
|
|
|
|
from pytest_mock import MockerFixture
|
|
from unittest import mock
|
|
|
|
from ahriman.application.handlers import Status
|
|
from ahriman.core.configuration import Configuration
|
|
from ahriman.models.build_status import BuildStatus, BuildStatusEnum
|
|
from ahriman.models.package import Package
|
|
|
|
|
|
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.ahriman = True
|
|
args.info = False
|
|
args.package = []
|
|
args.status = None
|
|
return args
|
|
|
|
|
|
def test_run(args: argparse.Namespace, configuration: Configuration, package_ahriman: Package,
|
|
package_python_schedule: Package, mocker: MockerFixture) -> None:
|
|
"""
|
|
must run command
|
|
"""
|
|
args = _default_args(args)
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.tree_create")
|
|
application_mock = mocker.patch("ahriman.core.status.client.Client.get_self")
|
|
packages_mock = mocker.patch("ahriman.core.status.client.Client.get",
|
|
return_value=[(package_ahriman, BuildStatus(BuildStatusEnum.Success)),
|
|
(package_python_schedule, BuildStatus(BuildStatusEnum.Failed))])
|
|
print_mock = mocker.patch("ahriman.application.formatters.printer.Printer.print")
|
|
|
|
Status.run(args, "x86_64", configuration, True)
|
|
application_mock.assert_called_once()
|
|
packages_mock.assert_called_once()
|
|
print_mock.assert_has_calls([mock.call(False) for _ in range(3)])
|
|
|
|
|
|
def test_run_verbose(args: argparse.Namespace, configuration: Configuration, package_ahriman: Package,
|
|
mocker: MockerFixture) -> None:
|
|
"""
|
|
must run command
|
|
"""
|
|
args = _default_args(args)
|
|
args.info = True
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.tree_create")
|
|
mocker.patch("ahriman.core.status.client.Client.get",
|
|
return_value=[(package_ahriman, BuildStatus(BuildStatusEnum.Success))])
|
|
print_mock = mocker.patch("ahriman.application.formatters.printer.Printer.print")
|
|
|
|
Status.run(args, "x86_64", configuration, True)
|
|
print_mock.assert_has_calls([mock.call(True) for _ in range(2)])
|
|
|
|
|
|
def test_run_with_package_filter(args: argparse.Namespace, configuration: Configuration, package_ahriman: Package,
|
|
mocker: MockerFixture) -> None:
|
|
"""
|
|
must run command with package filter
|
|
"""
|
|
args = _default_args(args)
|
|
args.package = [package_ahriman.base]
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.tree_create")
|
|
packages_mock = mocker.patch("ahriman.core.status.client.Client.get",
|
|
return_value=[(package_ahriman, BuildStatus(BuildStatusEnum.Success))])
|
|
|
|
Status.run(args, "x86_64", configuration, True)
|
|
packages_mock.assert_called_once_with(package_ahriman.base)
|
|
|
|
|
|
def test_run_by_status(args: argparse.Namespace, configuration: Configuration, package_ahriman: Package,
|
|
package_python_schedule: Package, mocker: MockerFixture) -> None:
|
|
"""
|
|
must filter packages by status
|
|
"""
|
|
args = _default_args(args)
|
|
args.status = BuildStatusEnum.Failed
|
|
mocker.patch("ahriman.core.status.client.Client.get",
|
|
return_value=[(package_ahriman, BuildStatus(BuildStatusEnum.Success)),
|
|
(package_python_schedule, BuildStatus(BuildStatusEnum.Failed))])
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.tree_create")
|
|
print_mock = mocker.patch("ahriman.application.formatters.printer.Printer.print")
|
|
|
|
Status.run(args, "x86_64", configuration, True)
|
|
print_mock.assert_has_calls([mock.call(False) for _ in range(2)])
|
|
|
|
|
|
def test_imply_with_report(args: argparse.Namespace, configuration: Configuration, mocker: MockerFixture) -> None:
|
|
"""
|
|
must create application object with native reporting
|
|
"""
|
|
args = _default_args(args)
|
|
mocker.patch("ahriman.models.repository_paths.RepositoryPaths.tree_create")
|
|
load_mock = mocker.patch("ahriman.core.status.client.Client.load")
|
|
|
|
Status.run(args, "x86_64", configuration, True)
|
|
load_mock.assert_called_once()
|
|
|
|
|
|
def test_disallow_auto_architecture_run() -> None:
|
|
"""
|
|
must not allow multi architecture run
|
|
"""
|
|
assert not Status.ALLOW_AUTO_ARCHITECTURE_RUN
|