Alessio Sergi
ebf1ea88f5
The pathlib backport module is no longer maintained. The development has moved to the pathlib2 module instead. Quoting from the pathlib's README: "Attention: this backport module isn't maintained anymore. If you want to report issues or contribute patches, please consider the pathlib2 project instead."
134 lines
5.5 KiB
Python
134 lines
5.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
from subprocess import PIPE
|
|
from mock import Mock
|
|
try:
|
|
from pathlib import Path
|
|
except ImportError:
|
|
from pathlib2 import Path
|
|
import pytest
|
|
from tests.utils import CorrectedCommand, Rule, Command
|
|
from thefuck import const
|
|
from thefuck.exceptions import EmptyCommand
|
|
|
|
|
|
class TestCorrectedCommand(object):
|
|
|
|
def test_equality(self):
|
|
assert CorrectedCommand('ls', None, 100) == \
|
|
CorrectedCommand('ls', None, 200)
|
|
assert CorrectedCommand('ls', None, 100) != \
|
|
CorrectedCommand('ls', lambda *_: _, 100)
|
|
|
|
def test_hashable(self):
|
|
assert {CorrectedCommand('ls', None, 100),
|
|
CorrectedCommand('ls', None, 200)} == {CorrectedCommand('ls')}
|
|
|
|
def test_representable(self):
|
|
assert '{}'.format(CorrectedCommand('ls', None, 100)) == \
|
|
'CorrectedCommand(script=ls, side_effect=None, priority=100)'
|
|
assert u'{}'.format(CorrectedCommand(u'echo café', None, 100)) == \
|
|
u'CorrectedCommand(script=echo café, side_effect=None, priority=100)'
|
|
|
|
|
|
class TestRule(object):
|
|
def test_from_path(self, mocker):
|
|
match = object()
|
|
get_new_command = object()
|
|
load_source = mocker.patch(
|
|
'thefuck.types.load_source',
|
|
return_value=Mock(match=match,
|
|
get_new_command=get_new_command,
|
|
enabled_by_default=True,
|
|
priority=900,
|
|
requires_output=True))
|
|
rule_path = os.path.join(os.sep, 'rules', 'bash.py')
|
|
assert Rule.from_path(Path(rule_path)) \
|
|
== Rule('bash', match, get_new_command, priority=900)
|
|
load_source.assert_called_once_with('bash', rule_path)
|
|
|
|
@pytest.mark.parametrize('rules, exclude_rules, rule, is_enabled', [
|
|
(const.DEFAULT_RULES, [], Rule('git', enabled_by_default=True), True),
|
|
(const.DEFAULT_RULES, [], Rule('git', enabled_by_default=False), False),
|
|
([], [], Rule('git', enabled_by_default=False), False),
|
|
([], [], Rule('git', enabled_by_default=True), False),
|
|
(const.DEFAULT_RULES + ['git'], [], Rule('git', enabled_by_default=False), True),
|
|
(['git'], [], Rule('git', enabled_by_default=False), True),
|
|
(const.DEFAULT_RULES, ['git'], Rule('git', enabled_by_default=True), False),
|
|
(const.DEFAULT_RULES, ['git'], Rule('git', enabled_by_default=False), False),
|
|
([], ['git'], Rule('git', enabled_by_default=True), False),
|
|
([], ['git'], Rule('git', enabled_by_default=False), False)])
|
|
def test_is_enabled(self, settings, rules, exclude_rules, rule, is_enabled):
|
|
settings.update(rules=rules,
|
|
exclude_rules=exclude_rules)
|
|
assert rule.is_enabled == is_enabled
|
|
|
|
def test_isnt_match(self):
|
|
assert not Rule('', lambda _: False).is_match(
|
|
Command('ls'))
|
|
|
|
def test_is_match(self):
|
|
rule = Rule('', lambda x: x.script == 'cd ..')
|
|
assert rule.is_match(Command('cd ..'))
|
|
|
|
@pytest.mark.usefixtures('no_colors')
|
|
def test_isnt_match_when_rule_failed(self, capsys):
|
|
rule = Rule('test', Mock(side_effect=OSError('Denied')),
|
|
requires_output=False)
|
|
assert not rule.is_match(Command('ls'))
|
|
assert capsys.readouterr()[1].split('\n')[0] == '[WARN] Rule test:'
|
|
|
|
def test_get_corrected_commands_with_rule_returns_list(self):
|
|
rule = Rule(get_new_command=lambda x: [x.script + '!', x.script + '@'],
|
|
priority=100)
|
|
assert list(rule.get_corrected_commands(Command(script='test'))) \
|
|
== [CorrectedCommand(script='test!', priority=100),
|
|
CorrectedCommand(script='test@', priority=200)]
|
|
|
|
def test_get_corrected_commands_with_rule_returns_command(self):
|
|
rule = Rule(get_new_command=lambda x: x.script + '!',
|
|
priority=100)
|
|
assert list(rule.get_corrected_commands(Command(script='test'))) \
|
|
== [CorrectedCommand(script='test!', priority=100)]
|
|
|
|
|
|
class TestCommand(object):
|
|
@pytest.fixture(autouse=True)
|
|
def Popen(self, monkeypatch):
|
|
Popen = Mock()
|
|
Popen.return_value.stdout.read.return_value = b'stdout'
|
|
Popen.return_value.stderr.read.return_value = b'stderr'
|
|
monkeypatch.setattr('thefuck.types.Popen', Popen)
|
|
return Popen
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def prepare(self, monkeypatch):
|
|
monkeypatch.setattr('thefuck.types.os.environ', {})
|
|
monkeypatch.setattr('thefuck.types.Command._wait_output',
|
|
staticmethod(lambda *_: True))
|
|
|
|
def test_from_script_calls(self, Popen, settings):
|
|
settings.env = {}
|
|
assert Command.from_raw_script(
|
|
['apt-get', 'search', 'vim']) == Command(
|
|
'apt-get search vim', 'stdout', 'stderr')
|
|
Popen.assert_called_once_with('apt-get search vim',
|
|
shell=True,
|
|
stdin=PIPE,
|
|
stdout=PIPE,
|
|
stderr=PIPE,
|
|
env={})
|
|
|
|
@pytest.mark.parametrize('script, result', [
|
|
([''], None),
|
|
(['', ''], None),
|
|
(['ls', '-la'], 'ls -la'),
|
|
(['ls'], 'ls')])
|
|
def test_from_script(self, script, result):
|
|
if result:
|
|
assert Command.from_raw_script(script).script == result
|
|
else:
|
|
with pytest.raises(EmptyCommand):
|
|
Command.from_raw_script(script)
|