[shared storage] Add a WPT to test cross-origin worklet functions more completely

yaoxiachromium · chromium-wpt-export-bot · commit d47590aae5f0 · 2024-04-24T17:16:37.000-07:00
Bug: 336832600 Change-Id: Ia02137d4a7e65dc33b10e6b1b9a29ebfa3099d08 Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5485328 Reviewed-by: Cammie Smith Barnes <cammie@chromium.org> Commit-Queue: Yao Xiao <yaoxia@chromium.org> Commit-Queue: Cammie Smith Barnes <cammie@chromium.org> Cr-Commit-Position: refs/heads/main@{#1292115}
diff --git a/shared-storage/cross-origin-worklet-select-url-and-verify-data-origin.tentative.https.sub.html b/shared-storage/cross-origin-worklet-select-url-and-verify-data-origin.tentative.https.sub.html
@@ -0,0 +1,46 @@
+<!doctype html>
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="/common/utils.js"></script>
+<script src="/shared-storage/resources/util.js"></script>
+<script src="/fenced-frame/resources/utils.js"></script>
+
+<body>
+<script>
+'use strict';
+
+promise_test(async () => {
+  const crossOrigin = 'https://{{domains[www]}}:{{ports[https][0]}}';
+  const script_url = crossOrigin +
+                     `/shared-storage/resources/simple-module.js`;
+
+  const worklet = await sharedStorage.createWorklet(
+    script_url,
+    { credentials: "omit" });
+
+  const ancestor_key = token();
+  let url0 = generateURL("/shared-storage/resources/frame0.html",
+                         [ancestor_key]);
+
+  let select_url_result = await worklet.selectURL(
+    "test-url-selection-operation",
+    [{ url: url0 }], {
+      data: {
+        'mockResult': 0,
+        'setKey': 'key0',
+        'setValue': 'value0'
+      },
+      resolveToConfig: true,
+      keepAlive: true
+    });
+
+  assert_true(validateSelectURLResult(select_url_result, true));
+  attachFencedFrame(select_url_result, 'opaque-ads');
+  const result0 = await nextValueFromServer(ancestor_key);
+  assert_equals(result0, "frame0_loaded");
+
+  await verifyKeyValueForOrigin('key0', 'value0', crossOrigin);
+}, 'For a cross-origin worklet, test selectURL() and verify its data origin');
+
+</script>
+</body>
diff --git a/shared-storage/resources/simple-module.js b/shared-storage/resources/simple-module.js
@@ -6,6 +6,10 @@ var globalVar = 0;
 
 class TestURLSelectionOperation {
   async run(urls, data) {
+    if (data && data.hasOwnProperty('setKey') && data.hasOwnProperty('setValue')) {
+      await sharedStorage.set(data['setKey'], data['setValue']);
+    }
+
     if (data && data.hasOwnProperty('mockResult')) {
       return data['mockResult'];
     }
diff --git a/shared-storage/resources/simple-module.js.headers b/shared-storage/resources/simple-module.js.headers
@@ -0,0 +1,2 @@
+Access-Control-Allow-Origin: *
+Shared-Storage-Cross-Origin-Worklet-Allowed: ?1

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+Access-Control-Allow-Origin: *`
	`2`	`+Shared-Storage-Cross-Origin-Worklet-Allowed: ?1`