mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-05 18:31:10 +01:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import pytest
|
|
from mock import Mock
|
|
from thefuck.utils import sudo_support, wrap_settings, memoize
|
|
from thefuck.types import Settings
|
|
from tests.utils import Command
|
|
|
|
|
|
@pytest.mark.parametrize('override, old, new', [
|
|
({'key': 'val'}, {}, {'key': 'val'}),
|
|
({'key': 'new-val'}, {'key': 'val'}, {'key': 'new-val'})])
|
|
def test_wrap_settings(override, old, new):
|
|
fn = lambda _, settings: settings
|
|
assert wrap_settings(override)(fn)(None, Settings(old)) == new
|
|
|
|
|
|
@pytest.mark.parametrize('return_value, command, called, result', [
|
|
('ls -lah', 'sudo ls', 'ls', 'sudo ls -lah'),
|
|
('ls -lah', 'ls', 'ls', 'ls -lah'),
|
|
(True, 'sudo ls', 'ls', True),
|
|
(True, 'ls', 'ls', True),
|
|
(False, 'sudo ls', 'ls', False),
|
|
(False, 'ls', 'ls', False)])
|
|
def test_sudo_support(return_value, command, called, result):
|
|
fn = Mock(return_value=return_value, __name__='')
|
|
assert sudo_support(fn)(Command(command), None) == result
|
|
fn.assert_called_once_with(Command(called), None)
|
|
|
|
|
|
def test_memoize():
|
|
fn = Mock(__name__='fn')
|
|
memoized = memoize(fn)
|
|
memoized()
|
|
memoized()
|
|
fn.assert_called_once_with()
|