def test_run_command_with_check_output_function(
self, mock_subprocess):
check_output = Mock()
check_output.__name__ = 'check_output'
mock_subprocess.check_output = check_output
run_command(self.command)
check_output.assert_called_once_with(
self.command, stderr=mock_subprocess.STDOUT, shell=True)
def test_run_command_with_check_call_function(
self, mock_subprocess):
mock_subprocess.check_call.__name__ = 'check_call'
if 'check_output' in dir(mock_subprocess):
mock_subprocess.check_output = mock_subprocess.check_call
run_command(self.command)
mock_subprocess.check_call.assert_called_once_with(
self.command, stderr=mock_subprocess.STDOUT, shell=True)
def test_run_command_in_check_output_occours_exception(
self, mock_subprocess):
check_output = Mock()
mock_subprocess.CalledProcessError = subprocess.CalledProcessError
check_output.side_effect = subprocess.CalledProcessError(1,
self.command)
result = run_command(self.command)
self.assertEqual(result[0], 1)
self.assertIsNone(result[1])
def test_run_command_check_call_return_tuple(
self, mock_subprocess):
mock_subprocess.check_call.__name__ = 'check_call'
mock_subprocess.check_call.return_value = 0
if 'check_output' in dir(mock_subprocess):
mock_subprocess.check_output = mock_subprocess.check_call
result = run_command(self.command)
self.assertEqual(result[0], 0)
self.assertIsNone(result[1])
def test_run_command_check_output_return_tuple(
self, mock_subprocess):
check_output = Mock()
check_output.__name__ = 'check_output'
check_output.return_value = 'command output'
mock_subprocess.check_output = check_output
result = run_command(self.command)
self.assertEqual(result[0], 0)
self.assertEqual(result[1], check_output.return_value)
def test_run_command_in_check_output_occours_exception(
self, mock_subprocess):
mock_subprocess.CalledProcessError = subprocess.CalledProcessError
mock_subprocess.check_call.side_effect = subprocess.CalledProcessError(
1, self.command)
if 'check_output' in dir(mock_subprocess):
mock_subprocess.check_output = mock_subprocess.check_call
result = run_command(self.command)
self.assertEqual(result[0], 1)
self.assertIsNone(result[1])
