x/cloudbase-init
Code Issues Proposed changes

Merge "Adds EC2 userdata execution support"

Browse Source
This commit is contained in:
Jenkins 2017-05-05 21:55:15 +00:00 committed by Gerrit Code Review
commit a80aa83360
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cloudbaseinit/metadata/services/ec2service.py
View File

@ -55,6 +55,10 @@ class EC2Service(base.BaseHTTPMetadataService):
return self._get_cache_data('%s/meta-data/instance-id' %
self._metadata_version, decode=True)
def get_user_data(self):
return self._get_cache_data('%s/user-data' %
self._metadata_version)
def get_public_keys(self):
ssh_keys = []

8
cloudbaseinit/tests/metadata/services/test_ec2service.py
View File

@ -88,3 +88,11 @@ class EC2ServiceTest(unittest.TestCase):
'idx': 'key'}, decode=True)]
self.assertEqual(expected, mock_get_cache_data.call_args_list)
self.assertEqual(['fake key'], response)
@mock.patch('cloudbaseinit.metadata.services.ec2service.EC2Service'
'._get_cache_data')
def test_get_user_data(self, mock_get_cache_data):
response = self._service.get_user_data()
path = '%s/user-data' % self._service._metadata_version
mock_get_cache_data.assert_called_once_with(path)
self.assertEqual(mock_get_cache_data.return_value, response)