1 files changed, 81 insertions, 0 deletions
diff --git a/dom/cache/test/mochitest/worker_driver.js b/dom/cache/test/mochitest/worker_driver.js
new file mode 100644
index 0000000000..03535c9839
--- /dev/null
+++ b/dom/cache/test/mochitest/worker_driver.js
@@ -0,0 +1,81 @@
+// Any copyright is dedicated to the Public Domain.
+// http://creativecommons.org/publicdomain/zero/1.0/
+//
+// Utility script for writing worker tests.  In your main document do:
+//
+//  <script type="text/javascript" src="worker_driver.js"></script>
+//  <script type="text/javascript">
+//    workerTestExec('myWorkerTestCase.js')
+//  </script>
+//
+// This will then spawn a worker, define some utility functions, and then
+// execute the code in myWorkerTestCase.js.  You can then use these
+// functions in your worker-side test:
+//
+//  ok() - like the SimpleTest assert
+//  is() - like the SimpleTest assert
+//  workerTestDone() - like SimpleTest.finish() indicating the test is complete
+//
+// There are also some functions for requesting information that requires
+// SpecialPowers or other main-thread-only resources:
+//
+//  workerTestGetPrefs() - request an array of prefs value from the main thread
+//  workerTestGetPermissions() - request an array permissions from the MT
+//  workerTestGetVersion() - request the current version string from the MT
+//  workerTestGetUserAgent() - request the user agent string from the MT
+//
+// For an example see test_worker_interfaces.html and test_worker_interfaces.js.
+
+function workerTestExec(script) {
+  return new Promise(function (resolve, reject) {
+    var worker = new Worker("worker_wrapper.js");
+    worker.onmessage = function (event) {
+      is(
+        event.data.context,
+        "Worker",
+        "Correct context for messages received on the worker"
+      );
+      if (event.data.type == "finish") {
+        worker.terminate();
+        SpecialPowers.forceGC();
+        resolve();
+      } else if (event.data.type == "status") {
+        ok(event.data.status, event.data.context + ": " + event.data.msg);
+      } else if (event.data.type == "getPrefs") {
+        let result = {};
+        event.data.prefs.forEach(function (pref) {
+          result[pref] = SpecialPowers.Services.prefs.getBoolPref(pref);
+        });
+        worker.postMessage({
+          type: "returnPrefs",
+          prefs: event.data.prefs,
+          result,
+        });
+      } else if (event.data.type == "getPermissions") {
+        let result = {};
+        event.data.permissions.forEach(function (permission) {
+          result[permission] = SpecialPowers.hasPermission(
+            permission,
+            window.document
+          );
+        });
+        worker.postMessage({
+          type: "returnPermissions",
+          permissions: event.data.permissions,
+          result,
+        });
+      } else if (event.data.type == "getUserAgent") {
+        worker.postMessage({
+          type: "returnUserAgent",
+          result: navigator.userAgent,
+        });
+      }
+    };
+
+    worker.onerror = function (event) {
+      reject("Worker had an error: " + event.data);
+    };
+
+    worker.postMessage({ script });
+  });
+}