test: map reference breaking

src/lifecycleEvents.ts

@@ -15,9 +15,9 @@ import { Logger } from './logger/logger';
 
 // Data of any type can be passed to the callback. Can be cast to any type that is given in emit().
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-type callback = (data: any) => Promise<void>;
+export type callback = (data: any) => Promise<void>;
 type ListenerMap = Map<string, callback>;
-type UniqueListenerMap = Map<string, ListenerMap>;
+export type UniqueListenerMap = Map<string, ListenerMap>;
 
 declare const global: {
   salesforceCoreLifecycle?: Lifecycle;
@@ -51,7 +51,7 @@ export class Lifecycle {
 
   private constructor(
     private readonly listeners: Dictionary<callback[]> = {},
-    private readonly uniqueListeners: Map<string, Map<string, callback>> = new Map<string, Map<string, callback>>()
+    private readonly uniqueListeners: UniqueListenerMap = new Map<string, Map<string, callback>>()
   ) {}
 
   /**
@@ -238,5 +238,5 @@ export class Lifecycle {
 }
 
 const cloneListeners: (listeners: ListenerMap) => ListenerMap = (listeners) => new Map(Array.from(listeners.entries()));
-const cloneUniqueListeners = (uniqueListeners: UniqueListenerMap): UniqueListenerMap =>
+export const cloneUniqueListeners = (uniqueListeners: UniqueListenerMap): UniqueListenerMap =>
   new Map(Array.from(uniqueListeners.entries()).map(([key, value]) => [key, cloneListeners(value)]));

test/unit/lifecycleEventsTest.ts

@@ -9,7 +9,7 @@
 import { Duration, sleep } from '@salesforce/kit/lib/duration';
 import { spyMethod } from '@salesforce/ts-sinon';
 import * as chai from 'chai';
-import { Lifecycle } from '../../src/lifecycleEvents';
+import { Lifecycle, callback, cloneUniqueListeners } from '../../src/lifecycleEvents';
 import { TestContext } from '../../src/testSetup';
 import { Logger } from '../../src/logger/logger';
 
@@ -257,3 +257,29 @@ describe('lifecycleEvents', () => {
     lifecycle2.removeAllListeners('test7');
   });
 });
+
+describe('listener map cloning', () => {
+  const cb = (): Promise<void> => Promise.resolve();
+  it('clones map, breaking event name reference', () => {
+    const map1 = new Map<string, Map<string, callback>>();
+    map1.set('evt', new Map([['uniqueId', cb]]));
+
+    const map2 = cloneUniqueListeners(map1);
+    chai.expect(map2).to.deep.equal(map1);
+    map1.delete('evt');
+    chai.expect(map1.has('evt')).to.be.false;
+    chai.expect(map2.has('evt')).to.be.true;
+  });
+  it('clones map, breaking uniqueId reference', () => {
+    const map1 = new Map<string, Map<string, callback>>();
+    map1.set('evt', new Map([['uniqueId', cb]]));
+
+    const map2 = cloneUniqueListeners(map1);
+    chai.expect(map2).to.deep.equal(map1);
+    map2.get('evt')?.set('uniqueId2', cb);
+    chai.expect(map1.has('evt')).to.be.true;
+    chai.expect(map2.has('evt')).to.be.true;
+    chai.expect(map1.get('evt')?.has('uniqueId2')).to.be.false;
+    chai.expect(map2.get('evt')?.has('uniqueId2')).to.be.true;
+  });
+});