S3 - Add RequestId to responses

moto/s3/responses.py

@@ -8,6 +8,7 @@
 from botocore.awsrequest import AWSPreparedRequest
 
 from moto.core.utils import (
+    amzn_request_id,
     str_to_rfc_1123_datetime,
     py2_strip_unicode_keys,
     unix_time_millis,
@@ -258,6 +259,7 @@ def ambiguous_response(self, request, full_url, headers):
             # Using path-based buckets
             return self.bucket_response(request, full_url, headers)
 
+    @amzn_request_id
     def bucket_response(self, request, full_url, headers):
         self.method = request.method
         self.path = self._get_path(request)
@@ -1025,6 +1027,7 @@ def _handle_v4_chunk_signatures(self, body, content_length):
             line = body_io.readline()
         return bytes(new_body)
 
+    @amzn_request_id
     def key_or_control_response(self, request, full_url, headers):
         # Key and Control are lumped in because splitting out the regex is too much of a pain :/
         self.method = request.method

tests/test_s3/test_s3_metadata.py

@@ -0,0 +1,27 @@
+import boto3
+
+from moto import mock_s3
+from moto.s3.responses import DEFAULT_REGION_NAME
+
+import sure  # noqa
+
+
+@mock_s3
+def test_s3_returns_requestid():
+    s3 = boto3.client("s3", region_name=DEFAULT_REGION_NAME)
+    resp = s3.create_bucket(Bucket="mybucket")
+    _check_metadata(resp)
+
+    resp = s3.put_object(Bucket="mybucket", Key="steve", Body=b"is awesome")
+    _check_metadata(resp)
+
+    resp = s3.get_object(Bucket="mybucket", Key="steve")
+    _check_metadata(resp)
+
+
+def _check_metadata(resp):
+    meta = resp["ResponseMetadata"]
+    headers = meta["HTTPHeaders"]
+    meta.should.have.key("RequestId")
+    headers.should.have.key("x-amzn-requestid")
+    meta["RequestId"].should.equal(headers["x-amzn-requestid"])