23 lines
680 B
Python
23 lines
680 B
Python
import pytest
|
|
from thefuck.rules.yarn_alias import match, get_new_command
|
|
from tests.utils import Command
|
|
|
|
|
|
stderr_remove = 'error Did you mean `yarn remove`?'
|
|
|
|
stderr_list = 'error Did you mean `yarn list`?'
|
|
|
|
|
|
@pytest.mark.parametrize('command', [
|
|
Command(script='yarn rm', stderr=stderr_remove),
|
|
Command(script='yarn ls', stderr=stderr_list)])
|
|
def test_match(command):
|
|
assert match(command)
|
|
|
|
|
|
@pytest.mark.parametrize('command, new_command', [
|
|
(Command('yarn rm', stderr=stderr_remove), 'yarn remove'),
|
|
(Command('yarn ls', stderr=stderr_list), 'yarn list')])
|
|
def test_get_new_command(command, new_command):
|
|
assert get_new_command(command) == new_command
|