Merge "Prevent unnecessary reconnection for wifi privacy" into main
This commit is contained in:
@@ -72,6 +72,10 @@ public class WifiPrivacyPreferenceController2 extends BasePreferenceController i
|
|||||||
@Override
|
@Override
|
||||||
public boolean onPreferenceChange(@NonNull Preference preference, Object newValue) {
|
public boolean onPreferenceChange(@NonNull Preference preference, Object newValue) {
|
||||||
final int privacy = Integer.parseInt((String) newValue);
|
final int privacy = Integer.parseInt((String) newValue);
|
||||||
|
if (mWifiEntry.getPrivacy() == privacy) {
|
||||||
|
// Prevent disconnection + reconnection if settings not changed.
|
||||||
|
return true;
|
||||||
|
}
|
||||||
mWifiEntry.setPrivacy(privacy);
|
mWifiEntry.setPrivacy(privacy);
|
||||||
|
|
||||||
// To activate changing, we need to reconnect network. WiFi will auto connect to
|
// To activate changing, we need to reconnect network. WiFi will auto connect to
|
||||||
|
@@ -1,124 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (C) 2019 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.wifi.details2;
|
|
||||||
|
|
||||||
import static com.google.common.truth.Truth.assertThat;
|
|
||||||
|
|
||||||
import static org.mockito.Mockito.doReturn;
|
|
||||||
import static org.mockito.Mockito.mock;
|
|
||||||
import static org.mockito.Mockito.spy;
|
|
||||||
import static org.mockito.Mockito.when;
|
|
||||||
|
|
||||||
import android.content.Context;
|
|
||||||
|
|
||||||
import androidx.preference.DropDownPreference;
|
|
||||||
|
|
||||||
import com.android.settings.R;
|
|
||||||
import com.android.wifitrackerlib.WifiEntry;
|
|
||||||
|
|
||||||
import org.junit.Before;
|
|
||||||
import org.junit.Test;
|
|
||||||
import org.junit.runner.RunWith;
|
|
||||||
import org.mockito.Mock;
|
|
||||||
import org.robolectric.RobolectricTestRunner;
|
|
||||||
import org.robolectric.RuntimeEnvironment;
|
|
||||||
|
|
||||||
@RunWith(RobolectricTestRunner.class)
|
|
||||||
public class WifiPrivacyPreferenceController2Test {
|
|
||||||
|
|
||||||
private static final int PRIVACY_RANDOMIZED = WifiEntry.PRIVACY_RANDOMIZED_MAC;
|
|
||||||
private static final int PRIVACY_TRUSTED = WifiEntry.PRIVACY_DEVICE_MAC;
|
|
||||||
|
|
||||||
@Mock private WifiEntry mMockWifiEntry;
|
|
||||||
|
|
||||||
private WifiPrivacyPreferenceController2 mPreferenceController;
|
|
||||||
private Context mContext;
|
|
||||||
private DropDownPreference mDropDownPreference;
|
|
||||||
private String[] mPerferenceStrings;
|
|
||||||
|
|
||||||
@Before
|
|
||||||
public void setUp() {
|
|
||||||
mContext = RuntimeEnvironment.application;
|
|
||||||
|
|
||||||
mMockWifiEntry = mock(WifiEntry.class);
|
|
||||||
WifiPrivacyPreferenceController2 preferenceController =
|
|
||||||
new WifiPrivacyPreferenceController2(mContext);
|
|
||||||
preferenceController.setWifiEntry(mMockWifiEntry);
|
|
||||||
mPreferenceController = spy(preferenceController);
|
|
||||||
mDropDownPreference = new DropDownPreference(mContext);
|
|
||||||
mDropDownPreference.setEntries(R.array.wifi_privacy_entries);
|
|
||||||
mDropDownPreference.setEntryValues(R.array.wifi_privacy_values);
|
|
||||||
|
|
||||||
mPerferenceStrings = mContext.getResources().getStringArray(R.array.wifi_privacy_entries);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testUpdateState_wifiPrivacy_setCorrectValue() {
|
|
||||||
doReturn(PRIVACY_TRUSTED).when(mPreferenceController).getRandomizationValue();
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
int prefValue = mPreferenceController.translateMacRandomizedValueToPrefValue(
|
|
||||||
PRIVACY_TRUSTED);
|
|
||||||
assertThat(mDropDownPreference.getEntry()).isEqualTo(mPerferenceStrings[prefValue]);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testUpdateState_wifiNotMetered_setCorrectValue() {
|
|
||||||
doReturn(PRIVACY_RANDOMIZED).when(mPreferenceController).getRandomizationValue();
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
int prefValue = mPreferenceController.translateMacRandomizedValueToPrefValue(
|
|
||||||
PRIVACY_RANDOMIZED);
|
|
||||||
assertThat(mDropDownPreference.getEntry()).isEqualTo(mPerferenceStrings[prefValue]);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testUpdateState_canSetPrivacyInNextUpdate_shouldBeSelectable() {
|
|
||||||
// Return false in WifiEntry#canSetPrivacy to make preference un-selectable first.
|
|
||||||
when(mMockWifiEntry.canSetPrivacy()).thenReturn(false);
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
assertThat(mDropDownPreference.isSelectable()).isFalse();
|
|
||||||
|
|
||||||
// Return true in WifiEntry#canSetPrivacy to verify preference back to selectable.
|
|
||||||
when(mMockWifiEntry.canSetPrivacy()).thenReturn(true);
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
assertThat(mDropDownPreference.isSelectable()).isTrue();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
public void testUpdateState_canNotSetPrivacyInNextUpdate_shouldNotBeSelectable() {
|
|
||||||
// Return true in WifiEntry#canSetPrivacy to make preference selectable first.
|
|
||||||
when(mMockWifiEntry.canSetPrivacy()).thenReturn(true);
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
assertThat(mDropDownPreference.isSelectable()).isTrue();
|
|
||||||
|
|
||||||
// Return false in WifiEntry#canSetPrivacy to verify preference back to un-selectable.
|
|
||||||
when(mMockWifiEntry.canSetPrivacy()).thenReturn(false);
|
|
||||||
|
|
||||||
mPreferenceController.updateState(mDropDownPreference);
|
|
||||||
|
|
||||||
assertThat(mDropDownPreference.isSelectable()).isFalse();
|
|
||||||
}
|
|
||||||
}
|
|
@@ -0,0 +1,143 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2023 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.wifi.details2
|
||||||
|
|
||||||
|
import android.content.Context
|
||||||
|
import androidx.preference.ListPreference
|
||||||
|
import androidx.test.core.app.ApplicationProvider
|
||||||
|
import androidx.test.ext.junit.runners.AndroidJUnit4
|
||||||
|
import com.android.settings.R
|
||||||
|
import com.android.wifitrackerlib.WifiEntry
|
||||||
|
import com.google.common.truth.Truth.assertThat
|
||||||
|
import org.junit.Test
|
||||||
|
import org.junit.runner.RunWith
|
||||||
|
import org.mockito.kotlin.any
|
||||||
|
import org.mockito.kotlin.doReturn
|
||||||
|
import org.mockito.kotlin.mock
|
||||||
|
import org.mockito.kotlin.never
|
||||||
|
import org.mockito.kotlin.spy
|
||||||
|
import org.mockito.kotlin.stub
|
||||||
|
import org.mockito.kotlin.verify
|
||||||
|
import org.mockito.kotlin.whenever
|
||||||
|
|
||||||
|
@RunWith(AndroidJUnit4::class)
|
||||||
|
class WifiPrivacyPreferenceController2Test {
|
||||||
|
private var mockWifiEntry = mock<WifiEntry>()
|
||||||
|
|
||||||
|
private var context: Context = ApplicationProvider.getApplicationContext()
|
||||||
|
|
||||||
|
private var controller = spy(WifiPrivacyPreferenceController2(context).apply {
|
||||||
|
setWifiEntry(mockWifiEntry)
|
||||||
|
})
|
||||||
|
|
||||||
|
private var preference = ListPreference(context).apply {
|
||||||
|
setEntries(R.array.wifi_privacy_entries)
|
||||||
|
setEntryValues(R.array.wifi_privacy_values)
|
||||||
|
}
|
||||||
|
|
||||||
|
private var preferenceStrings = context.resources.getStringArray(R.array.wifi_privacy_entries)
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun updateState_wifiPrivacy_setCorrectValue() {
|
||||||
|
controller.stub {
|
||||||
|
doReturn(WifiEntry.PRIVACY_DEVICE_MAC).whenever(mock).randomizationValue
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.updateState(preference)
|
||||||
|
|
||||||
|
val prefValue = WifiPrivacyPreferenceController2.translateMacRandomizedValueToPrefValue(
|
||||||
|
WifiEntry.PRIVACY_DEVICE_MAC
|
||||||
|
)
|
||||||
|
assertThat(preference.entry).isEqualTo(preferenceStrings[prefValue])
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun updateState_wifiNotMetered_setCorrectValue() {
|
||||||
|
controller.stub {
|
||||||
|
doReturn(WifiEntry.PRIVACY_RANDOMIZED_MAC).whenever(mock).randomizationValue
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.updateState(preference)
|
||||||
|
|
||||||
|
val prefValue = WifiPrivacyPreferenceController2.translateMacRandomizedValueToPrefValue(
|
||||||
|
WifiEntry.PRIVACY_RANDOMIZED_MAC
|
||||||
|
)
|
||||||
|
assertThat(preference.entry).isEqualTo(preferenceStrings[prefValue])
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun updateState_canSetPrivacyInNextUpdate_shouldBeSelectable() {
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
// Return false in WifiEntry#canSetPrivacy to make preference un-selectable first.
|
||||||
|
on { canSetPrivacy() } doReturn false
|
||||||
|
}
|
||||||
|
controller.updateState(preference)
|
||||||
|
assertThat(preference.isSelectable).isFalse()
|
||||||
|
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
// Return true in WifiEntry#canSetPrivacy to verify preference back to selectable.
|
||||||
|
on { canSetPrivacy() } doReturn true
|
||||||
|
}
|
||||||
|
controller.updateState(preference)
|
||||||
|
assertThat(preference.isSelectable).isTrue()
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun updateState_canNotSetPrivacyInNextUpdate_shouldNotBeSelectable() {
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
// Return true in WifiEntry#canSetPrivacy to make preference selectable first.
|
||||||
|
on { canSetPrivacy() } doReturn true
|
||||||
|
}
|
||||||
|
controller.updateState(preference)
|
||||||
|
assertThat(preference.isSelectable).isTrue()
|
||||||
|
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
// Return false in WifiEntry#canSetPrivacy to verify preference back to un-selectable.
|
||||||
|
on { canSetPrivacy() } doReturn false
|
||||||
|
}
|
||||||
|
controller.updateState(preference)
|
||||||
|
assertThat(preference.isSelectable).isFalse()
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun onPreferenceChange_sameNewValue_doNoting() {
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
on { privacy } doReturn 0
|
||||||
|
on { connectedState } doReturn WifiEntry.CONNECTED_STATE_CONNECTED
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.onPreferenceChange(preference, "0")
|
||||||
|
|
||||||
|
verify(mockWifiEntry, never()).privacy = any()
|
||||||
|
verify(mockWifiEntry, never()).disconnect(null)
|
||||||
|
verify(mockWifiEntry, never()).connect(null)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun onPreferenceChange_differentNewValue_setAndReconnect() {
|
||||||
|
mockWifiEntry.stub {
|
||||||
|
on { privacy } doReturn 0
|
||||||
|
on { connectedState } doReturn WifiEntry.CONNECTED_STATE_CONNECTED
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.onPreferenceChange(preference, "1")
|
||||||
|
|
||||||
|
verify(mockWifiEntry).privacy = 1
|
||||||
|
verify(mockWifiEntry).disconnect(null)
|
||||||
|
verify(mockWifiEntry).connect(null)
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user