mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-24 15:27:17 +00:00
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
import argparse
|
|
|
|
from pathlib import Path
|
|
from pytest_mock import MockerFixture
|
|
from unittest.mock import MagicMock
|
|
|
|
from ahriman.application.handlers import Backup
|
|
from ahriman.core.configuration import Configuration
|
|
from ahriman.models.repository_paths import RepositoryPaths
|
|
|
|
|
|
def _default_args(args: argparse.Namespace) -> argparse.Namespace:
|
|
"""
|
|
default arguments for these test cases
|
|
|
|
Args:
|
|
args(argparse.Namespace): command line arguments fixture
|
|
|
|
Returns:
|
|
argparse.Namespace: generated arguments for these test cases
|
|
"""
|
|
args.path = Path("result.tar.gz")
|
|
return args
|
|
|
|
|
|
def test_run(args: argparse.Namespace, configuration: Configuration, mocker: MockerFixture) -> None:
|
|
"""
|
|
must run command
|
|
"""
|
|
args = _default_args(args)
|
|
mocker.patch("ahriman.application.handlers.Backup.get_paths", return_value=[Path("path")])
|
|
tarfile = MagicMock()
|
|
add_mock = tarfile.__enter__.return_value = MagicMock()
|
|
mocker.patch("tarfile.TarFile.__new__", return_value=tarfile)
|
|
|
|
Backup.run(args, "x86_64", configuration, report=False, unsafe=False)
|
|
add_mock.add.assert_called_once_with(Path("path"))
|
|
|
|
|
|
def test_get_paths(configuration: Configuration, mocker: MockerFixture) -> None:
|
|
"""
|
|
must get paths to be archived
|
|
"""
|
|
# gnupg export mock
|
|
mocker.patch("pathlib.Path.is_dir", return_value=True)
|
|
mocker.patch.object(RepositoryPaths, "root_owner", (42, 42))
|
|
getpwuid_mock = mocker.patch("pwd.getpwuid", return_value=MagicMock())
|
|
# well database does not exist so we override it
|
|
database_mock = mocker.patch("ahriman.core.database.SQLite.database_path", return_value=configuration.path)
|
|
|
|
paths = Backup.get_paths(configuration)
|
|
getpwuid_mock.assert_called_once_with(42)
|
|
database_mock.assert_called_once_with(configuration)
|
|
assert configuration.path in paths
|
|
assert all(path.exists() for path in paths if path.name not in (".gnupg", "cache"))
|
|
|
|
|
|
def test_disallow_auto_architecture_run() -> None:
|
|
"""
|
|
must not allow multi architecture run
|
|
"""
|
|
assert not Backup.ALLOW_AUTO_ARCHITECTURE_RUN
|