|
@@ -1,49 +1,14 @@
|
|
|
from collections import OrderedDict
|
|
|
-try:
|
|
|
- # Python 2
|
|
|
- import __builtin__ as builtins
|
|
|
-except ImportError:
|
|
|
- # Python 3
|
|
|
- import builtins
|
|
|
|
|
|
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(builtins).should_receive('print')
|
|
|
-
|
|
|
- flexmock(module).subprocess = subprocess
|
|
|
- return subprocess
|
|
|
-
|
|
|
-
|
|
|
-def insert_subprocess_check_call_mock(call_command, **kwargs):
|
|
|
+def insert_subprocess_mock(check_call_command, **kwargs):
|
|
|
subprocess = flexmock()
|
|
|
-
|
|
|
- subprocess.should_receive('check_call').with_args(
|
|
|
- call_command,
|
|
|
- **kwargs
|
|
|
- ).once()
|
|
|
-
|
|
|
+ subprocess.should_receive('check_call').with_args(check_call_command, **kwargs).once()
|
|
|
flexmock(module).subprocess = subprocess
|
|
|
- return subprocess
|
|
|
|
|
|
|
|
|
def insert_platform_mock():
|
|
@@ -57,7 +22,7 @@ def insert_datetime_mock():
|
|
|
|
|
|
|
|
|
def test_create_archive_should_call_attic_with_parameters():
|
|
|
- insert_subprocess_check_output_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
('attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar'),
|
|
|
)
|
|
|
insert_platform_mock()
|
|
@@ -72,7 +37,7 @@ def test_create_archive_should_call_attic_with_parameters():
|
|
|
|
|
|
|
|
|
def test_create_archive_with_verbose_should_call_attic_with_verbose_parameters():
|
|
|
- insert_subprocess_check_output_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
(
|
|
|
'attic', 'create', '--exclude-from', 'excludes', 'repo::host-now', 'foo', 'bar',
|
|
|
'--verbose', '--stats',
|
|
@@ -88,39 +53,6 @@ def test_create_archive_with_verbose_should_call_attic_with_verbose_parameters()
|
|
|
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'),
|
|
@@ -148,7 +80,7 @@ def test_prune_archives_should_call_attic_with_parameters():
|
|
|
flexmock(module).should_receive('make_prune_flags').with_args(retention_config).and_return(
|
|
|
BASE_PRUNE_FLAGS,
|
|
|
)
|
|
|
- insert_subprocess_check_call_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
(
|
|
|
'attic', 'prune', 'repo', '--keep-daily', '1', '--keep-weekly', '2', '--keep-monthly',
|
|
|
'3',
|
|
@@ -167,7 +99,7 @@ def test_prune_archives_with_verbose_should_call_attic_with_verbose_parameters()
|
|
|
flexmock(module).should_receive('make_prune_flags').with_args(retention_config).and_return(
|
|
|
BASE_PRUNE_FLAGS,
|
|
|
)
|
|
|
- insert_subprocess_check_call_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
(
|
|
|
'attic', 'prune', 'repo', '--keep-daily', '1', '--keep-weekly', '2', '--keep-monthly',
|
|
|
'3', '--verbose',
|
|
@@ -183,7 +115,7 @@ def test_prune_archives_with_verbose_should_call_attic_with_verbose_parameters()
|
|
|
|
|
|
def test_check_archives_should_call_attic_with_parameters():
|
|
|
stdout = flexmock()
|
|
|
- insert_subprocess_check_call_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
('attic', 'check', 'repo'),
|
|
|
stdout=stdout,
|
|
|
)
|
|
@@ -199,7 +131,7 @@ def test_check_archives_should_call_attic_with_parameters():
|
|
|
|
|
|
|
|
|
def test_check_archives_with_verbose_should_call_attic_with_verbose_parameters():
|
|
|
- insert_subprocess_check_call_mock(
|
|
|
+ insert_subprocess_mock(
|
|
|
('attic', 'check', 'repo', '--verbose'),
|
|
|
stdout=None,
|
|
|
)
|