1 files changed, 45 insertions, 0 deletions
diff --git a/testing/web-platform/tests/loading/early-hints/resources/early-hints-test-loader.h2.py b/testing/web-platform/tests/loading/early-hints/resources/early-hints-test-loader.h2.py
new file mode 100644
index 0000000000..aa9188c6db
--- /dev/null
+++ b/testing/web-platform/tests/loading/early-hints/resources/early-hints-test-loader.h2.py
@@ -0,0 +1,45 @@
+# An HTTP/2 handler for testing Early Hints. Used as an entry point of Early
+# Hints related tests to inject Early Hints response. See comments in
+# `early-hints-helpers.sub.js`.
+
+import json
+import os
+import time
+
+
+def _remove_relative_resources_prefix(path):
+    if path.startswith("resources/"):
+        return path[len("resources/"):]
+    return path
+
+
+def handle_headers(frame, request, response):
+    preload_headers = []
+    for encoded_preload in request.GET.get_list(b"preloads"):
+        preload = json.loads(encoded_preload.decode("utf-8"))
+        header = "<{}>; rel=preload; as={}".format(preload["url"], preload["as_attr"])
+        preload_headers.append(header.encode())
+
+    # Send a 103 response.
+    early_hints = [(b":status", b"103")]
+    for header in preload_headers:
+        early_hints.append((b"link", header))
+    response.writer.write_raw_header_frame(headers=early_hints,
+                                           end_headers=True)
+
+    # Simulate the response generation is taking time.
+    time.sleep(0.2)
+    response.status = 200
+    response.headers[b"content-type"] = "text/html"
+    for header in preload_headers:
+        response.headers.append(b"link", header)
+    response.write_status_headers()
+
+
+def main(request, response):
+    test_path = _remove_relative_resources_prefix(
+        request.GET[b"test_url"].decode("utf-8"))
+    current_dir = os.path.dirname(os.path.realpath(__file__))
+    file_path = os.path.join(current_dir, test_path)
+    test_content = open(file_path, "r").read()
+    response.writer.write_data(item=test_content, last=True)