mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-06 02:41:10 +01:00
122 lines
4.6 KiB
Python
122 lines
4.6 KiB
Python
# -*- encoding: utf-8 -*-
|
|
|
|
from mock import Mock
|
|
import pytest
|
|
from itertools import islice
|
|
from thefuck import ui
|
|
from thefuck.types import CorrectedCommand, SortedCorrectedCommandsSequence
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_getch(monkeypatch):
|
|
def patch(vals):
|
|
def getch():
|
|
for val in vals:
|
|
if val == KeyboardInterrupt:
|
|
raise val
|
|
else:
|
|
yield val
|
|
|
|
getch_gen = getch()
|
|
monkeypatch.setattr('thefuck.ui.getch', lambda: next(getch_gen))
|
|
|
|
return patch
|
|
|
|
|
|
def test_read_actions(patch_getch):
|
|
patch_getch([ # Enter:
|
|
'\n',
|
|
# Enter:
|
|
'\r',
|
|
# Ignored:
|
|
'x', 'y',
|
|
# Up:
|
|
'\x1b', '[', 'A',
|
|
# Down:
|
|
'\x1b', '[', 'B',
|
|
# Ctrl+C:
|
|
KeyboardInterrupt], )
|
|
assert list(islice(ui.read_actions(), 5)) \
|
|
== [ui.SELECT, ui.SELECT, ui.PREVIOUS, ui.NEXT, ui.ABORT]
|
|
|
|
|
|
def test_command_selector():
|
|
selector = ui.CommandSelector([1, 2, 3])
|
|
assert selector.value == 1
|
|
changes = []
|
|
selector.on_change(changes.append)
|
|
selector.next()
|
|
assert selector.value == 2
|
|
selector.next()
|
|
assert selector.value == 3
|
|
selector.next()
|
|
assert selector.value == 1
|
|
selector.previous()
|
|
assert selector.value == 3
|
|
assert changes == [1, 2, 3, 1, 3]
|
|
|
|
|
|
class TestSelectCommand(object):
|
|
@pytest.fixture
|
|
def commands_with_side_effect(self, settings):
|
|
return SortedCorrectedCommandsSequence(
|
|
iter([CorrectedCommand('ls', lambda *_: None, 100),
|
|
CorrectedCommand('cd', lambda *_: None, 100)]),
|
|
settings)
|
|
|
|
@pytest.fixture
|
|
def commands(self, settings):
|
|
return SortedCorrectedCommandsSequence(
|
|
iter([CorrectedCommand('ls', None, 100),
|
|
CorrectedCommand('cd', None, 100)]),
|
|
settings)
|
|
|
|
def test_without_commands(self, capsys):
|
|
assert ui.select_command([], Mock(debug=False, no_color=True)) is None
|
|
assert capsys.readouterr() == ('', 'No fucks given\n')
|
|
|
|
def test_without_confirmation(self, capsys, commands):
|
|
assert ui.select_command(commands,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=False)) == commands[0]
|
|
assert capsys.readouterr() == ('', 'ls\n')
|
|
|
|
def test_without_confirmation_with_side_effects(self, capsys,
|
|
commands_with_side_effect):
|
|
assert ui.select_command(commands_with_side_effect,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=False)) \
|
|
== commands_with_side_effect[0]
|
|
assert capsys.readouterr() == ('', 'ls (+side effect)\n')
|
|
|
|
def test_with_confirmation(self, capsys, patch_getch, commands):
|
|
patch_getch(['\n'])
|
|
assert ui.select_command(commands,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=True)) == commands[0]
|
|
assert capsys.readouterr() == ('', u'\x1b[1K\rls [enter/↑/↓/ctrl+c]\n')
|
|
|
|
def test_with_confirmation_abort(self, capsys, patch_getch, commands):
|
|
patch_getch([KeyboardInterrupt])
|
|
assert ui.select_command(commands,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=True)) is None
|
|
assert capsys.readouterr() == ('', u'\x1b[1K\rls [enter/↑/↓/ctrl+c]\nAborted\n')
|
|
|
|
def test_with_confirmation_with_side_effct(self, capsys, patch_getch,
|
|
commands_with_side_effect):
|
|
patch_getch(['\n'])
|
|
assert ui.select_command(commands_with_side_effect,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=True))\
|
|
== commands_with_side_effect[0]
|
|
assert capsys.readouterr() == ('', u'\x1b[1K\rls (+side effect) [enter/↑/↓/ctrl+c]\n')
|
|
|
|
def test_with_confirmation_select_second(self, capsys, patch_getch, commands):
|
|
patch_getch(['\x1b', '[', 'B', '\n'])
|
|
assert ui.select_command(commands,
|
|
Mock(debug=False, no_color=True,
|
|
require_confirmation=True)) == commands[1]
|
|
assert capsys.readouterr() == (
|
|
'', u'\x1b[1K\rls [enter/↑/↓/ctrl+c]\x1b[1K\rcd [enter/↑/↓/ctrl+c]\n')
|