Use Referrer-Policy headers for CSS stylesheets

third_party/WebKit/LayoutTests/external/wpt/referrer-policy/generic/subresource/image.py

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 = "*")