diff --git a/tests/robotests/Android.bp b/tests/robotests/Android.bp
index 3fd4306..60a6803 100644
--- a/tests/robotests/Android.bp
+++ b/tests/robotests/Android.bp
@@ -69,6 +69,9 @@
         "com_android_server_accessibility_flags_lib",
         "flag-junit",
         "flag-junit-base",
+        "kotlin-test",
+        "mockito-robolectric-prebuilt", // mockito deps order matters!
+        "mockito-kotlin2",
         "notification_flags_lib",
         "platform-test-annotations",
         "testables",
diff --git a/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceControllerTest.java
index 87fdb22..5d7e82e 100644
--- a/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceControllerTest.java
+++ b/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceControllerTest.java
@@ -38,6 +38,7 @@
 
 import java.util.Arrays;
 
+// LINT.IfChange
 @RunWith(RobolectricTestRunner.class)
 public class BasebandVersionPreferenceControllerTest {
     @Mock
@@ -68,3 +69,4 @@
         assertThat(mController.getAvailabilityStatus()).isEqualTo(AVAILABLE);
     }
 }
+// LINT.ThenChange(BasebandVersionPreferenceTest.kt)
diff --git a/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceTest.kt b/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceTest.kt
new file mode 100644
index 0000000..2722038
--- /dev/null
+++ b/tests/robotests/src/com/android/settings/deviceinfo/firmwareversion/BasebandVersionPreferenceTest.kt
@@ -0,0 +1,60 @@
+/*
+ * 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.Context
+import android.content.ContextWrapper
+import android.sysprop.TelephonyProperties
+import android.telephony.TelephonyManager
+import androidx.test.core.app.ApplicationProvider
+import com.google.common.truth.Truth.assertThat
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.kotlin.doReturn
+import org.mockito.kotlin.mock
+import org.robolectric.RobolectricTestRunner
+
+// LINT.IfChange
+@RunWith(RobolectricTestRunner::class)
+class BasebandVersionPreferenceTest {
+    private lateinit var telephonyManager: TelephonyManager
+
+    private val context: Context =
+        object : ContextWrapper(ApplicationProvider.getApplicationContext()) {
+            override fun getSystemService(name: String): Any? =
+                when {
+                    name == getSystemServiceName(TelephonyManager::class.java) -> telephonyManager
+                    else -> super.getSystemService(name)
+                }
+        }
+
+    private val basebandVersionPreference = BasebandVersionPreference()
+
+    @Test
+    fun isAvailable_wifiOnly_unavailable() {
+        telephonyManager = mock { on { isDataCapable } doReturn false }
+        assertThat(basebandVersionPreference.isAvailable(context)).isFalse()
+    }
+
+    @Test
+    fun isAvailable_hasMobile_available() {
+        TelephonyProperties.baseband_version(listOf("test"))
+        telephonyManager = mock { on { isDataCapable } doReturn true }
+        assertThat(basebandVersionPreference.isAvailable(context)).isTrue()
+    }
+}
+// LINT.ThenChange(BasebandVersionPreferenceControllerTest.java)