fix

Author: Lyokone

auth/src/test/java/com/firebase/ui/auth/compose/FirebaseAuthUIAuthStateTest.kt

@@ -198,10 +198,8 @@ class FirebaseAuthUIAuthStateTest {
     }
 
     @Test
-    fun `authStateFlow() emits Success even with unverified email for now`() = runBlocking {
-        // Given a signed-in user with unverified email
-        // Note: The current implementation checks for password provider, which might not be
-        // matched properly due to mocking limitations. This test verifies current behavior.
+    fun `authStateFlow() emits RequiresEmailVerification for unverified password users`() = runBlocking {
+        // Given a signed-in user with unverified email using password authentication
         val mockProviderData = mock(UserInfo::class.java)
         `when`(mockProviderData.providerId).thenReturn("password")
 
@@ -213,10 +211,11 @@ class FirebaseAuthUIAuthStateTest {
         // When collecting auth state flow
         val state = authUI.authStateFlow().first()
 
-        // Then it should emit Success state (current behavior with mocked data)
-        assertThat(state).isInstanceOf(AuthState.Success::class.java)
-        val successState = state as AuthState.Success
-        assertThat(successState.user).isEqualTo(mockFirebaseUser)
+        // Then it should emit RequiresEmailVerification state
+        assertThat(state).isInstanceOf(AuthState.RequiresEmailVerification::class.java)
+        val verificationState = state as AuthState.RequiresEmailVerification
+        assertThat(verificationState.user).isEqualTo(mockFirebaseUser)
+        assertThat(verificationState.email).isEqualTo("[email protected]")
     }
 
     @Test

auth/src/test/java/com/firebase/ui/auth/compose/ui/screens/MfaEnrollmentScreenTest.kt

@@ -63,6 +63,9 @@ class MfaEnrollmentScreenTest {
     @Mock
     private lateinit var mockFirebaseApp: FirebaseApp
 
+    @Mock
+    private lateinit var mockMultiFactor: com.google.firebase.auth.MultiFactor
+
     private lateinit var capturedState: MfaEnrollmentContentState
 
     @Before
@@ -72,6 +75,8 @@ class MfaEnrollmentScreenTest {
         `when`(mockAuth.app).thenReturn(mockFirebaseApp)
         `when`(mockFirebaseApp.name).thenReturn("TestApp")
         `when`(mockUser.email).thenReturn("[email protected]")
+        `when`(mockUser.multiFactor).thenReturn(mockMultiFactor)
+        `when`(mockMultiFactor.enrolledFactors).thenReturn(emptyList())
     }
 
     @Test

e2eTest/src/test/java/com/firebase/ui/auth/compose/ui/screens/MfaEnrollmentScreenTest.kt

@@ -73,6 +73,9 @@ class MfaEnrollmentScreenTest {
     @Mock
     private lateinit var mockFirebaseUser: FirebaseUser
 
+    @Mock
+    private lateinit var mockMultiFactor: com.google.firebase.auth.MultiFactor
+
     @Before
     fun setUp() {
         MockitoAnnotations.openMocks(this)
@@ -100,6 +103,8 @@ class MfaEnrollmentScreenTest {
         // Use mock user instead of real Firebase user
         `when`(mockFirebaseUser.email).thenReturn("[email protected]")
         `when`(mockFirebaseUser.uid).thenReturn("test-uid-123")
+        `when`(mockFirebaseUser.multiFactor).thenReturn(mockMultiFactor)
+        `when`(mockMultiFactor.enrolledFactors).thenReturn(emptyList())
         testUser = mockFirebaseUser
     }