gecko-dev/browser/tools/mozscreenshots/head.js

65 строки
2.5 KiB
JavaScript

/* 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/. */
/* exported TestRunner, shouldCapture */
"use strict";
const {AddonWatcher} = Cu.import("resource://gre/modules/AddonWatcher.jsm", {});
const chromeRegistry = Cc["@mozilla.org/chrome/chrome-registry;1"].getService(Ci.nsIChromeRegistry);
const env = Cc["@mozilla.org/process/environment;1"].getService(Ci.nsIEnvironment);
const EXTENSION_DIR = "chrome://mochitests/content/extensions/mozscreenshots/browser/";
let TestRunner;
function* setup() {
// This timeout doesn't actually end the job even if it is hit - the buildbot timeout will
// handle things for us if the test actually hangs.
requestLongerTimeout(100);
info("installing extension temporarily");
let chromeURL = Services.io.newURI(EXTENSION_DIR);
let dir = chromeRegistry.convertChromeURL(chromeURL).QueryInterface(Ci.nsIFileURL).file;
yield AddonManager.installTemporaryAddon(dir);
info("Checking for mozscreenshots extension");
return new Promise((resolve) => {
AddonManager.getAddonByID("mozscreenshots@mozilla.org", function(aAddon) {
isnot(aAddon, null, "The mozscreenshots extension should be installed");
AddonWatcher.ignoreAddonPermanently(aAddon.id);
TestRunner = Cu.import("chrome://mozscreenshots/content/TestRunner.jsm", {}).TestRunner;
resolve();
});
});
}
/**
* Used by pre-defined sets of configurations to decide whether to run for a build.
* @note This is not used by browser_screenshots.js which handles when MOZSCREENSHOTS_SETS is set.
* @return {bool} whether to capture screenshots.
*/
function shouldCapture() {
if (env.get("MOZSCREENSHOTS_SETS")) {
ok(true, "MOZSCREENSHOTS_SETS was specified so only capture what was " +
"requested (in browser_screenshots.js)");
return false;
}
if (AppConstants.MOZ_UPDATE_CHANNEL == "nightly") {
ok(true, "Screenshots aren't captured on Nightlies");
return false;
}
// Don't run pre-defined sets (e.g. primaryUI) on try, require MOZSCREENSHOTS_SETS.
// The job is only scheduled on specific repos:
// https://dxr.mozilla.org/build-central/search?q=MOCHITEST_BC_SCREENSHOTS
let capture = !AppConstants.SOURCE_REVISION_URL.includes("/try/rev/");
if (!capture) {
ok(true, "Capturing is disabled for this REPO. You may need to use MOZSCREENSHOTS_SETS");
}
return capture;
}
add_task(setup);