summaryrefslogtreecommitdiffstats
path: root/addon-sdk/source/test/page-mod
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /addon-sdk/source/test/page-mod
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'addon-sdk/source/test/page-mod')
-rw-r--r--addon-sdk/source/test/page-mod/helpers.js117
1 files changed, 117 insertions, 0 deletions
diff --git a/addon-sdk/source/test/page-mod/helpers.js b/addon-sdk/source/test/page-mod/helpers.js
new file mode 100644
index 000000000..3aa3deb0d
--- /dev/null
+++ b/addon-sdk/source/test/page-mod/helpers.js
@@ -0,0 +1,117 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+"use strict";
+
+const { Cc, Ci } = require("chrome");
+const { setTimeout } = require("sdk/timers");
+const { Loader } = require("sdk/test/loader");
+const { openTab, getBrowserForTab, closeTab } = require("sdk/tabs/utils");
+const { getMostRecentBrowserWindow } = require("sdk/window/utils");
+const { merge } = require("sdk/util/object");
+const httpd = require("../lib/httpd");
+const { cleanUI } = require("sdk/test/utils");
+
+const PORT = 8099;
+const PATH = '/test-contentScriptWhen.html';
+
+function createLoader () {
+ let options = merge({}, require('@loader/options'),
+ { id: "testloader", prefixURI: require('../fixtures').url() });
+ return Loader(module, null, options);
+}
+exports.createLoader = createLoader;
+
+function openNewTab(url) {
+ return openTab(getMostRecentBrowserWindow(), url, {
+ inBackground: false
+ });
+}
+exports.openNewTab = openNewTab;
+
+// an evil function enables the creation of tests
+// that depend on delicate event timing. do not use.
+function testPageMod(assert, done, testURL, pageModOptions,
+ testCallback, timeout) {
+ let loader = createLoader();
+ let { PageMod } = loader.require("sdk/page-mod");
+ let pageMods = pageModOptions.map(opts => new PageMod(opts));
+ let newTab = openNewTab(testURL);
+ let b = 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.
+ setTimeout(testCallback, timeout,
+ b.contentWindow.wrappedJSObject, // TODO: remove this CPOW
+ function () {
+ pageMods.forEach(mod => mod.destroy());
+ // XXX leaks reported if we don't close the tab?
+ closeTab(newTab);
+ loader.unload();
+ done();
+ }
+ );
+ }
+ b.addEventListener("load", onPageLoad, true);
+
+ return pageMods;
+}
+exports.testPageMod = testPageMod;
+
+/**
+ * helper function that creates a PageMod and calls back the appropriate handler
+ * based on the value of document.readyState at the time contentScript is attached
+ */
+exports.handleReadyState = function(url, contentScriptWhen, callbacks) {
+ const loader = Loader(module);
+ const { PageMod } = loader.require('sdk/page-mod');
+
+ let pagemod = PageMod({
+ include: url,
+ attachTo: ['existing', 'top'],
+ contentScriptWhen: contentScriptWhen,
+ contentScript: "self.postMessage(document.readyState)",
+ onAttach: worker => {
+ let { tab } = worker;
+ worker.on('message', readyState => {
+ // generate event name from `readyState`, e.g. `"loading"` becomes `onLoading`.
+ let type = 'on' + readyState[0].toUpperCase() + readyState.substr(1);
+
+ if (type in callbacks)
+ callbacks[type](tab);
+
+ pagemod.destroy();
+ loader.unload();
+ })
+ }
+ });
+}
+
+// serves a slow page which takes 1.5 seconds to load,
+// 0.5 seconds in each readyState: uninitialized, loading, interactive.
+function contentScriptWhenServer() {
+ const URL = 'http://localhost:' + PORT + PATH;
+
+ const HTML = `/* polyglot js
+ <script src="${URL}"></script>
+ delay both the "DOMContentLoaded"
+ <script async src="${URL}"></script>
+ and "load" events */`;
+
+ let srv = httpd.startServerAsync(PORT);
+
+ srv.registerPathHandler(PATH, (_, response) => {
+ response.processAsync();
+ response.setHeader('Content-Type', 'text/html', false);
+ setTimeout(_ => response.finish(), 500);
+ response.write(HTML);
+ })
+
+ srv.URL = URL;
+ return srv;
+}
+exports.contentScriptWhenServer = contentScriptWhenServer;