Fix the password recognition.

 * doh! !== instead of ===
 * also skip attachment.cgi page

3 files changed, 207 insertions, 0 deletions

diff --git a/tests/pagemod-test-helpers.js b/tests/pagemod-test-helpers.js
new file mode 100644
index 0000000..b019810
--- /dev/null
+++ b/tests/pagemod-test-helpers.js
@@ -0,0 +1,55 @@
+"use strict";
+
+const {Cc,Ci} = require("chrome");
+
+/**
+ * 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);
+    testCallback(b.contentWindow.wrappedJSObject, function done() {
+      pageMods.forEach(function(mod) mod.destroy());
+      // XXX leaks reported if we don't close the tab?
+      tabBrowser.removeTab(newTab);
+      test.done();
+    });
+  }
+  b.addEventListener("load", onPageLoad, true);
+
+  return pageMods;
+}
diff --git a/tests/test-match-pattern.js b/tests/test-match-pattern.js
new file mode 100644
index 0000000..ed51b43
--- /dev/null
+++ b/tests/test-match-pattern.js
@@ -0,0 +1,11 @@
+var { MatchPattern } = require("match-pattern");
+
+exports.ensureMatchPattern = function (test) {
+    var pattern = new MatchPattern("https://bugzilla.redhat.com/attachment.cgi");
+    console.log("pattern = " + pattern);
+    test.assert(pattern.
+        test("https://bugzilla.redhat.com/attachment.cgi"), "testing match pattern");
+    test.assert(!pattern.
+        test("https://bugzilla.redhat.com/attachment.cgi?bugid=676538&action=enter"),
+        "testing failing match pattern");
+};
diff --git a/tests/test-pageMod.js b/tests/test-pageMod.js
new file mode 100644
index 0000000..8662a4c
--- /dev/null
+++ b/tests/test-pageMod.js
@@ -0,0 +1,141 @@
+/*jslint white: false, eqeqeq: false, plusplus: false, onevar: false, newcap: false */
+/*global exports: false, require: false, console: false, log: false */
+"use strict";
+var Cc = require("chrome").Cc;
+var Ci = require("chrome").Ci;
+var main = require("main");
+var utilMod = require("util");
+var testPageMod = require("pagemod-test-helpers").testPageMod;
+var self = require("self");
+
+var theURL = main.theURL;
+var testURL = self.data.url('tests/change-more-bugs01.html');
+var JSONifiedMessage = '{"cmd":"testMessage","data":{"a":"first","b":"second"}}';
+
+exports.ensureMessagesWork = function(test) {
+    var msg = new utilMod.Message("testMessage", { a: "first", b: "second" } );
+    test.assertEqual(msg.cmd, "testMessage",
+        "msg.cmd comes over well");
+    test.assertEqual(msg.data.a, "first",
+        "msg.data.a comes over well");
+    test.assertEqual(msg.data.b, "second",
+        "msg.data.b comes over well");
+    test.assertEqual(JSON.stringify(msg), JSONifiedMessage,
+        "JSONification of Message works as well");
+};
+
+/*
+
+var theURL = main.theURL;
+var testURL = self.data.url('tests/change-more-bugs01.html');
+
+var contentScriptLibraries = {
+    "bugzilla.redhat.com": [
+        self.data.url("util.js"),
+        self.data.url("color.js"),
+       self.data.url("rhbzpage.js"),
+        self.data.url("bzpage.js")
+    ]
+};
+
+libbz.initialize(libbz.config, function () {
+    pageMod.PageMod({
+        include: [
+            "https://bugzilla.redhat.com/show_bug.cgi?id=*"
+        ],
+        contentScriptWhen: 'ready',
+        contentScriptFile: contentScriptLibraries["bugzilla.redhat.com"],
+        onAttach: function onAttach(worker, msg) {
+            console.log("worker: " + worker);
+            worker.on('message', function (msg) {
+                messageHandler(worker, msg);
+            });
+        }
+    });
+});
+
+pageMod.PageMod({
+    include: [
+        "https://bugzilla.redhat.com/process_bug.cgi"
+    ],
+    contentScriptWhen: 'ready',
+    contentScriptFile: self.data.url("skip-bug.js")
+});
+ */
+
+var ensureSimplePageLoad = function (test) {
+	console.log("testURL = " + testURL);
+	testPageMod(test, testURL, [{
+		include: ["*"],
+		contentScriptWhen: 'ready',
+		contentScriptFile: [
+			self.data.url("libPW.js"),
+			self.data.url("simplePageWorker.js")
+		],
+		onAttach: function onAttach(worker) {
+			worker.on('message', function (msg) {
+				switch (msg.cmd) {
+					case "LogMessage":
+						log(msg.data);
+						break;
+					case "CallBack":
+						worker.postMessage(new utilMod.Message("Main", null));
+						break;
+					default:
+						console.error(msg);
+				}
+			});
+		}
+	}],
+	function (win, done) {
+		test.assertNotEqual(win.document.getElementsByTagName("form")[0],
+			null, "test of loading the page");
+		done();
+	});
+};
+
+var ensurePageLoadsWell = function (test) {
+  var wm = Cc['@mozilla.org/appshell/window-mediator;1']
+           .getService(Ci.nsIWindowMediator);
+  var browserWindow = wm.getMostRecentWindow("navigator:browser");
+  if (!browserWindow) {
+    test.fail("page-mod tests: could not find the browser window, so " +
+              "will not run. Use -a firefox to run the pagemod tests.");
+    return null;
+  }
+
+  var loader = test.makeSandboxedLoader();
+  var pageMod = loader.require("page-mod");
+  var testDoc = {}, b = {}, tabBrowser = {}, newTab  = {};
+
+  pageMod.PageMod({
+		include: ["*"],
+		contentScriptWhen: 'ready',
+		contentScriptFile: [
+			self.data.url("libPW.js"),
+			self.data.url("pageWorker.js")
+		],
+		onAttach: function onAttach(worker) {
+			worker.on('message', function (msg) {
+				switch (msg.cmd) {
+					case "testReady":
+						testDoc = b.contentWindow.wrappedJSObject.document;
+						test.assertNotEqual(testDoc.getElementById("dupeid_action"),
+							null, "test of DOM modifications of the page");
+						pageMod.destroy();
+						tabBrowser.removeTab(newTab);
+						test.done();
+						// the test itself
+						break;
+					default:
+						main.messageHandler(worker, msg);
+				}
+			});
+		}
+	});
+
+  tabBrowser = browserWindow.gBrowser;
+  newTab = tabBrowser.addTab(testURL);
+  tabBrowser.selectedTab = newTab;
+  b = tabBrowser.getBrowserForTab(newTab);
+};