diff --git a/tests/rules/test_git_remote_delete.py b/tests/rules/test_git_remote_delete.py index 660f0c8b..b7a1e4f2 100644 --- a/tests/rules/test_git_remote_delete.py +++ b/tests/rules/test_git_remote_delete.py @@ -16,6 +16,9 @@ def test_not_match(command): assert not match(command) -def test_get_new_command(): - new_command = get_new_command(Command('git remote delete foo', '')) - assert new_command == 'git remote remove foo' +@pytest.mark.parametrize('command, new_command', [ + (Command('git remote delete foo', ''), 'git remote remove foo'), + (Command('git remote delete delete', ''), 'git remote remove delete'), +]) +def test_get_new_command(command, new_command): + assert get_new_command(command) == new_command diff --git a/thefuck/rules/git_remote_delete.py b/thefuck/rules/git_remote_delete.py index ce316da1..f98a70dd 100644 --- a/thefuck/rules/git_remote_delete.py +++ b/thefuck/rules/git_remote_delete.py @@ -1,12 +1,13 @@ -from thefuck.utils import replace_argument +import re + from thefuck.specific.git import git_support @git_support def match(command): - return "git remote delete" in command.script + return "remote delete" in command.script @git_support def get_new_command(command): - return replace_argument(command.script, "delete", "remove") + return re.sub(r"delete", "remove", command.script, 1)