33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
import pytest
|
|
from mock import Mock
|
|
from thefuck.entrypoints.fix_command import _get_raw_command
|
|
|
|
|
|
class TestGetRawCommand(object):
|
|
def test_from_force_command_argument(self):
|
|
known_args = Mock(force_command='git brunch')
|
|
assert _get_raw_command(known_args) == ['git brunch']
|
|
|
|
def test_from_command_argument(self, os_environ):
|
|
os_environ['TF_HISTORY'] = None
|
|
known_args = Mock(
|
|
force_command=None,
|
|
command=['sl']
|
|
)
|
|
assert _get_raw_command(known_args) == ['sl']
|
|
|
|
@pytest.mark.parametrize(
|
|
'history, result',
|
|
[
|
|
('git br', 'git br'),
|
|
('git br\nfcuk', 'git br'),
|
|
('git br\nfcuk\nls', 'ls'),
|
|
('git br\nfcuk\nls\nfuk', 'ls')
|
|
]
|
|
)
|
|
def test_from_history(self, os_environ, history, result):
|
|
os_environ['TF_HISTORY'] = history
|
|
known_args = Mock(force_command=None,
|
|
command=None)
|
|
assert _get_raw_command(known_args) == [result]
|