Fix State Management Tests

aws-amplify · Aug 28, 2024 · 2cdfe15 · 2cdfe15
1 parent 9d0989d
commit 2cdfe15
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/auth/__tests__/providers/cognito/signInStateManagement.test.ts b/packages/auth/__tests__/providers/cognito/signInStateManagement.test.ts
@@ -42,6 +42,10 @@ describe('local sign-in state management tests', () => {
 		cognitoUserPoolsTokenProvider.setAuthConfig(authConfig);
 	});
 
+	afterAll(() => {
+		mockedGetCurrentUser.mockRestore();
+	});
+
 	test('local state management should return state after signIn returns a ChallengeName', async () => {
 		const handleUserSRPAuthflowSpy = jest
 			.spyOn(signInHelpers, 'handleUserSRPAuthFlow')