mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-05 18:31:10 +01:00
30 lines
953 B
Python
30 lines
953 B
Python
import pytest
|
|
from thefuck.rules.whois import match, get_new_command
|
|
from thefuck.types import Command
|
|
|
|
|
|
@pytest.mark.parametrize('command', [
|
|
Command('whois https://en.wikipedia.org/wiki/Main_Page', ''),
|
|
Command('whois https://en.wikipedia.org/', ''),
|
|
Command('whois meta.unix.stackexchange.com', '')])
|
|
def test_match(command):
|
|
assert match(command)
|
|
|
|
|
|
def test_not_match():
|
|
assert not match(Command('whois', ''))
|
|
|
|
|
|
# `whois com` actually makes sense
|
|
@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 meta.unix.stackexchange.com', ''),
|
|
['whois unix.stackexchange.com',
|
|
'whois stackexchange.com',
|
|
'whois com'])])
|
|
def test_get_new_command(command, new_command):
|
|
assert get_new_command(command) == new_command
|