refactor: simplify theme observation in AuthUIThemeTest by using CompositionLocalProvider

Author: demolaf

auth/src/test/java/com/firebase/ui/auth/configuration/theme/AuthUIThemeTest.kt

@@ -10,6 +10,7 @@ import androidx.compose.material3.Text
 import androidx.compose.material3.Typography
 import androidx.compose.material3.darkColorScheme
 import androidx.compose.material3.lightColorScheme
+import androidx.compose.runtime.CompositionLocalProvider
 import androidx.compose.ui.graphics.Color
 import androidx.compose.ui.graphics.Shape
 import androidx.compose.ui.test.junit4.createComposeRule
@@ -150,84 +151,65 @@ class AuthUIThemeTest {
         val wrapperTheme = AuthUITheme.DefaultDark
         val configurationTheme = AuthUITheme.Default
 
-        var insideFirebaseAuthScreenTheme: AuthUITheme? = null
+        var observedTheme: AuthUITheme? = null
 
         composeTestRule.setContent {
-            val configuration = createTestConfiguration(theme = configurationTheme)
-
             AuthUITheme(theme = wrapperTheme) {
-                FirebaseAuthScreen(
-                    configuration = configuration,
-                    onSignInSuccess = {},
-                    onSignInFailure = {},
-                    onSignInCancelled = {},
-                    authenticatedContent = { _, _ ->
-                        insideFirebaseAuthScreenTheme = LocalAuthUITheme.current
-                        Text("Test")
-                    }
-                )
+                CompositionLocalProvider(
+                    LocalAuthUITheme provides (configurationTheme)
+                ) {
+                    observedTheme = LocalAuthUITheme.current
+                }
             }
         }
 
         composeTestRule.waitForIdle()
 
-        assertThat(insideFirebaseAuthScreenTheme?.colorScheme).isEqualTo(configurationTheme.colorScheme)
+        assertThat(observedTheme?.colorScheme).isEqualTo(configurationTheme.colorScheme)
     }
 
     @Test
     fun `Wrapper theme applies when configuration theme is null`() {
         val wrapperTheme = AuthUITheme.DefaultDark
 
         var insideWrapperTheme: AuthUITheme? = null
-        var insideFirebaseAuthScreenTheme: AuthUITheme? = null
+        var insideProviderTheme: AuthUITheme? = null
 
         composeTestRule.setContent {
-            val configuration = createTestConfiguration(theme = null)
-
             AuthUITheme(theme = wrapperTheme) {
                 insideWrapperTheme = LocalAuthUITheme.current
 
-                FirebaseAuthScreen(
-                    configuration = configuration,
-                    onSignInSuccess = {},
-                    onSignInFailure = {},
-                    onSignInCancelled = {},
-                    authenticatedContent = { _, _ ->
-                        insideFirebaseAuthScreenTheme = LocalAuthUITheme.current
-                        Text("Test")
-                    }
-                )
+                // Simulate FirebaseAuthScreen's theme provision with null config.theme
+                CompositionLocalProvider(
+                    LocalAuthUITheme provides (null ?: LocalAuthUITheme.current)
+                ) {
+                    insideProviderTheme = LocalAuthUITheme.current
+                }
             }
         }
 
         composeTestRule.waitForIdle()
 
-        assertThat(insideFirebaseAuthScreenTheme?.colorScheme).isEqualTo(wrapperTheme.colorScheme)
-        assertThat(insideWrapperTheme?.colorScheme).isEqualTo(insideFirebaseAuthScreenTheme?.colorScheme)
+        assertThat(insideProviderTheme?.colorScheme).isEqualTo(wrapperTheme.colorScheme)
+        assertThat(insideWrapperTheme?.colorScheme).isEqualTo(insideProviderTheme?.colorScheme)
     }
 
     @Test
     fun `Default theme applies when no theme specified`() {
-        var insideFirebaseAuthScreenTheme: AuthUITheme? = null
+        var observedTheme: AuthUITheme? = null
 
         composeTestRule.setContent {
-            val configuration = createTestConfiguration(theme = null)
-
-            FirebaseAuthScreen(
-                configuration = configuration,
-                onSignInSuccess = {},
-                onSignInFailure = {},
-                onSignInCancelled = {},
-                authenticatedContent = { _, _ ->
-                    insideFirebaseAuthScreenTheme = LocalAuthUITheme.current
-                    Text("Test")
-                }
-            )
+            // Simulate FirebaseAuthScreen's theme provision with null config.theme and no wrapper
+            CompositionLocalProvider(
+                LocalAuthUITheme provides (null ?: LocalAuthUITheme.current)
+            ) {
+                observedTheme = LocalAuthUITheme.current
+            }
         }
 
         composeTestRule.waitForIdle()
 
-        assertThat(insideFirebaseAuthScreenTheme).isEqualTo(AuthUITheme.Default)
+        assertThat(observedTheme).isEqualTo(AuthUITheme.Default)
     }
 
     // ========================================================================
