mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-06 02:41:10 +01:00
44c06c483e
This avoids thefuck failing when there's no arguments. It fails with:
```
...
File "thefuck/rules/whois.py", line 26, in get_new_command
url = command.script.split()[1]
IndexError: list index out of range
```
Signed-off-by: Pablo Santiago Blum de Aguiar <scorphus@gmail.com>
24 lines
834 B
Python
24 lines
834 B
Python
import pytest
|
|
from thefuck.rules.whois import match, get_new_command
|
|
from tests.utils import Command
|
|
|
|
|
|
@pytest.mark.parametrize('command', [
|
|
Command(script='whois https://en.wikipedia.org/wiki/Main_Page'),
|
|
Command(script='whois https://en.wikipedia.org/'),
|
|
Command(script='whois en.wikipedia.org')])
|
|
def test_match(command):
|
|
assert match(command, None)
|
|
|
|
|
|
def test_not_match():
|
|
assert not match(Command(script='whois'), None)
|
|
|
|
|
|
@pytest.mark.parametrize('command, new_command', [
|
|
(Command('whois https://en.wikipedia.org/wiki/Main_Page'), 'whois en.wikipedia.org'),
|
|
(Command('whois https://en.wikipedia.org/'), 'whois en.wikipedia.org'),
|
|
(Command('whois en.wikipedia.org'), 'whois wikipedia.org')])
|
|
def test_get_new_command(command, new_command):
|
|
assert get_new_command(command, None) == new_command
|