1 files changed, 90 insertions, 0 deletions
diff --git a/browser/base/content/test/general/browser_homeDrop.js b/browser/base/content/test/general/browser_homeDrop.js
new file mode 100644
index 000000000..6e87963d5
--- /dev/null
+++ b/browser/base/content/test/general/browser_homeDrop.js
@@ -0,0 +1,90 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+add_task(function*() {
+  let HOMEPAGE_PREF = "browser.startup.homepage";
+
+  let homepageStr = Cc["@mozilla.org/supports-string;1"]
+                    .createInstance(Ci.nsISupportsString);
+  homepageStr.data = "about:mozilla";
+  yield pushPrefs([HOMEPAGE_PREF, homepageStr, Ci.nsISupportsString]);
+
+  let scriptLoader = Cc["@mozilla.org/moz/jssubscript-loader;1"].
+                     getService(Ci.mozIJSSubScriptLoader);
+  let EventUtils = {};
+  scriptLoader.loadSubScript("chrome://mochikit/content/tests/SimpleTest/EventUtils.js", EventUtils);
+
+  // Since synthesizeDrop triggers the srcElement, need to use another button.
+  let dragSrcElement = document.getElementById("downloads-button");
+  ok(dragSrcElement, "Downloads button exists");
+  let homeButton = document.getElementById("home-button");
+  ok(homeButton, "home button present");
+
+  function* drop(dragData, homepage) {
+    let setHomepageDialogPromise = BrowserTestUtils.domWindowOpened();
+
+    EventUtils.synthesizeDrop(dragSrcElement, homeButton, dragData, "copy", window);
+
+    let setHomepageDialog = yield setHomepageDialogPromise;
+    ok(true, "dialog appeared in response to home button drop");
+    yield BrowserTestUtils.waitForEvent(setHomepageDialog, "load", false);
+
+    let setHomepagePromise = new Promise(function(resolve) {
+      let observer = {
+        QueryInterface: XPCOMUtils.generateQI([Ci.nsIObserver]),
+        observe: function(subject, topic, data) {
+          is(topic, "nsPref:changed", "observed correct topic");
+          is(data, HOMEPAGE_PREF, "observed correct data");
+          let modified = Services.prefs.getComplexValue(HOMEPAGE_PREF,
+                                                        Ci.nsISupportsString);
+          is(modified.data, homepage, "homepage is set correctly");
+          Services.prefs.removeObserver(HOMEPAGE_PREF, observer);
+
+          Services.prefs.setComplexValue(HOMEPAGE_PREF,
+                                         Ci.nsISupportsString, homepageStr);
+
+          resolve();
+        }
+      };
+      Services.prefs.addObserver(HOMEPAGE_PREF, observer, false);
+    });
+
+    setHomepageDialog.document.documentElement.acceptDialog();
+
+    yield setHomepagePromise;
+  }
+
+  function dropInvalidURI() {
+    return new Promise(resolve => {
+      let consoleListener = {
+        observe: function (m) {
+          if (m.message.includes("NS_ERROR_DOM_BAD_URI")) {
+            ok(true, "drop was blocked");
+            resolve();
+          }
+        }
+      };
+      Services.console.registerListener(consoleListener);
+      registerCleanupFunction(function () {
+        Services.console.unregisterListener(consoleListener);
+      });
+
+      executeSoon(function () {
+        info("Attempting second drop, of a javascript: URI");
+        // The drop handler throws an exception when dragging URIs that inherit
+        // principal, e.g. javascript:
+        expectUncaughtException();
+        EventUtils.synthesizeDrop(dragSrcElement, homeButton, [[{type: "text/plain", data: "javascript:8888"}]], "copy", window);
+      });
+    });
+  }
+
+  yield* drop([[{type: "text/plain",
+                 data: "http://mochi.test:8888/"}]],
+              "http://mochi.test:8888/");
+  yield* drop([[{type: "text/plain",
+                 data: "http://mochi.test:8888/\nhttp://mochi.test:8888/b\nhttp://mochi.test:8888/c"}]],
+              "http://mochi.test:8888/|http://mochi.test:8888/b|http://mochi.test:8888/c");
+  yield dropInvalidURI();
+});
+