e89bdd0c6a
This change works around a bug in CloudStack's EC2 api implementation. That is filed upstream at [1]. The fix is safe for openstack and EC2 use cases as well. EC2 and OpenStacks' EC2 metadata service both return a list with access to either of: <url_base>/latest/meta-data <url_base>/latest/meta-data/ Additionally, the responses explicitly contain a trailing '/' for items that have a child. The ec2_utils code then just re-uses the trailng / there. Thus, only the top level request for 'meta-data/' needs the explicit fix. This also changes test cases. Those test cases failed without fixing them. If ever this regressed, those would fail again. -- [1] https://issues.apache.org/jira/browse/CLOUDSTACK-7405
140 lines
6.6 KiB
Python
140 lines
6.6 KiB
Python
from . import helpers
|
|
|
|
from cloudinit import ec2_utils as eu
|
|
from cloudinit import url_helper as uh
|
|
|
|
import httpretty as hp
|
|
|
|
|
|
class TestEc2Util(helpers.HttprettyTestCase):
|
|
VERSION = 'latest'
|
|
|
|
@hp.activate
|
|
def test_userdata_fetch(self):
|
|
hp.register_uri(hp.GET,
|
|
'http://169.254.169.254/%s/user-data' % (self.VERSION),
|
|
body='stuff',
|
|
status=200)
|
|
userdata = eu.get_instance_userdata(self.VERSION)
|
|
self.assertEquals('stuff', userdata)
|
|
|
|
@hp.activate
|
|
def test_userdata_fetch_fail_not_found(self):
|
|
hp.register_uri(hp.GET,
|
|
'http://169.254.169.254/%s/user-data' % (self.VERSION),
|
|
status=404)
|
|
userdata = eu.get_instance_userdata(self.VERSION, retries=0)
|
|
self.assertEquals('', userdata)
|
|
|
|
@hp.activate
|
|
def test_userdata_fetch_fail_server_dead(self):
|
|
hp.register_uri(hp.GET,
|
|
'http://169.254.169.254/%s/user-data' % (self.VERSION),
|
|
status=500)
|
|
userdata = eu.get_instance_userdata(self.VERSION, retries=0)
|
|
self.assertEquals('', userdata)
|
|
|
|
@hp.activate
|
|
def test_userdata_fetch_fail_server_not_found(self):
|
|
hp.register_uri(hp.GET,
|
|
'http://169.254.169.254/%s/user-data' % (self.VERSION),
|
|
status=404)
|
|
userdata = eu.get_instance_userdata(self.VERSION)
|
|
self.assertEquals('', userdata)
|
|
|
|
@hp.activate
|
|
def test_metadata_fetch_no_keys(self):
|
|
base_url = 'http://169.254.169.254/%s/meta-data/' % (self.VERSION)
|
|
hp.register_uri(hp.GET, base_url, status=200,
|
|
body="\n".join(['hostname',
|
|
'instance-id',
|
|
'ami-launch-index']))
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'hostname'),
|
|
status=200, body='ec2.fake.host.name.com')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'instance-id'),
|
|
status=200, body='123')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'ami-launch-index'),
|
|
status=200, body='1')
|
|
md = eu.get_instance_metadata(self.VERSION, retries=0)
|
|
self.assertEquals(md['hostname'], 'ec2.fake.host.name.com')
|
|
self.assertEquals(md['instance-id'], '123')
|
|
self.assertEquals(md['ami-launch-index'], '1')
|
|
|
|
@hp.activate
|
|
def test_metadata_fetch_key(self):
|
|
base_url = 'http://169.254.169.254/%s/meta-data/' % (self.VERSION)
|
|
hp.register_uri(hp.GET, base_url, status=200,
|
|
body="\n".join(['hostname',
|
|
'instance-id',
|
|
'public-keys/']))
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'hostname'),
|
|
status=200, body='ec2.fake.host.name.com')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'instance-id'),
|
|
status=200, body='123')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'public-keys/'),
|
|
status=200, body='0=my-public-key')
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url, 'public-keys/0/openssh-key'),
|
|
status=200, body='ssh-rsa AAAA.....wZEf my-public-key')
|
|
md = eu.get_instance_metadata(self.VERSION, retries=0, timeout=0.1)
|
|
self.assertEquals(md['hostname'], 'ec2.fake.host.name.com')
|
|
self.assertEquals(md['instance-id'], '123')
|
|
self.assertEquals(1, len(md['public-keys']))
|
|
|
|
@hp.activate
|
|
def test_metadata_fetch_with_2_keys(self):
|
|
base_url = 'http://169.254.169.254/%s/meta-data/' % (self.VERSION)
|
|
hp.register_uri(hp.GET, base_url, status=200,
|
|
body="\n".join(['hostname',
|
|
'instance-id',
|
|
'public-keys/']))
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'hostname'),
|
|
status=200, body='ec2.fake.host.name.com')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'instance-id'),
|
|
status=200, body='123')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'public-keys/'),
|
|
status=200,
|
|
body="\n".join(['0=my-public-key', '1=my-other-key']))
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url, 'public-keys/0/openssh-key'),
|
|
status=200, body='ssh-rsa AAAA.....wZEf my-public-key')
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url, 'public-keys/1/openssh-key'),
|
|
status=200, body='ssh-rsa AAAA.....wZEf my-other-key')
|
|
md = eu.get_instance_metadata(self.VERSION, retries=0, timeout=0.1)
|
|
self.assertEquals(md['hostname'], 'ec2.fake.host.name.com')
|
|
self.assertEquals(md['instance-id'], '123')
|
|
self.assertEquals(2, len(md['public-keys']))
|
|
|
|
@hp.activate
|
|
def test_metadata_fetch_bdm(self):
|
|
base_url = 'http://169.254.169.254/%s/meta-data/' % (self.VERSION)
|
|
hp.register_uri(hp.GET, base_url, status=200,
|
|
body="\n".join(['hostname',
|
|
'instance-id',
|
|
'block-device-mapping/']))
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'hostname'),
|
|
status=200, body='ec2.fake.host.name.com')
|
|
hp.register_uri(hp.GET, uh.combine_url(base_url, 'instance-id'),
|
|
status=200, body='123')
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url, 'block-device-mapping/'),
|
|
status=200,
|
|
body="\n".join(['ami', 'ephemeral0']))
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url, 'block-device-mapping/ami'),
|
|
status=200,
|
|
body="sdb")
|
|
hp.register_uri(hp.GET,
|
|
uh.combine_url(base_url,
|
|
'block-device-mapping/ephemeral0'),
|
|
status=200,
|
|
body="sdc")
|
|
md = eu.get_instance_metadata(self.VERSION, retries=0, timeout=0.1)
|
|
self.assertEquals(md['hostname'], 'ec2.fake.host.name.com')
|
|
self.assertEquals(md['instance-id'], '123')
|
|
bdm = md['block-device-mapping']
|
|
self.assertEquals(2, len(bdm))
|
|
self.assertEquals(bdm['ami'], 'sdb')
|
|
self.assertEquals(bdm['ephemeral0'], 'sdc')
|