/* Any copyright is dedicated to the Public Domain. http://creativecommons.org/publicdomain/zero/1.0/ */ "use strict"; ChromeUtils.defineESModuleGetters(this, { MockRegistrar: "resource://testing-common/MockRegistrar.sys.mjs", sinon: "resource://testing-common/Sinon.sys.mjs", ShellService: "moz-src:///browser/components/shell/ShellService.sys.mjs", TaskbarTabsPin: "resource:///modules/taskbartabs/TaskbarTabsPin.sys.mjs", TaskbarTabsRegistry: "resource:///modules/taskbartabs/TaskbarTabsRegistry.sys.mjs", XPCOMUtils: "resource://gre/modules/XPCOMUtils.sys.mjs", }); XPCOMUtils.defineLazyServiceGetters(this, { imgTools: ["@mozilla.org/image/tools;1", Ci.imgITools], }); const gRegistry = new TaskbarTabsRegistry(); const patchedSpy = sinon.stub(); gRegistry.on(TaskbarTabsRegistry.events.patched, patchedSpy); sinon.stub(ShellService, "writeShortcutIcon").resolves(); sinon.stub(ShellService, "createLinuxDesktopEntry").resolves(); sinon.stub(ShellService, "deleteLinuxDesktopEntry").resolves(); sinon.stub(TaskbarTabsPin, "_getLocalization").returns({ formatValue(msg) { // Slash must also be sanitized, so it should appear as '_' in paths. return `[formatValue/${msg}]`; }, }); registerCleanupFunction(() => { sinon.restore(); }); // Favicons are written to the profile directory, ensure it exists. do_get_profile(); function checkCreateLinuxDesktopEntryCall(aTaskbarTab) { Assert.equal( ShellService.createLinuxDesktopEntry.callCount, 1, "createLinuxDesktopEntry was called once." ); Assert.equal( ShellService.createLinuxDesktopEntry.firstCall.args[0], "org.mozilla.firefox.webapp-" + aTaskbarTab.id, "Correct application ID was specified." ); Assert.equal( ShellService.createLinuxDesktopEntry.firstCall.args[1], aTaskbarTab.name, "Correct name was specified." ); Assert.ok( ShellService.createLinuxDesktopEntry.firstCall.args[2].includes( "-taskbar-tab" ), "Reasonable arguments were specified." ); Assert.equal(patchedSpy.callCount, 1, "A single patched event was emitted"); Assert.equal( aTaskbarTab.shortcutRelativePath, "org.mozilla.firefox.webapp-" + aTaskbarTab.id + ".desktop", "Correct relative path is saved to the taskbar tab" ); } add_task(async function test_pinCreatesDesktopEntry() { const parsedURI = Services.io.newURI("https://www.example.com"); const taskbarTab = createTaskbarTab(gRegistry, parsedURI, 0); sinon.resetHistory(); await TaskbarTabsPin.pinTaskbarTab(taskbarTab, gRegistry); checkCreateLinuxDesktopEntryCall(taskbarTab); gRegistry.removeTaskbarTab(taskbarTab.id); }); // This is more of a Windows problem, so make sure it doesn't affect Linux. add_task(async function test_pinUnusualName() { const parsedURI = Services.io.newURI("https://another.test"); const invalidTaskbarTab = createTaskbarTab(gRegistry, parsedURI, 0, { manifest: { name: "** :\t\r\n \\\\ >> Not a valid. filename??! << // |||: **.", }, }); sinon.resetHistory(); await TaskbarTabsPin.pinTaskbarTab(invalidTaskbarTab, gRegistry); checkCreateLinuxDesktopEntryCall(invalidTaskbarTab); gRegistry.removeTaskbarTab(invalidTaskbarTab.id); }); add_task(async function test_unpin() { const parsedURI = Services.io.newURI("https://example.com"); const tt = createTaskbarTab(gRegistry, parsedURI, 0); gRegistry.patchTaskbarTab(tt, { shortcutRelativePath: "this.is.an.app.id.desktop", }); sinon.resetHistory(); await TaskbarTabsPin.unpinTaskbarTab(tt, gRegistry); Assert.equal( ShellService.deleteLinuxDesktopEntry.callCount, 1, "deleteLinuxDesktopEntry was called once" ); Assert.deepEqual( ShellService.deleteLinuxDesktopEntry.firstCall.args, ["this.is.an.app.id"], "deleteLinuxDesktopEntry was called with the value in shortcutRelativePath without the extension" ); Assert.equal( tt.shortcutRelativePath, null, "Shortcut relative path was removed from the taskbar tab" ); Assert.equal(patchedSpy.callCount, 1, "A single patched event was emitted"); });