borgmatic/atticmatic/tests/integration/test_command.py

import os
import sys

from flexmock import flexmock
import pytest

from atticmatic import command as module


COMMAND_NAME = 'foomatic'


def test_parse_arguments_with_no_arguments_uses_defaults():
    flexmock(os.path).should_receive('exists').and_return(True)

    parser = module.parse_arguments(COMMAND_NAME)

    assert parser.config_filename == module.DEFAULT_CONFIG_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.excludes_filename == module.DEFAULT_EXCLUDES_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.verbosity == None


def test_parse_arguments_with_filename_arguments_overrides_defaults():
    flexmock(os.path).should_receive('exists').and_return(True)

    parser = module.parse_arguments(COMMAND_NAME, '--config', 'myconfig', '--excludes', 'myexcludes')

    assert parser.config_filename == 'myconfig'
    assert parser.excludes_filename == 'myexcludes'
    assert parser.verbosity == None


def test_parse_arguments_with_missing_default_excludes_file_sets_filename_to_none():
    flexmock(os.path).should_receive('exists').and_return(False)

    parser = module.parse_arguments(COMMAND_NAME)

    assert parser.config_filename == module.DEFAULT_CONFIG_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.excludes_filename == None
    assert parser.verbosity == None


def test_parse_arguments_with_missing_overridden_excludes_file_retains_filename():
    flexmock(os.path).should_receive('exists').and_return(False)

    parser = module.parse_arguments(COMMAND_NAME, '--excludes', 'myexcludes')

    assert parser.config_filename == module.DEFAULT_CONFIG_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.excludes_filename == 'myexcludes'
    assert parser.verbosity == None


def test_parse_arguments_with_verbosity_flag_overrides_default():
    flexmock(os.path).should_receive('exists').and_return(True)

    parser = module.parse_arguments(COMMAND_NAME, '--verbosity', '1')

    assert parser.config_filename == module.DEFAULT_CONFIG_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.excludes_filename == module.DEFAULT_EXCLUDES_FILENAME_PATTERN.format(COMMAND_NAME)
    assert parser.verbosity == 1


def test_parse_arguments_with_invalid_arguments_exits():
    flexmock(os.path).should_receive('exists').and_return(True)
    original_stderr = sys.stderr
    sys.stderr = sys.stdout

    try:
        with pytest.raises(SystemExit):
            module.parse_arguments(COMMAND_NAME, '--posix-me-harder')
    finally:
        sys.stderr = original_stderr