1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import datetime
import json
from typing import Any
from urllib.parse import urlparse
from moto.core.common_types import TYPE_RESPONSE
from moto.core.responses import BaseResponse
from moto.core.utils import utcnow
class InstanceMetadataResponse(BaseResponse):
def __init__(self) -> None:
super().__init__(service_name=None)
def backends(self) -> None:
pass
@staticmethod
def metadata_response( # type: ignore
request: Any,
full_url: str,
headers: Any,
) -> TYPE_RESPONSE:
"""
Mock response for localhost metadata
http://docs.aws.amazon.com/AWSEC2/latest/UserGuide/AESDG-chapter-instancedata.html
"""
parsed_url = urlparse(full_url)
tomorrow = utcnow() + datetime.timedelta(days=1)
credentials = {
"AccessKeyId": "test-key",
"SecretAccessKey": "test-secret-key",
"Token": "test-session-token",
"Expiration": tomorrow.strftime("%Y-%m-%dT%H:%M:%SZ"),
}
path = parsed_url.path
meta_data_prefix = "/latest/meta-data/"
# Strip prefix if it is there
if path.startswith(meta_data_prefix):
path = path[len(meta_data_prefix) :]
if path == "":
result = "iam"
elif path == "iam":
result = json.dumps({"security-credentials": {"default-role": credentials}})
elif path == "iam/security-credentials/":
result = "default-role"
elif path == "iam/security-credentials/default-role":
result = json.dumps(credentials)
else:
raise NotImplementedError(
f"The {path} metadata path has not been implemented"
)
try:
from werkzeug.datastructures.headers import EnvironHeaders
if isinstance(headers, EnvironHeaders):
# We should be returning a generic dict here, not werkzeug-specific classes
headers = dict(headers)
except ImportError:
pass
return 200, headers, result
|
