borgmatic/atticmatic/tests/unit/test_attic.py

from collections import OrderedDict
import sys

from flexmock import flexmock
from nose.tools import assert_raises

from atticmatic import attic as module


class MockCalledProcessError(Exception):
    def __init__(self, output):
        self.output = output


def insert_subprocess_check_output_mock(call_command, error_output=None, **kwargs):
    subprocess = flexmock(CalledProcessError=MockCalledProcessError, STDOUT=flexmock())

    expectation = subprocess.should_receive('check_output').with_args(
        call_command,
        stderr=subprocess.STDOUT,
        **kwargs
    ).once()

    if error_output:
        expectation.and_raise(MockCalledProcessError, output=error_output)
        flexmock(sys.modules['__builtin__']).should_receive('print')

    flexmock(module).subprocess = subprocess
    return subprocess


def insert_subprocess_check_call_mock(call_command, **kwargs):
    subprocess = flexmock()

    subprocess.should_receive('check_call').with_args(
        call_command,
        **kwargs
    ).once()

    flexmock(module).subprocess = subprocess
    return subprocess


def insert_platform_mock():
    flexmock(module).platform = flexmock().should_receive('node').and_return('host').mock


def insert_datetime_mock():
    flexmock(module).datetime = flexmock().should_receive('now').and_return(
        flexmock().should_receive('isoformat').and_return('now').mock
    ).mock


def test_create_archive_should_call_attic_with_parameters():
    insert_subprocess_check_output_mock(
        ('attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar'),
    )
    insert_platform_mock()
    insert_datetime_mock()

    module.create_archive(
        excludes_filename='excludes',
        verbose=False,
        source_directories='foo bar',
        repository='repo',
    )


def test_create_archive_with_verbose_should_call_attic_with_verbose_parameters():
    insert_subprocess_check_output_mock(
        (
            'attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar',
            '--verbose', '--stats',
        ),
    )
    insert_platform_mock()
    insert_datetime_mock()

    module.create_archive(
        excludes_filename='excludes',
        verbose=True,
        source_directories='foo bar',
        repository='repo',
    )

def test_create_archive_with_missing_repository_should_raise():
    insert_subprocess_check_output_mock(
        ('attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar'),
        error_output='Error: Repository repo does not exist',
    )
    insert_platform_mock()
    insert_datetime_mock()

    with assert_raises(RuntimeError):
        module.create_archive(
            excludes_filename='excludes',
            verbose=False,
            source_directories='foo bar',
            repository='repo',
        )


def test_create_archive_with_other_error_should_raise():
    subprocess = insert_subprocess_check_output_mock(
        ('attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar'),
        error_output='Something went wrong',
    )
    insert_platform_mock()
    insert_datetime_mock()

    with assert_raises(subprocess.CalledProcessError):
        module.create_archive(
            excludes_filename='excludes',
            verbose=False,
            source_directories='foo bar',
            repository='repo',
        )


BASE_PRUNE_FLAGS = (
    ('--keep-daily', '1'),
    ('--keep-weekly', '2'),
    ('--keep-monthly', '3'),
)


def test_make_prune_flags_should_return_flags_from_config():
    retention_config = OrderedDict(
        (
            ('keep_daily', 1),
            ('keep_weekly', 2),
            ('keep_monthly', 3),
        )
    )

    result = module.make_prune_flags(retention_config)

    assert tuple(result) == BASE_PRUNE_FLAGS


def test_prune_archives_should_call_attic_with_parameters():
    retention_config = flexmock()
    flexmock(module).should_receive('make_prune_flags').with_args(retention_config).and_return(
        BASE_PRUNE_FLAGS,
    )
    insert_subprocess_check_call_mock(
        (
            'attic', 'prune', 'repo', '--keep-daily', '1', '--keep-weekly', '2', '--keep-monthly',
            '3',
        ),
    )

    module.prune_archives(
        verbose=False,
        repository='repo',
        retention_config=retention_config,
    )


def test_prune_archives_with_verbose_should_call_attic_with_verbose_parameters():
    retention_config = flexmock()
    flexmock(module).should_receive('make_prune_flags').with_args(retention_config).and_return(
        BASE_PRUNE_FLAGS,
    )
    insert_subprocess_check_call_mock(
        (
            'attic', 'prune', 'repo', '--keep-daily', '1', '--keep-weekly', '2', '--keep-monthly',
            '3', '--verbose',
        ),
    )

    module.prune_archives(
        repository='repo',
        verbose=True,
        retention_config=retention_config,
    )


def test_check_archives_should_call_attic_with_parameters():
    stdout = flexmock()
    insert_subprocess_check_call_mock(
        ('attic', 'check', 'repo'),
        stdout=stdout,
    )
    insert_platform_mock()
    insert_datetime_mock()
    flexmock(module).open = lambda filename, mode: stdout
    flexmock(module).os = flexmock().should_receive('devnull').mock

    module.check_archives(
        verbose=False,
        repository='repo',
    )


def test_check_archives_with_verbose_should_call_attic_with_verbose_parameters():
    insert_subprocess_check_call_mock(
        ('attic', 'check', 'repo', '--verbose'),
        stdout=None,
    )
    insert_platform_mock()
    insert_datetime_mock()

    module.check_archives(
        verbose=True,
        repository='repo',
    )