mirror of
https://github.com/nvbn/thefuck.git
synced 2025-01-19 04:21:14 +00:00
ab91eaf73d
This reverts commit 28a5c1214b730e727b88648cf56dd36d51862ff6, reversing changes made to 3f0cbb9326895cf516eb9f4f0f834d23200dc21e.
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
from subprocess import PIPE
|
|
from mock import patch, Mock
|
|
import pytest
|
|
from thefuck.rules.no_command import match, get_new_command
|
|
from thefuck.main import Command
|
|
|
|
|
|
@pytest.fixture
|
|
def command_found():
|
|
return b'''No command 'aptget' found, did you mean:
|
|
Command 'apt-get' from package 'apt' (main)
|
|
aptget: command not found
|
|
'''
|
|
|
|
@pytest.fixture
|
|
def command_not_found():
|
|
return b'''No command 'vom' found, but there are 19 similar ones
|
|
vom: command not found
|
|
'''
|
|
|
|
|
|
@pytest.fixture
|
|
def bins_exists(request):
|
|
p = patch('thefuck.rules.no_command.which',
|
|
return_value=True)
|
|
p.start()
|
|
request.addfinalizer(p.stop)
|
|
|
|
|
|
@pytest.fixture
|
|
def settings():
|
|
class _Settings(object):
|
|
pass
|
|
return _Settings
|
|
|
|
|
|
@pytest.mark.usefixtures('bins_exists')
|
|
def test_match(command_found, command_not_found, settings):
|
|
with patch('thefuck.rules.no_command.Popen') as Popen:
|
|
Popen.return_value.stderr.read.return_value = command_found
|
|
assert match(Command('aptget install vim', '', ''), settings)
|
|
Popen.assert_called_once_with('/usr/lib/command-not-found aptget',
|
|
shell=True, stderr=PIPE)
|
|
Popen.return_value.stderr.read.return_value = command_not_found
|
|
assert not match(Command('ls', '', ''), settings)
|
|
|
|
with patch('thefuck.rules.no_command.Popen') as Popen:
|
|
Popen.return_value.stderr.read.return_value = command_found
|
|
assert match(Command('sudo aptget install vim', '', ''),
|
|
Mock(command_not_found='test'))
|
|
Popen.assert_called_once_with('test aptget',
|
|
shell=True, stderr=PIPE)
|
|
|
|
|
|
@pytest.mark.usefixtures('bins_exists')
|
|
def test_get_new_command(command_found):
|
|
with patch('thefuck.rules.no_command._get_output',
|
|
return_value=command_found.decode()):
|
|
assert get_new_command(Command('aptget install vim', '', ''), settings)\
|
|
== 'apt-get install vim'
|
|
assert get_new_command(Command('sudo aptget install vim', '', ''), settings) \
|
|
== 'sudo apt-get install vim'
|