def test_execute_command_not_found(mocker):
mocker.patch(
"aws_gate.utils.subprocess.run",
side_effect=OSError(errno.ENOENT, os.strerror(errno.ENOENT)),
)
with pytest.raises(ValueError):
execute("/usr/bin/ls", ["-l"])
def test_execute_environment(mocker):
mock_output = mocker.MagicMock(stdout=b"output")
m = mocker.patch("aws_gate.utils.subprocess.run", return_value=mock_output)
execute("ls", ["-l"])
assert m.call_args_list[0][1]["env"] is not None
assert DEFAULT_GATE_BIN_PATH in m.call_args_list[0][1]["env"]["PATH"]
def test_execute_command_exited_with_nonzero_rc(mocker):
mock = mocker.patch(
"aws_gate.utils.subprocess.run",
side_effect=subprocess.CalledProcessError(returncode=1, cmd="error"),
)
execute("/usr/bin/ls", ["-l"])
assert mock.called
def open(self):
self._ssh_cmd = self._build_ssh_command()
return execute(self._ssh_cmd[0], self._ssh_cmd[1:])
def test_execute(mocker, cmd, args):
mock_output = mocker.MagicMock(stdout=b"output")
mocker.patch("aws_gate.utils.subprocess.run", return_value=mock_output)
assert execute(cmd, args) == "output"
def _check_plugin_version(path=PLUGIN_INSTALL_PATH):
return execute(path, ["--version"], capture_output=True)
def test_execute_command_not_found(self):
with patch('aws_gate.utils.subprocess.run',
side_effect=OSError(errno.ENOENT, os.strerror(errno.ENOENT))):
with self.assertRaises(ValueError):
execute('/usr/bin/ls', ['-l'])
def test_execute_command_exited_with_nonzero_rc(self):
with patch('aws_gate.utils.subprocess.run',
side_effect=subprocess.CalledProcessError(returncode=1, cmd='error')) as mock:
execute('/usr/bin/ls', ['-l'])
self.assertTrue(mock.called)
def test_execute(self, cmd, args):
mock_output = MagicMock(stdout=b'output')
with patch('aws_gate.utils.subprocess.run', return_value=mock_output):
self.assertEqual(execute(cmd, args), 'output')
def _check_plugin_version(path=PLUGIN_INSTALL_PATH):
return execute(path, ["--version"], stdout=PIPE, stderr=PIPE)
