Address testResendCode comment and situation

AzureAD · Nov 21, 2024 · 242b66c · 242b66c
1 parent 6a1710d
commit 242b66c
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/...est/java/com/microsoft/identity/client/e2e/tests/network/nativeauth/SignUpEmailOTPTest.kt b/...est/java/com/microsoft/identity/client/e2e/tests/network/nativeauth/SignUpEmailOTPTest.kt
@@ -78,13 +78,18 @@ class SignUpEmailOTPTest : NativeAuthPublicClientApplicationAnotherAbstractTest(
         config = getConfig(defaultConfigType)
         application = setupPCA(config, defaultChallengeTypes)
 
-        runBlocking { // Running with runBlocking to avoid default 10 second execution timeout.
-            val user = tempEmailApi.generateRandomEmailAddress()
-            val signUpResult = application.signUp(user)
-            assertResult<SignUpResult.CodeRequired>(signUpResult)
-            val codeRequiredState = (signUpResult as SignUpResult.CodeRequired).nextState
-            val resendCodeResult = codeRequiredState.resendCode()
-            assertResult<SignUpResendCodeResult.Success>(resendCodeResult)
+        retryOperation {
+            runBlocking { // Running with runBlocking to avoid default 10 second execution timeout.
+                val user = tempEmailApi.generateRandomEmailAddress()
+                val signUpResult = application.signUp(user)
+                assertResult<SignUpResult.CodeRequired>(signUpResult)
+                val otp1 = tempEmailApi.retrieveCodeFromInbox(user)
+                val codeRequiredState = (signUpResult as SignUpResult.CodeRequired).nextState
+                val resendCodeResult = codeRequiredState.resendCode()
+                assertResult<SignUpResendCodeResult.Success>(resendCodeResult)
+                val otp2 = tempEmailApi.retrieveCodeFromInbox(user)
+                Assert.assertNotEquals(otp1, otp2)
+            }
         }
     }