1 files changed, 153 insertions, 0 deletions
diff --git a/dom/indexedDB/test/head.js b/dom/indexedDB/test/head.js
new file mode 100644
index 0000000000..434334ff0b
--- /dev/null
+++ b/dom/indexedDB/test/head.js
@@ -0,0 +1,153 @@
+/**
+ * Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+var gActiveListeners = {};
+
+// These event (un)registration handlers only work for one window, DONOT use
+// them with multiple windows.
+function registerPopupEventHandler(eventName, callback, win) {
+  if (!win) {
+    win = window;
+  }
+  gActiveListeners[eventName] = function(event) {
+    if (event.target != win.PopupNotifications.panel) {
+      return;
+    }
+    win.PopupNotifications.panel.removeEventListener(
+      eventName,
+      gActiveListeners[eventName]
+    );
+    delete gActiveListeners[eventName];
+
+    callback.call(win.PopupNotifications.panel);
+  };
+  win.PopupNotifications.panel.addEventListener(
+    eventName,
+    gActiveListeners[eventName]
+  );
+}
+
+function unregisterPopupEventHandler(eventName, win) {
+  if (!win) {
+    win = window;
+  }
+  win.PopupNotifications.panel.removeEventListener(
+    eventName,
+    gActiveListeners[eventName]
+  );
+  delete gActiveListeners[eventName];
+}
+
+function unregisterAllPopupEventHandlers(win) {
+  if (!win) {
+    win = window;
+  }
+  for (let eventName in gActiveListeners) {
+    win.PopupNotifications.panel.removeEventListener(
+      eventName,
+      gActiveListeners[eventName]
+    );
+  }
+  gActiveListeners = {};
+}
+
+function triggerMainCommand(popup) {
+  info("triggering main command");
+  let notifications = popup.childNodes;
+  ok(notifications.length, "at least one notification displayed");
+  let notification = notifications[0];
+  info("triggering command: " + notification.getAttribute("buttonlabel"));
+
+  EventUtils.synthesizeMouseAtCenter(notification.button, {});
+}
+
+function triggerSecondaryCommand(popup, win) {
+  if (!win) {
+    win = window;
+  }
+  info("triggering secondary command");
+  let notifications = popup.childNodes;
+  ok(notifications.length, "at least one notification displayed");
+  let notification = notifications[0];
+  EventUtils.synthesizeMouseAtCenter(notification.secondaryButton, {}, win);
+}
+
+function dismissNotification(popup) {
+  info("dismissing notification");
+  executeSoon(function() {
+    EventUtils.synthesizeKey("KEY_Escape");
+  });
+}
+
+function waitForMessage(aMessage, browser) {
+  // We cannot capture aMessage inside the checkFn, so we override the
+  // checkFn.toSource to tunnel aMessage instead.
+  let checkFn = function() {};
+  checkFn.toSource = function() {
+    return `function checkFn(event) {
+      let message = ${aMessage.toSource()};
+      if (event.data == message) {
+        return true;
+      }
+      throw new Error(
+       \`Unexpected result: \$\{event.data\}, expected \$\{message\}\`
+      );
+    }`;
+  };
+
+  return BrowserTestUtils.waitForContentEvent(
+    browser.selectedBrowser,
+    "message",
+    /* capture */ true,
+    checkFn,
+    /* wantsUntrusted */ true
+  ).then(() => {
+    // An assertion in checkFn wouldn't be recorded as part of the test, so we
+    // use this assertion to confirm that we've successfully received the
+    // message (we'll only reach this point if that's the case).
+    ok(true, "Received message: " + aMessage);
+  });
+}
+
+function dispatchEvent(eventName) {
+  info("dispatching event: " + eventName);
+  let event = document.createEvent("Events");
+  event.initEvent(eventName, false, false);
+  gBrowser.selectedBrowser.contentWindow.dispatchEvent(event);
+}
+
+function setPermission(url, permission, originAttributes = {}) {
+  let uri = Services.io.newURI(url);
+  let principal = Services.scriptSecurityManager.createContentPrincipal(
+    uri,
+    originAttributes
+  );
+
+  Services.perms.addFromPrincipal(
+    principal,
+    permission,
+    Ci.nsIPermissionManager.ALLOW_ACTION
+  );
+}
+
+function removePermission(url, permission, originAttributes = {}) {
+  let uri = Services.io.newURI(url);
+  let principal = Services.scriptSecurityManager.createContentPrincipal(
+    uri,
+    originAttributes
+  );
+
+  Services.perms.removeFromPrincipal(principal, permission);
+}
+
+function getPermission(url, permission, originAttributes = {}) {
+  let uri = Services.io.newURI(url);
+  let principal = Services.scriptSecurityManager.createContentPrincipal(
+    uri,
+    originAttributes
+  );
+
+  return Services.perms.testPermissionFromPrincipal(principal, permission);
+}