MirroredRepos/thefuck
1
0
Fork 0
mirror of https://github.com/nvbn/thefuck.git synced 2024-10-05 18:31:10 +01:00
Code Issues Releases Wiki Activity

#1039: Remove all leading $ not just one

Browse Source
This commit is contained in:
theslimshaney 2020-02-08 04:37:19 -08:00 committed by Pablo Santiago Blum de Aguiar
parent 7f97818663
commit 9201ce79cf
2 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/rules/test_remove_shell_prompt_literal.py
View File

@ -8,7 +8,15 @@ def output():
return "$: command not found"
@pytest.mark.parametrize("script", ["$ cd newdir", " $ cd newdir"])
@pytest.mark.parametrize(
"script",
[
"$ cd newdir",
" $ cd newdir",
"$ $ cd newdir"
" $ $ cd newdir",
],
)
def test_match(script, output):
assert match(Command(script, output))
@ -31,7 +39,9 @@ def test_not_match(command):
"script, new_command",
[
("$ cd newdir", "cd newdir"),
("$ $ cd newdir", "cd newdir"),
("$ python3 -m virtualenv env", "python3 -m virtualenv env"),
(" $ $ $ python3 -m virtualenv env", "python3 -m virtualenv env"),
],
)
def test_get_new_command(script, new_command, output):

5
thefuck/rules/remove_shell_prompt_literal.py
View File

@ -1,4 +1,5 @@
"""Fixes error for commands containing the shell prompt symbol '$'.
"""Fixes error for commands containing one or more occurrences of the shell
prompt symbol '$'.
This usually happens when commands are copied from documentations
including them in their code blocks.
@ -19,4 +20,4 @@ def match(command):
def get_new_command(command):
return command.script.replace("$", "", 1).strip()
return command.script.lstrip("$ ")