mirror of
https://github.com/nvbn/thefuck.git
synced 2025-01-18 20:11:17 +00:00
0f4a523dc4
Using the `force_command` argument will run into issues as the `_get_raw_command` method simply returns the value of `force_command` (which is a string) while it should actually return a list. Fix #1240
27 lines
975 B
Python
27 lines
975 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]
|