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')