1 files changed, 79 insertions, 0 deletions
diff --git a/dom/serviceworkers/test/ForceRefreshParent.jsm b/dom/serviceworkers/test/ForceRefreshParent.jsm
new file mode 100644
index 0000000000..ad22b216de
--- /dev/null
+++ b/dom/serviceworkers/test/ForceRefreshParent.jsm
@@ -0,0 +1,79 @@
+var EXPORTED_SYMBOLS = ["ForceRefreshParent"];
+
+var maxCacheLoadCount = 3;
+var cachedLoadCount = 0;
+var baseLoadCount = 0;
+var done = false;
+
+class ForceRefreshParent extends JSWindowActorParent {
+  constructor() {
+    super();
+  }
+
+  receiveMessage(msg) {
+    // if done is called, ignore the msg.
+    if (done) {
+      return;
+    }
+    if (msg.data.type === "base-load") {
+      baseLoadCount += 1;
+      if (cachedLoadCount === maxCacheLoadCount) {
+        ForceRefreshParent.SimpleTest.is(
+          baseLoadCount,
+          2,
+          "cached load should occur before second base load"
+        );
+        done = true;
+        return ForceRefreshParent.done();
+      }
+      if (baseLoadCount !== 1) {
+        ForceRefreshParent.SimpleTest.ok(
+          false,
+          "base load without cached load should only occur once"
+        );
+        done = true;
+        return ForceRefreshParent.done();
+      }
+    } else if (msg.data.type === "base-register") {
+      ForceRefreshParent.SimpleTest.ok(
+        !cachedLoadCount,
+        "cached load should not occur before base register"
+      );
+      ForceRefreshParent.SimpleTest.is(
+        baseLoadCount,
+        1,
+        "register should occur after first base load"
+      );
+    } else if (msg.data.type === "base-sw-ready") {
+      ForceRefreshParent.SimpleTest.ok(
+        !cachedLoadCount,
+        "cached load should not occur before base ready"
+      );
+      ForceRefreshParent.SimpleTest.is(
+        baseLoadCount,
+        1,
+        "ready should occur after first base load"
+      );
+      ForceRefreshParent.refresh();
+    } else if (msg.data.type === "cached-load") {
+      ForceRefreshParent.SimpleTest.ok(
+        cachedLoadCount < maxCacheLoadCount,
+        "cached load should not occur too many times"
+      );
+      ForceRefreshParent.SimpleTest.is(
+        baseLoadCount,
+        1,
+        "cache load occur after first base load"
+      );
+      cachedLoadCount += 1;
+      if (cachedLoadCount < maxCacheLoadCount) {
+        return ForceRefreshParent.refresh();
+      }
+      ForceRefreshParent.forceRefresh();
+    } else if (msg.data.type === "cached-failure") {
+      ForceRefreshParent.SimpleTest.ok(false, "failure: " + msg.data.detail);
+      done = true;
+      ForceRefreshParent.done();
+    }
+  }
+}