def test_sudo_support(): fn = Mock(return_value=True, __name__='') assert sudo_support(fn)(Command('sudo ls', 'out', 'err'), None) fn.assert_called_once_with(Command('ls', 'out', 'err'), None) fn.return_value = False assert not sudo_support(fn)(Command('sudo ls', 'out', 'err'), None) fn.return_value = 'pwd' assert sudo_support(fn)(Command('sudo ls', 'out', 'err'), None) == 'sudo pwd' assert sudo_support(fn)(Command('ls', 'out', 'err'), None) == 'pwd'
def test_sudo_support(): fn = Mock(return_value=True, __name__='') assert sudo_support(fn)(Command('sudo ls'), None) fn.assert_called_once_with(Command('ls'), None) fn.return_value = False assert not sudo_support(fn)(Command('sudo ls'), None) fn.return_value = 'pwd' assert sudo_support(fn)(Command('sudo ls'), None) == 'sudo pwd' assert sudo_support(fn)(Command('ls'), None) == 'pwd'
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)