2015-05-16 14:36:27 +01:00
|
|
|
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/'),
|
2015-07-31 21:12:10 +01:00
|
|
|
Command(script='whois meta.unix.stackexchange.com')])
|
2015-05-16 14:36:27 +01:00
|
|
|
def test_match(command):
|
|
|
|
assert match(command, None)
|
|
|
|
|
|
|
|
|
2015-05-20 17:35:31 +01:00
|
|
|
def test_not_match():
|
|
|
|
assert not match(Command(script='whois'), None)
|
|
|
|
|
|
|
|
|
2015-07-31 21:12:10 +01:00
|
|
|
# `whois com` actually makes sense
|
2015-05-16 14:36:27 +01:00
|
|
|
@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'),
|
2015-07-31 21:12:10 +01:00
|
|
|
(Command('whois meta.unix.stackexchange.com'), ['whois unix.stackexchange.com',
|
|
|
|
'whois stackexchange.com',
|
|
|
|
'whois com'])])
|
2015-05-16 14:36:27 +01:00
|
|
|
def test_get_new_command(command, new_command):
|
|
|
|
assert get_new_command(command, None) == new_command
|