Index: third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py |
diff --git a/third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py b/third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py |
index b6306181e585aafefec68825849ac1709b3e35e7..7bc9c657e6b35528d63b6f2d72d73391dedfdf5b 100644 |
--- a/third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py |
+++ b/third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py |
@@ -88,13 +88,27 @@ def encode_string_as_bmp_image(string_data): |
return f.read() |
-def generate_payload(server_data): |
+def generate_payload(request, server_data): |
data = ('{"headers": %(headers)s}') % server_data |
- return encode_string_as_bmp_image(data) |
+ if "id" in request.GET: |
+ request.server.stash.put(request.GET["id"], data) |
+ data = encode_string_as_bmp_image(data) |
+ return data |
+ |
+def generate_report_headers_payload(request, server_data): |
+ stashed_data = request.server.stash.take(request.GET["id"]) |
+ return stashed_data |
def main(request, response): |
+ handler = lambda data: generate_payload(request, data) |
+ content_type = 'image/bmp' |
+ |
+ if "report-headers" in request.GET: |
+ handler = lambda data: generate_report_headers_payload(request, data) |
+ content_type = 'application/json' |
+ |
subresource.respond(request, |
response, |
- payload_generator = generate_payload, |
- content_type = "image/bmp", |
+ payload_generator = handler, |
+ content_type = content_type, |
access_control_allow_origin = "*") |