Update tests and run formatting

Author: riggaroo

core/network/src/main/java/com/android/developers/androidify/startup/FirebaseAppCheckInitializer.kt

@@ -36,13 +36,13 @@ class FirebaseAppCheckInitializer : Initializer<FirebaseAppCheck> {
             if (BuildConfig.DEBUG) {
                 Log.i("AndroidifyAppCheck", "Firebase debug")
                 installAppCheckProviderFactory(
-                    DebugAppCheckProviderFactory.getInstance()
+                    DebugAppCheckProviderFactory.getInstance(),
                 )
             } else {
                 Log.i("AndroidifyAppCheck", "Play integrity")
                 installAppCheckProviderFactory(
                     PlayIntegrityAppCheckProviderFactory.getInstance(),
-                    )
+                )
             }
         }
     }

core/testing/src/main/java/com/android/developers/testing/network/TestRemoteConfigDataSource.kt

@@ -29,9 +29,9 @@ class TestRemoteConfigDataSource(private val useGeminiNano: Boolean) : RemoteCon
     override fun imageModelName(): String {
         TODO("Not yet implemented")
     }
-
+    var backgroundVibeEnabled: Boolean = true
     override fun isBackgroundVibesFeatureEnabled(): Boolean {
-        return true
+        return backgroundVibeEnabled
     }
 
     override fun promptTextVerify(): String {

feature/creation/src/main/java/com/android/developers/androidify/creation/CreationScreen.kt

@@ -493,8 +493,8 @@ private fun MainCreationPane(
                     // Workaround for https://issuetracker.google.com/432431393
                     val showTextPrompt by remember {
                         derivedStateOf {
-                            pagerState.currentPage == PromptType.TEXT.ordinal
-                                    && pagerState.targetPage == pagerState.currentPage
+                            pagerState.currentPage == PromptType.TEXT.ordinal &&
+                                pagerState.targetPage == pagerState.currentPage
                         }
                     }
                     if (showTextPrompt) {

feature/creation/src/test/kotlin/com/android/developers/androidify/creation/CreationViewModelTest.kt

@@ -72,7 +72,7 @@ class CreationViewModelTest {
             viewModel.uiState.value.screenState,
         )
         assertEquals(false, viewModel.uiState.value.promptGenerationInProgress)
-        assertEquals( fakeUri, viewModel.uiState.value.imageUri)
+        assertEquals(fakeUri, viewModel.uiState.value.imageUri)
     }
 
     @Test
@@ -91,7 +91,7 @@ class CreationViewModelTest {
             viewModel.uiState.value.screenState,
         )
         assertEquals(false, viewModel.uiState.value.promptGenerationInProgress)
-        assertEquals( null, viewModel.uiState.value.imageUri)
+        assertEquals(null, viewModel.uiState.value.imageUri)
     }
 
     @Test

feature/results/src/main/java/com/android/developers/androidify/customize/ComposableBitmapRendererImpl.kt

@@ -22,7 +22,6 @@ import android.content.Context.DISPLAY_SERVICE
 import android.graphics.Bitmap
 import android.graphics.SurfaceTexture
 import android.hardware.display.DisplayManager
-import android.hardware.display.VirtualDisplay
 import android.view.Display
 import android.view.Surface
 import android.view.ViewGroup

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeExportViewModel.kt

@@ -95,9 +95,9 @@ class CustomizeExportViewModel @AssistedInject constructor(
                 toolState = mapOf(
                     CustomizeTool.Size to AspectRatioToolState(),
                     CustomizeTool.Background to BackgroundToolState(
-                        options = backgrounds
+                        options = backgrounds,
                     ),
-                )
+                ),
             )
         }
         loadInitialBitmap(resultImageUrl)

feature/results/src/main/java/com/android/developers/androidify/customize/CustomizeState.kt

@@ -60,7 +60,7 @@ data class BackgroundToolState(
         BackgroundOption.None,
         BackgroundOption.Plain,
         BackgroundOption.Lightspeed,
-        BackgroundOption.IO
+        BackgroundOption.IO,
     ),
 ) : ToolState
 

