def test_get_command(): with patch('thefuck.main.Popen') as Popen, \ patch('thefuck.main.os.environ', new_callable=lambda: {}), \ patch('thefuck.main.wait_output', return_value=True): Popen.return_value.stdout.read.return_value = b'stdout' Popen.return_value.stderr.read.return_value = b'stderr' assert main.get_command(Mock(), Mock(), ['thefuck', 'apt-get', 'search', 'vim']) \ == Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE, env={'LANG': 'C'}) assert main.get_command(Mock(), Mock(), ['']) is None # When command is `f**k`: assert main.get_command( Mock(), Mock(last_script='ls', last_fixed_script='ls -la'), ['thefuck', 'f**k']).script == 'ls -la' # When command equals to last command: assert main.get_command( Mock(), Mock(last_script='ls', last_fixed_script='ls -la'), ['thefuck', 'ls']).script == 'ls -la' # When last command is `f**k` and no last fixed script: assert main.get_command( Mock(), Mock(last_script='ls', last_fixed_script=''), ['thefuck', 'ls']).script == 'ls'
def test_get_command(): with patch('thefuck.main.Popen') as Popen, \ patch('thefuck.main.os.environ', new_callable=lambda: {}), \ patch('thefuck.main.wait_output', return_value=True): Popen.return_value.stdout.read.return_value = b'stdout' Popen.return_value.stderr.read.return_value = b'stderr' assert main.get_command(Mock(), ['thefuck', 'apt-get', 'search', 'vim']) \ == Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE, env={'LANG': 'C'}) assert main.get_command(Mock(), ['']) is None
def test_get_command(): with patch('thefuck.main.Popen') as Popen, \ patch('thefuck.main.os.environ', new_callable=lambda: {}), \ patch('thefuck.main.wait_output', return_value=True): Popen.return_value.stdout.read.return_value = b'stdout' Popen.return_value.stderr.read.return_value = b'stderr' assert main.get_command(Mock(), ['thefuck', 'apt-get', 'search', 'vim']) \ == types.Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE, env={'LANG': 'C'}) assert main.get_command(Mock(), ['']) is None
def test_get_command_calls(self, Popen): assert main.get_command(Mock(), ['thefuck', 'apt-get', 'search', 'vim']) \ == Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE, env={'LANG': 'C'})
def test_get_command_calls(self, Popen): assert main.get_command(Mock(env={}), ['thefuck', 'apt-get', 'search', 'vim']) \ == Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE, env={})
def test_get_command(): with patch('thefuck.main.Popen') as Popen: Popen.return_value.stdout.read.return_value = b'stdout' Popen.return_value.stderr.read.return_value = b'stderr' assert main.get_command(['thefuck', 'apt-get', 'search', 'vim']) \ == main.Command('apt-get search vim', 'stdout', 'stderr') Popen.assert_called_once_with('apt-get search vim', shell=True, stdout=PIPE, stderr=PIPE)
def test_get_command_script(self, args, result): if result: assert main.get_command(Mock(), args).script == result else: assert main.get_command(Mock(), args) is None
def test_get_command_script(self, args, result): if result: assert main.get_command(Mock(env={}), args).script == result else: assert main.get_command(Mock(env={}), args) is None