1 files changed, 108 insertions, 0 deletions
diff --git a/testing/web-platform/tests/worklets/resources/service-worker-interception-tests.js b/testing/web-platform/tests/worklets/resources/service-worker-interception-tests.js
new file mode 100644
index 0000000000..1298a626e4
--- /dev/null
+++ b/testing/web-platform/tests/worklets/resources/service-worker-interception-tests.js
@@ -0,0 +1,108 @@
+function openWindow(t, url) {
+  return new Promise(resolve => {
+    const win = window.open(url, '_blank');
+    t.add_cleanup(() => win.close());
+    window.onmessage = e => {
+      assert_equals(e.data, 'LOADED');
+      resolve(win);
+    };
+  });
+}
+
+// Runs a series of tests related to service worker interception for a worklet.
+//
+// Usage:
+// runServiceWorkerInterceptionTests("paint");
+function runServiceWorkerInterceptionTests(worklet_type) {
+  const worklet = get_worklet(worklet_type);
+
+  // Tests that a worklet should be served by the owner document's service
+  // worker.
+  //
+  // [Current document] registers a service worker for Window's URL.
+  // --(open)--> [Window] should be controlled by the service worker.
+  //   --(addModule)--> [Worklet] should be served by the service worker.
+  promise_test(async t => {
+    const kWindowURL = 'resources/addmodule-window.html';
+    const kServiceWorkerScriptURL = 'resources/service-worker.js';
+    // This doesn't contain the 'resources/' prefix because this will be
+    // imported from a html file under resources/.
+    const kWorkletScriptURL = 'non-existent-worklet-script.js';
+
+    const registration = await service_worker_unregister_and_register(
+        t, kServiceWorkerScriptURL, kWindowURL);
+    t.add_cleanup(() => registration.unregister());
+    await wait_for_state(t, registration.installing, 'activated');
+
+    const win = await openWindow(t, kWindowURL);
+    assert_not_equals(win.navigator.serviceWorker.controller, null,
+                      'The document should be controlled.');
+
+    // The worklet script on kWorkletScriptURL doesn't exist but the service
+    // worker serves it, so the addModule() should succeed.
+    win.postMessage({ type: worklet_type, script_url: kWorkletScriptURL }, '*');
+    const msgEvent = await new Promise(resolve => window.onmessage = resolve);
+    assert_equals(msgEvent.data, 'RESOLVED');
+  }, 'addModule() on a controlled document should be intercepted by a ' +
+     'service worker.');
+
+  // Tests that a worklet should not be served by a service worker other than
+  // the owner document's service worker.
+  //
+  // [Current document] registers a service worker for Worklet's URL.
+  // --(open)--> [Window] should not be controlled by the service worker.
+  //   --(addModule)--> [Worklet] should not be served by the service worker.
+  promise_test(async t => {
+    const kWindowURL = 'resources/addmodule-window.html';
+    const kServiceWorkerScriptURL = 'resources/service-worker.js';
+    // This doesn't contain the 'resources/' prefix because this will be
+    // imported from a html file under resources/.
+    const kWorkletScriptURL = 'non-existent-worklet-script.js';
+
+    const registration = await service_worker_unregister_and_register(
+        t, kServiceWorkerScriptURL, 'resources/' + kWorkletScriptURL);
+    t.add_cleanup(() => registration.unregister());
+    await wait_for_state(t, registration.installing, 'activated');
+
+    const win = await openWindow(t, kWindowURL);
+    assert_equals(win.navigator.serviceWorker.controller, null,
+                  'The document should not be controlled.');
+
+    // The worklet script on kWorkletScriptURL doesn't exist and the service
+    // worker doesn't serve it, so the addModule() should fail.
+    win.postMessage({ type: worklet_type, script_url: kWorkletScriptURL }, '*');
+    const msgEvent = await new Promise(resolve => window.onmessage = resolve);
+    assert_equals(msgEvent.data, 'REJECTED');
+  }, 'addModule() on a non-controlled document should not be intercepted by ' +
+     'a service worker even if the script is under the service worker scope.');
+
+  // Tests that static import should be served by the owner document's service
+  // worker.
+  //
+  // [Current document] registers a service worker for Window's URL.
+  // --(open)--> [Window] should be controlled by the service worker.
+  //   --(addModule)--> [Worklet] should be served by the service worker.
+  //     --(static import)--> [Script] should be served by the service worker.
+  promise_test(async t => {
+    const kWindowURL = 'resources/addmodule-window.html';
+    const kServiceWorkerScriptURL = 'resources/service-worker.js';
+    // This doesn't contain the 'resources/' prefix because this will be
+    // imported from a html file under resources/.
+    const kWorkletScriptURL = 'import-non-existent-worklet-script.js';
+
+    const registration = await service_worker_unregister_and_register(
+        t, kServiceWorkerScriptURL, kWindowURL);
+    t.add_cleanup(() => registration.unregister());
+    await wait_for_state(t, registration.installing, 'activated');
+
+    const win = await openWindow(t, kWindowURL);
+    assert_not_equals(win.navigator.serviceWorker.controller, null,
+                      'The document should be controlled.');
+
+    // A script statically imported by the worklet doesn't exist but the service
+    // worker serves it, so the addModule() should succeed.
+    win.postMessage({ type: worklet_type, script_url: kWorkletScriptURL }, '*');
+    const msgEvent = await new Promise(resolve => window.onmessage = resolve);
+    assert_equals(msgEvent.data, 'RESOLVED');
+  }, 'Static import should be intercepted by a service worker.');
+}