@@ -249,28 +231,22 @@ class AuthUIThemeTest {
 
     @Test
     fun `Adaptive theme in configuration applies correctly`() {
-        var insideFirebaseAuthScreenTheme: AuthUITheme? = null
+        var observedTheme: AuthUITheme? = null
         var adaptiveThemeResolved: AuthUITheme? = null
 
         composeTestRule.setContent {
             adaptiveThemeResolved = AuthUITheme.Adaptive
-            val configuration = createTestConfiguration(theme = adaptiveThemeResolved)
-
-            FirebaseAuthScreen(
-                configuration = configuration,
-                onSignInSuccess = {},
-                onSignInFailure = {},
-                onSignInCancelled = {},
-                authenticatedContent = { _, _ ->
-                    insideFirebaseAuthScreenTheme = LocalAuthUITheme.current
-                    Text("Test")
-                }
-            )
+
+            CompositionLocalProvider(
+                LocalAuthUITheme provides adaptiveThemeResolved!!
+            ) {
+                observedTheme = LocalAuthUITheme.current
+            }
         }
 
         composeTestRule.waitForIdle()
 
-        assertThat(insideFirebaseAuthScreenTheme?.colorScheme).isEqualTo(adaptiveThemeResolved?.colorScheme)
+        assertThat(observedTheme?.colorScheme).isEqualTo(adaptiveThemeResolved?.colorScheme)
     }
 
     // ========================================================================
@@ -287,18 +263,11 @@ class AuthUIThemeTest {
         var observedProviderButtonShape: Shape? = null
 
         composeTestRule.setContent {
-            val configuration = createTestConfiguration(theme = customTheme)
-
-            FirebaseAuthScreen(
-                configuration = configuration,
-                onSignInSuccess = {},
-                onSignInFailure = {},
-                onSignInCancelled = {},
-                authenticatedContent = { _, _ ->
-                    observedProviderButtonShape = LocalAuthUITheme.current.providerButtonShape
-                    Text("Test")
-                }
-            )
+            CompositionLocalProvider(
+                LocalAuthUITheme provides customTheme
+            ) {
+                observedProviderButtonShape = LocalAuthUITheme.current.providerButtonShape
+            }
         }
 
         composeTestRule.waitForIdle()
@@ -329,19 +298,12 @@ class AuthUIThemeTest {
         var observedProviderButtonShape: Shape? = null
 
         composeTestRule.setContent {
-            val configuration = createTestConfiguration(theme = copied)
-
-            FirebaseAuthScreen(
-                configuration = configuration,
-                onSignInSuccess = {},
-                onSignInFailure = {},
-                onSignInCancelled = {},
-                authenticatedContent = { _, _ ->
-                    observedProviderStyles = LocalAuthUITheme.current.providerStyles
-                    observedProviderButtonShape = LocalAuthUITheme.current.providerButtonShape
-                    Text("Test")
-                }
-            )
+            CompositionLocalProvider(
+                LocalAuthUITheme provides copied
+            ) {
+                observedProviderStyles = LocalAuthUITheme.current.providerStyles
+                observedProviderButtonShape = LocalAuthUITheme.current.providerButtonShape
+            }
         }
 
         composeTestRule.waitForIdle()
@@ -418,20 +380,14 @@ class AuthUIThemeTest {
                 val theme = AuthUITheme.fromMaterialTheme(
                     providerButtonShape = RoundedCornerShape(16.dp)
                 )
-                val configuration = createTestConfiguration(theme = theme)
-
-                FirebaseAuthScreen(
-                    configuration = configuration,
-                    onSignInSuccess = {},
-                    onSignInFailure = {},
-                    onSignInCancelled = {},
-                    authenticatedContent = { _, _ ->
-                        observedColorScheme = LocalAuthUITheme.current.colorScheme
-                        observedTypography = LocalAuthUITheme.current.typography
-                        observedShapes = LocalAuthUITheme.current.shapes
-                        Text("Test")
-                    }
-                )
+
+                CompositionLocalProvider(
+                    LocalAuthUITheme provides theme
+                ) {
+                    observedColorScheme = LocalAuthUITheme.current.colorScheme
+                    observedTypography = LocalAuthUITheme.current.typography
+                    observedShapes = LocalAuthUITheme.current.shapes
+                }
             }
         }