1
0
mirror of https://github.com/nvbn/thefuck.git synced 2025-01-18 20:11:17 +00:00

Fix test cases

This commit is contained in:
dhilipsiva 2017-02-09 09:16:20 +05:30
parent 8a8ade1e6b
commit 309fe8f6ee
No known key found for this signature in database
GPG Key ID: D3A33A90ADCDC5BF
2 changed files with 9 additions and 19 deletions

View File

@ -1,17 +1,9 @@
import pytest
from thefuck.rules.hostscli import no_command, no_website, get_new_command, \
match
from thefuck.rules.hostscli import no_website, get_new_command, match
from tests.utils import Command
no_command = '''
Usage: hostscli [OPTIONS] COMMAND [ARGS]...
%s "invalid".
''' % no_command
no_website = '''
no_website_long = '''
%s:
No Domain list found for website: a_website_that_does_not_exist
@ -24,17 +16,16 @@ type `hostscli websites` to see a list of websites that you can block/unblock
@pytest.mark.parametrize('command', [
Command('hostscli invalid', stderr=no_command)])
Command(
'sudo hostscli block a_website_that_does_not_exist',
stderr=no_website_long)])
def test_match(command):
assert match(command)
@pytest.mark.parametrize('command, result', [
(Command(
'hostscli invalid', stderr=no_command), ['hostscli --help']),
(Command(
@pytest.mark.parametrize('command, result', [(
Command(
'sudo hostscli block a_website_that_does_not_exist',
stderr=no_website),
['hostscli websites'])])
stderr=no_website_long), ['hostscli websites'])])
def test_get_new_command(command, result):
assert get_new_command(command) == result

View File

@ -1,12 +1,11 @@
import re
from thefuck.utils import for_app, replace_command
from thefuck.utils import replace_command
no_command = "Error: No such command"
no_website = "hostscli.errors.WebsiteImportError"
@for_app("hostscli")
def match(command):
errors = [no_command, no_website]
for error in errors: