mirror of
https://github.com/nvbn/thefuck.git
synced 2024-10-06 02:41:10 +01:00
27 lines
983 B
Python
27 lines
983 B
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]
|