[Catalyst] Simplify KeyValueStore API calls

NO_IFTTT=Catalyst only

Bug: 325144964
Flag: com.android.settings.flags.catalyst
Test: atest
Change-Id: I518b9c861eac6d515dfc7f0ac2d028f9395e06a1
This commit is contained in:
Jacky Wang
2024-12-08 19:20:16 +08:00
parent 6c0260c0ec
commit dc1a52a2ee
8 changed files with 21 additions and 40 deletions

View File

@@ -152,7 +152,7 @@ class AirplaneModePreference :
data: Intent?,
): Boolean {
if (requestCode == REQUEST_CODE_EXIT_ECM && resultCode == Activity.RESULT_OK) {
storage(context).setValue(KEY, Boolean::class.javaObjectType, true)
storage(context).setBoolean(KEY, true)
}
return true
}

View File

@@ -72,6 +72,6 @@ class VibrationMainSwitchPreferenceTest {
preference.createAndBindWidget(context)
private fun setVibrateOn(enabled: Boolean?) =
preference.storage(context).setValue(VIBRATE_ON, Boolean::class.javaObjectType, enabled)
preference.storage(context).setBoolean(VIBRATE_ON, enabled)
}
// LINT.ThenChange(VibrationMainSwitchPreferenceControllerTest.java)

View File

@@ -29,21 +29,17 @@ import org.junit.runner.RunWith
import org.mockito.kotlin.spy
import org.mockito.kotlin.verify
import org.mockito.kotlin.whenever
import org.robolectric.RuntimeEnvironment
@RunWith(AndroidJUnit4::class)
class BluetoothMainSwitchPreferenceTest {
@get:Rule
val setFlagsRule = SetFlagsRule()
@get:Rule val setFlagsRule = SetFlagsRule()
private val context: Context = ApplicationProvider.getApplicationContext()
private lateinit var bluetoothAdapter: BluetoothAdapter
private lateinit var bluetoothMainSwitchPreference: BluetoothMainSwitchPreference
@Before
fun setUp() {
bluetoothAdapter = spy(
BluetoothAdapter.getDefaultAdapter()
)
bluetoothAdapter = spy(BluetoothAdapter.getDefaultAdapter())
whenever(bluetoothAdapter.state).thenReturn(BluetoothAdapter.STATE_ON)
bluetoothMainSwitchPreference = BluetoothMainSwitchPreference(bluetoothAdapter)
}
@@ -62,19 +58,19 @@ class BluetoothMainSwitchPreferenceTest {
@Test
fun storageSetOff_turnOff() {
bluetoothMainSwitchPreference.storage(context).setValue(
bluetoothMainSwitchPreference.key, Boolean::class.javaObjectType, false
)
bluetoothMainSwitchPreference
.storage(context)
.setBoolean(bluetoothMainSwitchPreference.key, false)
verify(bluetoothAdapter).disable()
}
@Test
fun storageSetOn_turnOn() {
bluetoothMainSwitchPreference.storage(context).setValue(
bluetoothMainSwitchPreference.key, Boolean::class.javaObjectType, true
)
bluetoothMainSwitchPreference
.storage(context)
.setBoolean(bluetoothMainSwitchPreference.key, true)
verify(bluetoothAdapter).enable()
}
}
}

View File

@@ -117,11 +117,8 @@ class BatteryPercentageSwitchPreferenceTest {
batteryPercentageSwitchPreference.createAndBindWidget(context)
private fun showBatteryPercentage(on: Boolean) =
batteryPercentageSwitchPreference
.storage(context)
.setValue(
batteryPercentageSwitchPreference.storage(context).setBoolean(
Settings.System.SHOW_BATTERY_PERCENT,
Boolean::class.javaObjectType,
on,
)
}

View File

@@ -75,18 +75,14 @@ class BatterySaverPreferenceTest {
@Test
fun storeSetOn_setPowerSaveMode() {
batterySaverPreference
.storage(context)
.setValue(batterySaverPreference.key, Boolean::class.javaObjectType, true)
batterySaverPreference.storage(context).setBoolean(batterySaverPreference.key, true)
verify(powerManager).setPowerSaveModeEnabled(true)
}
@Test
fun storeSetOff_unsetPowerSaveMode() {
batterySaverPreference
.storage(context)
.setValue(batterySaverPreference.key, Boolean::class.javaObjectType, false)
batterySaverPreference.storage(context).setBoolean(batterySaverPreference.key, false)
verify(powerManager).setPowerSaveModeEnabled(false)
}

View File

@@ -96,11 +96,11 @@ class AdaptiveConnectivityTogglePreferenceTest {
private fun setAdaptiveConnectivityEnabled(enabled: Boolean) =
adaptiveConnectivityTogglePreference
.storage(context)
.setValue(ADAPTIVE_CONNECTIVITY_ENABLED, Boolean::class.javaObjectType, enabled)
.setBoolean(ADAPTIVE_CONNECTIVITY_ENABLED, enabled)
private fun getAdaptiveConnectivityEnabled() =
adaptiveConnectivityTogglePreference
.storage(context)
.getValue(ADAPTIVE_CONNECTIVITY_ENABLED, Boolean::class.javaObjectType)
.getBoolean(ADAPTIVE_CONNECTIVITY_ENABLED)
}
// LINT.ThenChange(AdaptiveConnectivityTogglePreferenceControllerTest.java)

View File

@@ -88,9 +88,7 @@ class AirplaneModePreferenceTest {
SettingsGlobalStore.get(context).setInt(Settings.Global.AIRPLANE_MODE_ON, 1)
val getValue =
airplaneModePreference
.storage(context)
.getValue(AirplaneModePreference.KEY, Boolean::class.javaObjectType)
airplaneModePreference.storage(context).getBoolean(AirplaneModePreference.KEY)
assertThat(getValue).isTrue()
}
@@ -100,9 +98,7 @@ class AirplaneModePreferenceTest {
SettingsGlobalStore.get(context).setInt(Settings.Global.AIRPLANE_MODE_ON, 0)
val getValue =
airplaneModePreference
.storage(context)
.getValue(AirplaneModePreference.KEY, Boolean::class.javaObjectType)
airplaneModePreference.storage(context).getBoolean(AirplaneModePreference.KEY)
assertThat(getValue).isFalse()
}

View File

@@ -49,9 +49,7 @@ class WifiSwitchPreferenceTest {
fun getValue_defaultOn_returnOn() {
mockWifiManager.stub { on { isWifiEnabled } doReturn true }
val getValue = wifiSwitchPreference
.storage(context)
.getValue(WifiSwitchPreference.KEY, Boolean::class.javaObjectType)
val getValue = wifiSwitchPreference.storage(context).getBoolean(WifiSwitchPreference.KEY)
assertThat(getValue).isTrue()
}
@@ -60,9 +58,7 @@ class WifiSwitchPreferenceTest {
fun getValue_defaultOff_returnOff() {
mockWifiManager.stub { on { isWifiEnabled } doReturn false }
val getValue = wifiSwitchPreference
.storage(context)
.getValue(WifiSwitchPreference.KEY, Boolean::class.javaObjectType)
val getValue = wifiSwitchPreference.storage(context).getBoolean(WifiSwitchPreference.KEY)
assertThat(getValue).isFalse()
}
@@ -87,4 +83,4 @@ class WifiSwitchPreferenceTest {
private fun getSwitchPreference(): SwitchPreferenceCompat =
wifiSwitchPreference.createAndBindWidget(context)
}
}