1 files changed, 62 insertions, 0 deletions
diff --git a/tools/addon-sdk-1.3/packages/addon-kit/tests/pagemod-test-helpers.js b/tools/addon-sdk-1.3/packages/addon-kit/tests/pagemod-test-helpers.js
new file mode 100644
index 0000000..01799ec
--- /dev/null
+++ b/tools/addon-sdk-1.3/packages/addon-kit/tests/pagemod-test-helpers.js
@@ -0,0 +1,62 @@
+"use strict";
+
+const {Cc,Ci} = require("chrome");
+const timer = require("timer");
+
+/**
+ * A helper function that creates a PageMod, then opens the specified URL
+ * and checks the effect of the page mod on 'onload' event via testCallback.
+ */
+exports.testPageMod = function testPageMod(test, testURL, pageModOptions,
+                                           testCallback, timeout) {
+  var xulApp = require("xul-app");
+  if (!xulApp.versionInRange(xulApp.platformVersion, "1.9.3a3", "*") &&
+      !xulApp.versionInRange(xulApp.platformVersion, "1.9.2.7", "1.9.2.*")) {
+    test.pass("Note: not testing PageMod, as it doesn't work on this platform version");
+    return null;
+  }
+
+  var wm = Cc['@mozilla.org/appshell/window-mediator;1']
+           .getService(Ci.nsIWindowMediator);
+  var browserWindow = wm.getMostRecentWindow("navigator:browser");
+  if (!browserWindow) {
+    test.pass("page-mod tests: could not find the browser window, so " +
+              "will not run. Use -a firefox to run the pagemod tests.")
+    return null;
+  }
+
+  if (timeout !== undefined)
+    test.waitUntilDone(timeout);
+  else
+    test.waitUntilDone();
+
+  let loader = test.makeSandboxedLoader();
+  let pageMod = loader.require("page-mod");
+
+  var pageMods = [new pageMod.PageMod(opts) for each(opts in pageModOptions)];
+
+  var tabBrowser = browserWindow.gBrowser;
+  var newTab = tabBrowser.addTab(testURL);
+  tabBrowser.selectedTab = newTab;
+  var b = tabBrowser.getBrowserForTab(newTab);
+
+  function onPageLoad() {
+    b.removeEventListener("load", onPageLoad, true);
+    // Delay callback execute as page-mod content scripts may be executed on
+    // load event. So page-mod actions may not be already done.
+    // If we delay even more contentScriptWhen:'end', we may want to modify
+    // this code again.
+    timer.setTimeout(testCallback, 0,
+      b.contentWindow.wrappedJSObject, 
+      function done() {
+        pageMods.forEach(function(mod) mod.destroy());
+        // XXX leaks reported if we don't close the tab?
+        tabBrowser.removeTab(newTab);
+        loader.unload();
+        test.done();
+      });
+  }
+  b.addEventListener("load", onPageLoad, true);
+
+  return pageMods;
+}