1
0
mirror of https://github.com/nvbn/thefuck.git synced 2025-02-21 04:18:55 +00:00

#609: Use replace_command in yarn_command_not_found

This commit is contained in:
Vladimir Iakovlev 2017-03-06 17:31:57 +01:00
parent df5428c5e4
commit 02f3250d39
2 changed files with 3 additions and 5 deletions

View File

@ -108,4 +108,4 @@ def test_not_match(command):
@pytest.mark.parametrize('command, result', [
(Command('yarn whyy webpack', stderr=stderr('whyy')), 'yarn why webpack')])
def test_get_new_command(command, result):
assert get_new_command(command) == result
assert get_new_command(command)[0] == result

View File

@ -1,6 +1,6 @@
import re
from subprocess import Popen, PIPE
from thefuck.utils import for_app, eager, get_closest
from thefuck.utils import for_app, eager, replace_command
regex = re.compile(r'error Command "(.*)" not found.')
@ -28,6 +28,4 @@ def _get_all_tasks():
def get_new_command(command):
misspelled_task = regex.findall(command.stderr)[0]
tasks = _get_all_tasks()
fixed = get_closest(misspelled_task, tasks)
return command.script.replace(' {}'.format(misspelled_task),
' {}'.format(fixed))
return replace_command(command, misspelled_task, tasks)