summaryrefslogtreecommitdiffstats
path: root/toolkit/modules/tests/xpcshell/test_firstStartup.js
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--toolkit/modules/tests/xpcshell/test_firstStartup.js82
1 files changed, 82 insertions, 0 deletions
diff --git a/toolkit/modules/tests/xpcshell/test_firstStartup.js b/toolkit/modules/tests/xpcshell/test_firstStartup.js
new file mode 100644
index 0000000000..42d9cb2757
--- /dev/null
+++ b/toolkit/modules/tests/xpcshell/test_firstStartup.js
@@ -0,0 +1,82 @@
+"use strict";
+
+const { AppConstants } = ChromeUtils.importESModule(
+ "resource://gre/modules/AppConstants.sys.mjs"
+);
+const { FirstStartup } = ChromeUtils.importESModule(
+ "resource://gre/modules/FirstStartup.sys.mjs"
+);
+const { updateAppInfo } = ChromeUtils.importESModule(
+ "resource://testing-common/AppInfo.sys.mjs"
+);
+
+const PREF_TIMEOUT = "first-startup.timeout";
+
+add_setup(function test_setup() {
+ // FOG needs a profile directory to put its data in.
+ do_get_profile();
+
+ // FOG needs to be initialized in order for data to flow.
+ Services.fog.initializeFOG();
+});
+
+add_task(async function test_success() {
+ updateAppInfo();
+
+ let submissionPromise;
+
+ if (AppConstants.MOZ_NORMANDY || AppConstants.MOZ_UPDATE_AGENT) {
+ submissionPromise = new Promise(resolve => {
+ GleanPings.firstStartup.testBeforeNextSubmit(() => {
+ Assert.equal(FirstStartup.state, FirstStartup.SUCCESS);
+ Assert.equal(
+ Glean.firstStartup.statusCode.testGetValue(),
+ FirstStartup.SUCCESS
+ );
+ resolve();
+ });
+ });
+ } else {
+ submissionPromise = new Promise(resolve => {
+ GleanPings.firstStartup.testBeforeNextSubmit(() => {
+ Assert.equal(FirstStartup.state, FirstStartup.UNSUPPORTED);
+ Assert.equal(
+ Glean.firstStartup.statusCode.testGetValue(),
+ FirstStartup.UNSUPPORTED
+ );
+ resolve();
+ });
+ });
+ }
+
+ FirstStartup.init();
+ await submissionPromise;
+});
+
+add_task(async function test_timeout() {
+ updateAppInfo();
+ Services.prefs.setIntPref(PREF_TIMEOUT, 0);
+
+ let submissionPromise;
+
+ if (AppConstants.MOZ_NORMANDY || AppConstants.MOZ_UPDATE_AGENT) {
+ submissionPromise = new Promise(resolve => {
+ GleanPings.firstStartup.testBeforeNextSubmit(() => {
+ Assert.equal(FirstStartup.state, FirstStartup.TIMED_OUT);
+ Assert.ok(Glean.firstStartup.elapsed.testGetValue() > 0);
+ resolve();
+ });
+ });
+ } else {
+ submissionPromise = new Promise(resolve => {
+ GleanPings.firstStartup.testBeforeNextSubmit(() => {
+ Assert.equal(FirstStartup.state, FirstStartup.UNSUPPORTED);
+ Assert.equal(Glean.firstStartup.elapsed.testGetValue(), 0);
+ resolve();
+ });
+ });
+ }
+
+ FirstStartup.init();
+ await submissionPromise;
+});