def open(self):
execute_plugin([json.dumps(self._response),
self._region_name,
'StartSession',
self._profile_name,
json.dumps(self._session_parameters),
self._ssm.meta.endpoint_url])
def test_execute_plugin_args(mocker):
m = mocker.patch("aws_gate.utils.execute", return_value="output")
execute_plugin(["--version"], capture_output=True)
assert m.called
assert "['--version'], capture_output=True" in str(m.call_args)
def test_execute_plugin_args(mocker):
m = mocker.patch("aws_gate.utils.execute", return_value="output")
execute_plugin(["--version"], stdout=PIPE, stderr=PIPE)
assert m.called
assert "['--version']" in str(m.call_args[0])
assert m.call_args[1]["stdout"] == PIPE
assert m.call_args[1]["stderr"] == PIPE
def wrapper(wrapped_function, instance, args, kwargs): # pylint: disable=unused-argument
version = execute_plugin(['--version'], capture_output=True)
logger.debug(
'session-manager-plugin version: %s (required version: %s)',
version, required_version)
if version and int(version.replace('.', '')) < int(
required_version.replace('.', '')):
raise ValueError('Invalid plugin version: {}'.format(version))
return wrapped_function(*args, **kwargs)
def wrapper(wrapped_function, instance, args, kwargs): # pylint: disable=unused-argument
version = execute_plugin(["--version"], stdout=PIPE, stderr=PIPE)
logger.debug(
"session-manager-plugin version: %s (required version: %s)",
version,
required_version,
)
if version and parse_version(version) < parse_version(
required_version):
raise ValueError("Invalid plugin version: {}".format(version))
return wrapped_function(*args, **kwargs)
def test_execute_plugin(mocker):
mocker.patch("aws_gate.utils.execute", return_value="output")
output = execute_plugin(["--version"], stdout=PIPE, stderrr=PIPE)
assert output == "output"
def test_execute_plugin(mocker):
mocker.patch("aws_gate.utils.execute", return_value="output")
output = execute_plugin(["--version"], capture_output=True)
assert output == "output"
def test_execute_plugin_args(self):
with patch('aws_gate.utils.execute', return_value='output') as m:
execute_plugin(['--version'], capture_output=True)
self.assertTrue(m.called)
self.assertIn('[\'--version\'], capture_output=True', str(m.call_args))
def test_execute_plugin(self):
with patch('aws_gate.utils.execute', return_value='output'):
output = execute_plugin(['--version'], capture_output=True)
self.assertEqual(output, 'output')
