Update unit tests

aws · Sep 10, 2020 · 4ed57cf · 4ed57cf
1 parent 17bcd52
commit 4ed57cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/Amazon.AspNetCore.Identity.Cognito/CognitoUserManager.cs b/src/Amazon.AspNetCore.Identity.Cognito/CognitoUserManager.cs
@@ -198,7 +198,7 @@ public virtual Task<AuthFlowResponse> CheckPasswordAsync(TUser user, string pass
         /// <param name="code">The 2fa code to check</param>
         /// <param name="authWorkflowSessionId">The ongoing Cognito authentication workflow id.</param>
         /// <returns>The <see cref="Task"/> that represents the asynchronous operation, containing the AuthFlowResponse object linked to that authentication workflow.</returns>
-        public virtual Task<AuthFlowResponse> RespondToTwoFactorChallengeAsync2(TUser user, string code, string authWorkflowSessionId)
+        public virtual Task<AuthFlowResponse> RespondToTwoFactorChallengeAsync(TUser user, string code, string authWorkflowSessionId)
         {
             return RespondToTwoFactorChallengeAsync(user, code, ChallengeNameType.SMS_MFA, authWorkflowSessionId);
         }

diff --git a/test/unit/Amazon.AspNetCore.Identity.Cognito.Tests/CognitoSigninManagerTests.cs b/test/unit/Amazon.AspNetCore.Identity.Cognito.Tests/CognitoSigninManagerTests.cs
@@ -144,7 +144,7 @@ public async void Test_GivenAUserWith2FA_WhenRespondToTwoFactorChallengeWithCorr
             var context = MockUtils.MockContext(cognitoUser, IdentityConstants.TwoFactorUserIdScheme);
             contextAccessorMock.Setup(a => a.HttpContext).Returns(context).Verifiable();
 
-            var authFlowResponse = new AuthFlowResponse("sessionId", null, null, null, null);
+            var authFlowResponse = new AuthFlowResponse("sessionId", null, ChallengeNameType.SMS_MFA, null, null);
 
             userManagerMock.Setup(mock => mock.FindByIdAsync(It.IsAny<string>())).Returns(Task.FromResult(cognitoUser)).Verifiable();
             userManagerMock.Setup(mock => mock.RespondToTwoFactorChallengeAsync(It.IsAny<CognitoUser>(), It.IsAny<string>(), It.IsAny<string>()))