gecko-dev/browser/components/newtab/test/xpcshell/test_AboutWelcomeAttribution.js

38 lines
1.3 KiB
JavaScript

/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/
*/
"use strict";
const { AboutWelcomeChild } = ChromeUtils.import(
"resource:///actors/AboutWelcomeChild.jsm"
);
const { sinon } = ChromeUtils.import("resource://testing-common/Sinon.jsm");
const TEST_ATTRIBUTION_DATA = {
source: "addons.mozilla.org",
medium: "referral",
campaign: "non-fx-button",
content: "iridium%40particlecore.github.io",
};
add_task(async function test_handleAddonInfoNotFound() {
let AWChild = new AboutWelcomeChild();
const stub = sinon.stub(AWChild, "getAddonInfo").resolves(null);
let result = await AWChild.formatAttributionData(TEST_ATTRIBUTION_DATA);
equal(stub.callCount, 1, "Call was made");
equal(result.template, undefined, "No template returned");
});
add_task(async function test_formatAttributionData() {
let AWChild = new AboutWelcomeChild();
const TEST_ADDON_INFO = {
name: "Test Add-on",
url: "https://test.xpi",
iconURL: "http://test.svg",
};
sinon.stub(AWChild, "getAddonInfo").resolves(TEST_ADDON_INFO);
let result = await AWChild.formatAttributionData(TEST_ATTRIBUTION_DATA);
equal(result.template, "return_to_amo", "RTAMO template returned");
equal(result.extraProps, TEST_ADDON_INFO, "AddonInfo returned");
});