chore: Fix an integration test edge case while testing MFA/OTP codes

aws-amplify · Jan 28, 2025 · ae4d682 · ae4d682
1 parent 08da562
commit ae4d682
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/...c/androidTest/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginEmailMFATests.kt b/...c/androidTest/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginEmailMFATests.kt
@@ -75,8 +75,10 @@ class AWSCognitoAuthPluginEmailMFATests {
             { println("====== Subscription Established ======") },
             {
                 println("====== Received some MFA Info ======")
-                mfaCode = it.data.code
-                latch?.countDown()
+                if (it.data.username == userName)   {
+                    mfaCode = it.data.code
+                    latch?.countDown()
+                }
             },
             { println("====== Subscription Failed $it ======") },
             { }

diff --git a/...c/androidTest/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginUserAuthTests.kt b/...c/androidTest/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginUserAuthTests.kt
@@ -89,8 +89,10 @@ class AWSCognitoAuthPluginUserAuthTests {
             { println("====== Subscription Established ======") },
             {
                 println("====== Received some MFA Info ======")
-                otpCode = it.data.code
-                latch?.countDown()
+                if (it.data.username == userName)   {
+                    otpCode = it.data.code
+                    latch?.countDown()
+                }
             },
             { println("====== Subscription Failed $it ======") },
             { }