thefuck/tests/rules/test_no_command.py

import pytest
from thefuck.rules.no_command import match, get_new_command
from tests.utils import Command


@pytest.fixture(autouse=True)
def get_all_executables(mocker):
    mocker.patch('thefuck.rules.no_command.get_all_executables',
                 return_value=['vim', 'apt-get', 'fsck'])


@pytest.mark.usefixtures('no_memoize')
def test_match():
    assert match(Command(stderr='vom: not found', script='vom file.py'), None)
    assert match(Command(stderr='fucck: not found', script='fucck'), None)
    assert not match(Command(stderr='qweqwe: not found', script='qweqwe'), None)
    assert not match(Command(stderr='some text', script='vom file.py'), None)


@pytest.mark.usefixtures('no_memoize')
def test_get_new_command():
    assert get_new_command(
        Command(stderr='vom: not found',
                script='vom file.py'),
        None) == ['vim file.py']
    assert get_new_command(
        Command(stderr='fucck: not found',
                script='fucck'),
        Command) == ['fsck']