2018-12-21 11:28:56 +00:00
|
|
|
from __future__ import unicode_literals
|
|
|
|
|
import sure # noqa
|
|
|
|
|
from nose.tools import assert_raises
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
from moto import mock_ec2, settings
|
|
|
|
|
|
|
|
|
|
if settings.TEST_SERVER_MODE:
|
|
|
|
|
BASE_URL = 'http://localhost:5000'
|
|
|
|
|
else:
|
|
|
|
|
BASE_URL = 'http://169.254.169.254'
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_latest_meta_data():
|
|
|
|
|
res = requests.get("{0}/latest/meta-data/".format(BASE_URL))
|
|
|
|
|
res.content.should.equal(b"iam")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_meta_data_iam():
|
|
|
|
|
res = requests.get("{0}/latest/meta-data/iam".format(BASE_URL))
|
|
|
|
|
json_response = res.json()
|
|
|
|
|
default_role = json_response['security-credentials']['default-role']
|
|
|
|
|
default_role.should.contain('AccessKeyId')
|
|
|
|
|
default_role.should.contain('SecretAccessKey')
|
|
|
|
|
default_role.should.contain('Token')
|
|
|
|
|
default_role.should.contain('Expiration')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_meta_data_security_credentials():
|
|
|
|
|
res = requests.get(
|
|
|
|
|
"{0}/latest/meta-data/iam/security-credentials/".format(BASE_URL))
|
|
|
|
|
res.content.should.equal(b"default-role")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_meta_data_default_role():
|
|
|
|
|
res = requests.get(
|
|
|
|
|
"{0}/latest/meta-data/iam/security-credentials/default-role".format(BASE_URL))
|
|
|
|
|
json_response = res.json()
|
|
|
|
|
json_response.should.contain('AccessKeyId')
|
|
|
|
|
json_response.should.contain('SecretAccessKey')
|
|
|
|
|
json_response.should.contain('Token')
|
|
|
|
|
json_response.should.contain('Expiration')
|
