unit test update removeAllListeners

aws-amplify · Jul 6, 2021 · ff26889 · ff26889
1 parent 5d565fb
commit ff26889
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/storage/__tests__/providers/AWSS3Provider-unit-test.ts b/packages/storage/__tests__/providers/AWSS3Provider-unit-test.ts
@@ -235,9 +235,11 @@ describe('StorageProvider test', () => {
 				return Promise.resolve(credentials);
 			});
 			const mockCallback = jest.fn();
+			const mockRemoveAllListeners = jest.fn();
 			const mockEventEmitter = {
 				emit: jest.fn(),
 				on: jest.fn(),
+				removeAllListeners: mockRemoveAllListeners,
 			};
 			jest
 				.spyOn(events, 'EventEmitter')
@@ -265,6 +267,7 @@ describe('StorageProvider test', () => {
 			// Manully invoke it for testing
 			emitterOnFn('arg');
 			expect(mockCallback).toBeCalledWith('arg');
+			expect(mockRemoveAllListeners).toHaveBeenCalled();
 		});
 
 		test('get object with incorrect progressCallback type', async () => {
@@ -275,6 +278,7 @@ describe('StorageProvider test', () => {
 			const mockEventEmitter = {
 				emit: jest.fn(),
 				on: jest.fn(),
+				removeAllListeners: jest.fn(),
 			};
 			jest
 				.spyOn(events, 'EventEmitter')