Migrate mainline module version preference

Bug: 365886251
Flag: com.android.settings.flags.catalyst_firmware_version
Test: Unit
Change-Id: I1f3f69b08fe22c66d1b1c6786f9cd7e7ff413529

tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/MainlineModuleVersionPreferenceControllerTest.java

@@ -47,6 +47,7 @@ import org.mockito.MockitoAnnotations;
 import org.robolectric.RobolectricTestRunner;
 import org.robolectric.RuntimeEnvironment;
 
+// LINT.IfChange
 @RunWith(RobolectricTestRunner.class)
 public class MainlineModuleVersionPreferenceControllerTest {
 
@@ -222,3 +223,4 @@ public class MainlineModuleVersionPreferenceControllerTest {
         when(mPackageManager.getPackageInfo(eq(provider), anyInt())).thenReturn(info);
     }
 }
+// LINT.ThenChange(MainlineModuleVersionPreferenceTest.kt)

tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/MainlineModuleVersionPreferenceTest.kt

@@ -0,0 +1,185 @@
+/*
+ * Copyright (C) 2024 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.settings.deviceinfo.firmwareversion
+
+import android.content.ContextWrapper
+import android.content.Intent
+import android.content.pm.PackageInfo
+import android.content.pm.PackageManager
+import android.content.pm.ResolveInfo
+import android.content.res.Resources
+import android.platform.test.annotations.DisableFlags
+import android.platform.test.annotations.EnableFlags
+import android.platform.test.flag.junit.SetFlagsRule
+import androidx.preference.Preference
+import androidx.test.core.app.ApplicationProvider
+import com.android.settings.R
+import com.android.settings.deviceinfo.firmwareversion.MainlineModuleVersionPreference.Companion.MODULE_UPDATE_ACTION
+import com.android.settings.deviceinfo.firmwareversion.MainlineModuleVersionPreference.Companion.MODULE_UPDATE_ACTION_V2
+import com.android.settings.flags.Flags
+import com.google.common.truth.Truth.assertThat
+import org.junit.Rule
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.ArgumentMatchers.any
+import org.mockito.ArgumentMatchers.anyInt
+import org.mockito.ArgumentMatchers.eq
+import org.mockito.kotlin.KStubbing
+import org.mockito.kotlin.doAnswer
+import org.mockito.kotlin.doReturn
+import org.mockito.kotlin.doThrow
+import org.mockito.kotlin.mock
+import org.robolectric.RobolectricTestRunner
+
+// LINT.IfChange
+@RunWith(RobolectricTestRunner::class)
+class MainlineModuleVersionPreferenceTest {
+    @get:Rule val setFlagsRule = SetFlagsRule()
+
+    private lateinit var mockPackageManager: PackageManager
+    private lateinit var mockResources: Resources
+
+    private val context =
+        object : ContextWrapper(ApplicationProvider.getApplicationContext()) {
+            override fun getPackageManager(): PackageManager = mockPackageManager
+
+            override fun getResources(): Resources = mockResources
+        }
+
+    private val mainlineModuleVersionPreference = MainlineModuleVersionPreference()
+
+    @Test
+    fun isAvailable_noMainlineModuleProvider_unavailable() {
+        createMocks("", null)
+        assertThat(mainlineModuleVersionPreference.isAvailable(context)).isFalse()
+    }
+
+    @Test
+    fun isAvailable_noMainlineModulePackageInfo_unavailable() {
+        createMocks("test.provider", PackageManager.NameNotFoundException())
+        assertThat(mainlineModuleVersionPreference.isAvailable(context)).isFalse()
+    }
+
+    @Test
+    fun isAvailable_hasMainlineModulePackageInfo_available() {
+        createMocks("test.provider", "test version 123")
+        assertThat(mainlineModuleVersionPreference.isAvailable(context)).isTrue()
+    }
+
+    @Test
+    fun getSummary_versionIsNull() {
+        createMocks("test.provider", PackageInfo())
+        assertThat(mainlineModuleVersionPreference.getSummary(context)).isNull()
+    }
+
+    @Test
+    fun getSummary_versionIsEmpty() {
+        createMocks("test.provider", "")
+        assertThat(mainlineModuleVersionPreference.getSummary(context)).isNull()
+    }
+
+    @Test
+    fun getSummary_versionIsNotDate() {
+        createMocks("test.provider", "a")
+        assertThat(mainlineModuleVersionPreference.getSummary(context)).isEqualTo("a")
+    }
+
+    @Test
+    fun getSummary_versionIsMonth() {
+        createMocks("test.provider", "2019-05")
+        assertThat(mainlineModuleVersionPreference.getSummary(context)).isEqualTo("May 1, 2019")
+    }
+
+    @Test
+    fun getSummary_versionIsDate() {
+        createMocks("test.provider", "2019-05-13")
+        assertThat(mainlineModuleVersionPreference.getSummary(context)).isEqualTo("May 13, 2019")
+    }
+
+    @Test
+    @EnableFlags(Flags.FLAG_MAINLINE_MODULE_EXPLICIT_INTENT)
+    fun intentV2_preferenceShouldBeSelectable() {
+        intent_preferenceShouldBeSelectable(MODULE_UPDATE_ACTION_V2, MODULE_PACKAGE)
+    }
+
+    @Test
+    @DisableFlags(Flags.FLAG_MAINLINE_MODULE_EXPLICIT_INTENT)
+    fun intent_preferenceShouldBeSelectable() {
+        intent_preferenceShouldBeSelectable(MODULE_UPDATE_ACTION, null)
+    }
+
+    private fun intent_preferenceShouldBeSelectable(action: String, intentPackage: String?) {
+        createMocks("test.provider", "test version 123") {
+            on { resolveActivity(any(), anyInt()) } doAnswer
+                {
+                    when {
+                        (it.arguments[0] as Intent).action == action -> ResolveInfo()
+                        else -> null
+                    }
+                }
+        }
+
+        val preference = Preference(context)
+        mainlineModuleVersionPreference.bind(preference, mainlineModuleVersionPreference)
+
+        val intent = preference.intent!!
+        assertThat(intent.action).isEqualTo(action)
+        assertThat(preference.isSelectable).isTrue()
+        assertThat(intent.`package`).isEqualTo(intentPackage)
+    }
+
+    @Test
+    fun intent_null() {
+        createMocks("test.provider", "test version 123")
+
+        val preference = Preference(context)
+        mainlineModuleVersionPreference.bind(preference, mainlineModuleVersionPreference)
+
+        assertThat(preference.intent).isNull()
+        assertThat(preference.isSelectable).isFalse()
+    }
+
+    private fun createMocks(
+        pkg: String,
+        pkgInfo: Any?,
+        stubbing: KStubbing<PackageManager>.() -> Unit = {},
+    ) {
+        mockResources = mock {
+            on { getString(R.string.config_mainline_module_update_package) } doReturn MODULE_PACKAGE
+            on {
+                getString(com.android.internal.R.string.config_defaultModuleMetadataProvider)
+            } doReturn pkg
+        }
+
+        mockPackageManager = mock {
+            when (pkgInfo) {
+                is PackageInfo -> on { getPackageInfo(eq(pkg), anyInt()) } doReturn pkgInfo
+                is String ->
+                    on { getPackageInfo(eq(pkg), anyInt()) } doReturn
+                        PackageInfo().apply { versionName = pkgInfo }
+                is Exception -> on { getPackageInfo(eq(pkg), anyInt()) } doThrow pkgInfo
+                else -> {}
+            }
+            stubbing.invoke(this)
+        }
+    }
+
+    companion object {
+        const val MODULE_PACKAGE = "com.android.vending"
+    }
+}
+// LINT.ThenChange(MainlineModuleVersionPreferenceControllerTest.java)