summaryrefslogtreecommitdiffstats
path: root/browser/tools/mozscreenshots/head.js
diff options
context:
space:
mode:
Diffstat (limited to 'browser/tools/mozscreenshots/head.js')
-rw-r--r--browser/tools/mozscreenshots/head.js67
1 files changed, 67 insertions, 0 deletions
diff --git a/browser/tools/mozscreenshots/head.js b/browser/tools/mozscreenshots/head.js
new file mode 100644
index 0000000000..a1876b9c30
--- /dev/null
+++ b/browser/tools/mozscreenshots/head.js
@@ -0,0 +1,67 @@
+/* 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 chromeRegistry = Cc["@mozilla.org/chrome/chrome-registry;1"].getService(
+ Ci.nsIChromeRegistry
+);
+const EXTENSION_DIR =
+ "chrome://mochitests/content/browser/browser/tools/mozscreenshots/mozscreenshots/extension/mozscreenshots/browser/";
+
+let TestRunner;
+
+async 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);
+
+ // Generate output so mozprocess knows we're still alive for the long session.
+ SimpleTest.requestCompleteLog();
+
+ info("installing extension temporarily");
+ let chromeURL = Services.io.newURI(EXTENSION_DIR);
+ let dir = chromeRegistry
+ .convertChromeURL(chromeURL)
+ .QueryInterface(Ci.nsIFileURL).file;
+ await AddonManager.installTemporaryAddon(dir);
+
+ info("Checking for mozscreenshots extension");
+
+ let aAddon = await AddonManager.getAddonByID("mozscreenshots@mozilla.org");
+ isnot(aAddon, null, "The mozscreenshots extension should be installed");
+ TestRunner = ChromeUtils.importESModule(
+ "resource://mozscreenshots/TestRunner.sys.mjs"
+ ).TestRunner;
+ TestRunner.initTest(this);
+}
+
+/**
+ * Used by pre-defined sets of configurations to decide whether to run for a build.
+ *
+ * This is not used by browser_screenshots.js which handles when MOZSCREENSHOTS_SETS is set.
+ *
+ * @returns {bool} whether to capture screenshots.
+ */
+function shouldCapture() {
+ if (Services.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;
+ }
+
+ return true;
+}
+
+add_setup(setup);