Improve test coverage.

google · Sep 28, 2024 · 9c0e918 · 9c0e918
1 parent b835233
commit 9c0e918
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/...les/reader-revenue-manager/components/dashboard/RRMSetupSuccessSubtleNotification.test.js b/...les/reader-revenue-manager/components/dashboard/RRMSetupSuccessSubtleNotification.test.js
@@ -87,6 +87,8 @@ describe( 'RRMSetupSuccessSubtleNotification', () => {
 		'^/google-site-kit/v1/modules/reader-revenue-manager/data/settings'
 	);
 
+	const setValueMock = jest.fn();
+
 	beforeEach( () => {
 		registry = createTestRegistry();
 
@@ -99,7 +101,6 @@ describe( 'RRMSetupSuccessSubtleNotification', () => {
 		] );
 
 		useQueryArg.mockImplementation( ( arg ) => {
-			const setValueMock = jest.fn();
 			switch ( arg ) {
 				case 'notification':
 					return [ 'authentication_success', setValueMock ];
@@ -113,6 +114,8 @@ describe( 'RRMSetupSuccessSubtleNotification', () => {
 	} );
 
 	afterEach( () => {
+		setValueMock.mockClear();
+		useQueryArg.mockClear();
 		global.open.mockClear();
 	} );
 
@@ -161,10 +164,6 @@ describe( 'RRMSetupSuccessSubtleNotification', () => {
 
 			const dismissElement = getByText( dismissText );
 			expect( dismissElement ).toBeInTheDocument();
-
-			act( () => {
-				fireEvent.click( ctaElement );
-			} );
 		}
 	);
 
@@ -208,6 +207,9 @@ describe( 'RRMSetupSuccessSubtleNotification', () => {
 			act( () => {
 				fireEvent.click( dismissElement );
 			} );
+
+			expect( setValueMock ).toHaveBeenCalledTimes( 2 );
+			expect( setValueMock ).toHaveBeenCalledWith( undefined );
 		}
 	);