2 files changed, 58 insertions, 0 deletions
diff --git a/testing/web-platform/tests/client-hints/critical-ch/resources/echo-critical-hint.py b/testing/web-platform/tests/client-hints/critical-ch/resources/echo-critical-hint.py
new file mode 100644
index 0000000000..e4e77ad2a9
--- /dev/null
+++ b/testing/web-platform/tests/client-hints/critical-ch/resources/echo-critical-hint.py
@@ -0,0 +1,43 @@
+import sys
+
+def main(request, response):
+    """
+    Simple handler that sets a response header based on which client hint
+    request headers were received.
+    """
+
+    response.headers.append(b"Content-Type", b"text/html; charset=UTF-8")
+    response.headers.append(b"Access-Control-Allow-Origin", b"*")
+    response.headers.append(b"Access-Control-Allow-Headers", b"*")
+    response.headers.append(b"Access-Control-Expose-Headers", b"*")
+
+    response.headers.append(b"Accept-CH", b"sec-ch-device-memory,device-memory")
+
+    critical = b"sec-ch-device-memory,device-memory"
+    if(request.GET.first(b"mismatch", None) is not None):
+      critical = b"sec-ch-viewport-width,viewport-width"
+
+    response.headers.append(b"Critical-CH", critical)
+
+    response.headers.append(b"Cache-Control", b"no-store")
+
+    result = "FAIL"
+
+    if b"sec-ch-device-memory" in request.headers and b"device-memory" in request.headers:
+      result = "PASS"
+
+    token = request.GET.first(b"token", None)
+    if(token is not None):
+      with request.server.stash.lock:
+        count = request.server.stash.take(token)
+        if(count == None):
+          count = 1
+        else:
+          count += 1
+        request.server.stash.put(token, count)
+        result = str(count)
+
+    if b"sec-ch-viewport-width" in request.headers and b"viewport-width" in request.headers:
+      result = "MISMATCH"
+
+    response.content = "<script>window.postMessage('{0}', '*')</script>".format(result)
diff --git a/testing/web-platform/tests/client-hints/critical-ch/resources/util.js b/testing/web-platform/tests/client-hints/critical-ch/resources/util.js
new file mode 100644
index 0000000000..7471d9022e
--- /dev/null
+++ b/testing/web-platform/tests/client-hints/critical-ch/resources/util.js
@@ -0,0 +1,15 @@
+ECHO_URL = "resources/echo-critical-hint.py"
+
+message_listener = (t, message) =>
+  (e) => {
+    t.step(()=>{assert_equals(e.data, message)});
+    t.done();
+  }
+
+make_message_test = (url, message) =>
+  (t) => {
+    popup_window = window.open("/common/blank.html");
+    assert_not_equals(popup_window, null, "Popup windows not allowed?");
+    popup_window.addEventListener('message', message_listener(t, message));
+    popup_window.location = url;
+  }