ClickHouse/tests/integration/test_s3_imds/metadata_servers/simple_server.py
2023-01-06 15:55:26 +01:00

37 lines
1.0 KiB
Python

import http.server
import sys
class RequestHandler(http.server.BaseHTTPRequestHandler):
def get_response(self):
if self.path == "/":
return "OK", 200
if self.path == "/latest/meta-data/iam/security-credentials":
return "myrole", 200
if self.path == "/latest/meta-data/iam/security-credentials/myrole":
return (
'{ "Code" : "Success", "Type" : "AWS-HMAC", "AccessKeyId" : "minio", "SecretAccessKey" : "minio123" }',
200,
)
# Resource not found.
return 404
def do_HEAD(self):
response, code = self.get_response()
self.send_response(code)
self.send_header("Content-Type", "text/plain")
self.send_header("Content-Length", len(response.encode()))
self.end_headers()
return response, code
def do_GET(self):
response, _ = self.do_HEAD()
self.wfile.write(response.encode())
httpd = http.server.HTTPServer(("0.0.0.0", int(sys.argv[1])), RequestHandler)
httpd.serve_forever()