feature/results/src/test/kotlin/com/android/developers/androidify/customize/CustomizeViewModelTest.kt

@@ -21,6 +21,7 @@ import android.net.Uri
 import androidx.test.core.app.ApplicationProvider
 import com.android.developers.testing.data.TestFileProvider
 import com.android.developers.testing.data.bitmapSample
+import com.android.developers.testing.network.TestRemoteConfigDataSource
 import com.android.developers.testing.repository.FakeImageGenerationRepository
 import com.android.developers.testing.util.FakeComposableBitmapRenderer
 import com.android.developers.testing.util.MainDispatcherRule
@@ -52,13 +53,16 @@ class CustomizeViewModelTest {
 
     @Before
     fun setup() {
+        val remoteConfigDataSource = TestRemoteConfigDataSource(true)
+        remoteConfigDataSource.backgroundVibeEnabled = false
         viewModel = CustomizeExportViewModel(
             fakeUri,
             originalFakeUri,
             FakeImageGenerationRepository(),
             composableBitmapRenderer = FakeComposableBitmapRenderer(),
             application = ApplicationProvider.getApplicationContext(),
             localFileProvider = TestFileProvider(),
+            remoteConfigDataSource = remoteConfigDataSource,
         )
     }
 
@@ -82,13 +86,16 @@ class CustomizeViewModelTest {
 
     @Test
     fun setArgumentsWithPrompt() = runTest {
+        val remoteConfigDataSource = TestRemoteConfigDataSource(true)
+        remoteConfigDataSource.backgroundVibeEnabled = false
         val viewModel = CustomizeExportViewModel(
             fakeUri,
             null,
             FakeImageGenerationRepository(),
             composableBitmapRenderer = FakeComposableBitmapRenderer(),
             application = ApplicationProvider.getApplicationContext(),
             localFileProvider = TestFileProvider(),
+            remoteConfigDataSource = remoteConfigDataSource,
         )
         assertEquals(
             CustomizeExportState(
@@ -141,6 +148,7 @@ class CustomizeViewModelTest {
             composableBitmapRenderer = FakeComposableBitmapRenderer(),
             application = ApplicationProvider.getApplicationContext(),
             localFileProvider = TestFileProvider(),
+            remoteConfigDataSource = TestRemoteConfigDataSource(false),
         )
         val values = mutableListOf<CustomizeExportState>()
         // Launch collector on the backgroundScope directly to use runTest's scheduler
@@ -190,4 +198,23 @@ class CustomizeViewModelTest {
         assertTrue { !values[values.lastIndex].showImageEditProgress }
         assertNull(values.last().exportImageCanvas.imageWithEdit)
     }
+
+    @Test
+    fun remoteConfigDataSource_BackgroundVibesFeatureEnabled_ContainsVibeList() = runTest {
+        val remoteConfigDataSource = TestRemoteConfigDataSource(true)
+        remoteConfigDataSource.backgroundVibeEnabled = true
+        val viewModel = CustomizeExportViewModel(
+            fakeUri,
+            null,
+            FakeImageGenerationRepository(),
+            composableBitmapRenderer = FakeComposableBitmapRenderer(),
+            application = ApplicationProvider.getApplicationContext(),
+            localFileProvider = TestFileProvider(),
+            remoteConfigDataSource = remoteConfigDataSource,
+        )
+        val state = viewModel.state.value.toolState[CustomizeTool.Background] as BackgroundToolState
+
+        assertTrue(state.options.size > 5)
+        assertTrue(state.options.any { it.aiBackground })
+    }
 }

gradle/libs.versions.toml

@@ -34,7 +34,7 @@ hiltAndroid = "2.56.2"
 hiltLifecycleViewmodel = "1.0.0-alpha03"
 hiltNavigationCompose = "1.2.0"
 junit = "4.13.2"
-junitVersion = "1.2.1"
+junitVersion = "1.3.0"
 kotlin = "2.2.0"
 ksp = "2.2.0-2.0.2"
 kotlinxCoroutines = "1.10.2"