Bug 1578885 - use AttributionCode directly from ClientEnvironment, rather than Telemetry r=leplatrem

Differential Revision: https://phabricator.services.mozilla.com/D45433

--HG--
extra : moz-landing-system : lando

toolkit/components/utils/ClientEnvironment.jsm

@@ -42,6 +42,12 @@ ChromeUtils.defineModuleGetter(
   "resource://gre/modules/FirstStartup.jsm"
 );
 
+ChromeUtils.defineModuleGetter(
+  this,
+  "AttributionCode",
+  "resource:///modules/AttributionCode.jsm"
+);
+
 var EXPORTED_SYMBOLS = ["ClientEnvironmentBase"];
 
 /**
@@ -217,4 +223,8 @@ class ClientEnvironmentBase {
       return rv;
     })();
   }
+
+  static get attribution() {
+    return AttributionCode.getAttrDataAsync();
+  }
 }

toolkit/components/utils/test/unit/test_ClientEnvironment.js

@@ -44,7 +44,7 @@ add_task(async () => {
     equal(os.darwinVersion, null, "Darwin version should not be set");
   }
 
-  // if on Mac, Mac versions should be set, and Windows versions should not b e
+  // if on Mac, Mac versions should be set, and Windows versions should not be
   if (os.isMac) {
     equal(typeof os.macVersion, "number", "Mac version should be a number");
     equal(
@@ -72,3 +72,15 @@ add_task(async () => {
     );
   }
 });
+
+add_task(async () => {
+  try {
+    await ClientEnvironmentBase.attribution;
+  } catch (ex) {
+    equal(
+      ex.name,
+      "NS_ERROR_FILE_NOT_FOUND",
+      "Test environment does not have attribution data"
+    );
+  }
+});