PawletOS/app_Settings
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update setDenylist() input structure

Browse Source 
- Change the input type from list of String to Integer
 - Update test case

Bug: 306329984
Test: make SettingsRoboTests
Change-Id: I018ef49523810577b17375628e45c24270a3bdf4
This commit is contained in:
Wesley Wang
2023-11-16 13:12:30 +08:00
committed by YK Hung
parent 3185a36b90
commit 24278e7a92
2 changed files with 35 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
tests/robotests/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManagerTest.java
View File

@@ -27,7 +27,6 @@ import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
@@ -37,8 +36,8 @@ import static org.mockito.Mockito.when;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.pm.PackageManager;
import android.net.NetworkPolicyManager;
import android.util.ArraySet;
import org.junit.After;
import org.junit.Before;
@@ -51,11 +50,12 @@ import org.robolectric.RuntimeEnvironment;
import java.util.Collections;
import java.util.List;
import java.util.Set;
@RunWith(RobolectricTestRunner.class)
public class DynamicDenylistManagerTest {
private static final int[] EMPTY_ARRAY = new int[]{};
private static final int[] EMPTY_ARRAY = new int[] {};
private static final String FAKE_UID_1 = "1001";
private static final String FAKE_UID_2 = "1002";
private static final int FAKE_UID_1_INT = Integer.parseInt(FAKE_UID_1);
@@ -67,8 +67,6 @@ public class DynamicDenylistManagerTest {
@Mock
private NetworkPolicyManager mNetworkPolicyManager;
@Mock
private PackageManager mPackageManager;
@Before
public void setUp() {
@@ -90,7 +88,7 @@ public class DynamicDenylistManagerTest {
@Test
public void init_withExistedRejectPolicy_createWithExpectedValue() {
initDynamicDenylistManager(new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
assertThat(mManualDenyListPref.getAll()).hasSize(3);
assertTrue(mManualDenyListPref.contains(PREF_KEY_MANUAL_DENYLIST_SYNCED));
@@ -166,43 +164,36 @@ public class DynamicDenylistManagerTest {
public void setDenylist_emptyListAndNoData_doNothing() {
initDynamicDenylistManager(EMPTY_ARRAY);
mDynamicDenylistManager.setDenylist(Collections.emptyList());
setDenylist(Collections.emptySet());
verify(mNetworkPolicyManager, never()).setUidPolicy(anyInt(), anyInt());
}
@Test
public void setDenylist_uidDeniedAlready_doNothing()
throws PackageManager.NameNotFoundException {
when(mPackageManager.getPackageUid(anyString(), eq(0))).thenReturn(FAKE_UID_1_INT);
initDynamicDenylistManager(new int[]{FAKE_UID_1_INT});
public void setDenylist_uidDeniedAlready_doNothing() {
initDynamicDenylistManager(new int[] {FAKE_UID_1_INT});
mDynamicDenylistManager.setDenylist(List.of(FAKE_UID_1));
setDenylist(new ArraySet<>(List.of(FAKE_UID_1_INT)));
verify(mNetworkPolicyManager, never()).setUidPolicy(anyInt(), anyInt());
}
@Test
public void setDenylist_sameList_doNothing() throws PackageManager.NameNotFoundException {
when(mPackageManager.getPackageUid(eq(FAKE_UID_1), eq(0))).thenReturn(FAKE_UID_1_INT);
when(mPackageManager.getPackageUid(eq(FAKE_UID_2), eq(0))).thenReturn(FAKE_UID_2_INT);
public void setDenylist_sameList_doNothing() {
initDynamicDenylistManager(EMPTY_ARRAY);
setupPreference(mDynamicDenyListPref, FAKE_UID_2, FAKE_UID_1);
mDynamicDenylistManager.setDenylist(List.of(FAKE_UID_1, FAKE_UID_2));
setDenylist(new ArraySet<>(List.of(FAKE_UID_1_INT, FAKE_UID_2_INT)));
verify(mNetworkPolicyManager, never()).setUidPolicy(anyInt(), anyInt());
}
@Test
public void setDenylist_newListWithOldData_modifyPolicyNoneAndReject()
throws PackageManager.NameNotFoundException {
when(mPackageManager.getPackageUid(anyString(), eq(0))).thenReturn(
Integer.parseInt(FAKE_UID_1));
public void setDenylist_newListWithOldData_modifyPolicyNoneAndReject() {
initDynamicDenylistManager(EMPTY_ARRAY);
setupPreference(mDynamicDenyListPref, FAKE_UID_2);
mDynamicDenylistManager.setDenylist(List.of(FAKE_UID_1));
setDenylist(new ArraySet<>(List.of(FAKE_UID_1_INT)));
verify(mNetworkPolicyManager).setUidPolicy(FAKE_UID_2_INT, POLICY_NONE);
verify(mNetworkPolicyManager).setUidPolicy(FAKE_UID_1_INT,
@@ -212,13 +203,10 @@ public class DynamicDenylistManagerTest {
}
@Test
public void setDenylist_newListWithoutOldData_modifyPolicyReject()
throws PackageManager.NameNotFoundException {
when(mPackageManager.getPackageUid(anyString(), eq(0))).thenReturn(
Integer.parseInt(FAKE_UID_1));
public void setDenylist_newListWithoutOldData_modifyPolicyReject() {
initDynamicDenylistManager(EMPTY_ARRAY);
mDynamicDenylistManager.setDenylist(List.of(FAKE_UID_1));
setDenylist(new ArraySet<>(List.of(FAKE_UID_1_INT)));
verify(mNetworkPolicyManager, never()).setUidPolicy(anyInt(), eq(POLICY_NONE));
verify(mNetworkPolicyManager).setUidPolicy(FAKE_UID_1_INT,
@@ -232,7 +220,7 @@ public class DynamicDenylistManagerTest {
initDynamicDenylistManager(EMPTY_ARRAY);
setupPreference(mDynamicDenyListPref, FAKE_UID_2);
mDynamicDenylistManager.setDenylist(Collections.emptyList());
setDenylist(Collections.emptySet());
verify(mNetworkPolicyManager).setUidPolicy(FAKE_UID_2_INT, POLICY_NONE);
verify(mNetworkPolicyManager, never()).setUidPolicy(anyInt(),
@@ -267,7 +255,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_nullPackageName_doNothing() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(null, false);
@@ -277,7 +265,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_invalidPackageName_doNothing() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded("invalid_package_name", false);
@@ -287,7 +275,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_denylistUnchanged_doNothingWithPolicy() {
initDynamicDenylistManager(new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(SETTINGS_PACKAGE_NAME, false);
@@ -296,7 +284,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_denylistChanged_resetAndClear() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(SETTINGS_PACKAGE_NAME, false);
@@ -306,7 +294,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_forceResetWithNullPackageName_resetAndClear() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(null, true);
@@ -314,9 +302,9 @@ public class DynamicDenylistManagerTest {
verify(mNetworkPolicyManager).setUidPolicy(eq(FAKE_UID_2_INT), eq(POLICY_NONE));
}
@Test// 4
@Test
public void resetDenylistIfNeeded_forceResetWithInvalidPackageName_resetAndClear() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded("invalid_package_name", true);
@@ -326,7 +314,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_forceResetButDenylistUnchanged_doNothingWithPolicy() {
initDynamicDenylistManager(new int[]{FAKE_UID_1_INT});
initDynamicDenylistManager(new int[] {FAKE_UID_1_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(SETTINGS_PACKAGE_NAME, true);
@@ -336,7 +324,7 @@ public class DynamicDenylistManagerTest {
@Test
public void resetDenylistIfNeeded_forceResetWithDenylistChanged_resetAndClear() {
initDynamicDenylistManager(new int[0], new int[]{FAKE_UID_1_INT, FAKE_UID_2_INT});
initDynamicDenylistManager(new int[0], new int[] {FAKE_UID_1_INT, FAKE_UID_2_INT});
mDynamicDenylistManager.resetDenylistIfNeeded(SETTINGS_PACKAGE_NAME, true);
@@ -372,10 +360,9 @@ public class DynamicDenylistManagerTest {
private void initDynamicDenylistManager(int[] preload) {
initDynamicDenylistManager(preload, preload);
}
private void initDynamicDenylistManager(int[] preload1, int[] preload2) {
final Context context = spy(RuntimeEnvironment.application.getApplicationContext());
when(context.getApplicationContext()).thenReturn(context);
when(context.getPackageManager()).thenReturn(mPackageManager);
when(mNetworkPolicyManager.getUidsWithPolicy(anyInt()))
.thenReturn(preload1).thenReturn(preload2);
mDynamicDenylistManager = new DynamicDenylistManager(context, mNetworkPolicyManager);
@@ -383,9 +370,15 @@ public class DynamicDenylistManagerTest {
mDynamicDenyListPref = mDynamicDenylistManager.getDynamicDenylistPref();
}
private void setDenylist(Set<Integer> packageNameList) {
mDynamicDenylistManager.setDenylist(packageNameList);
}
private void setupPreference(SharedPreferences sharedPreferences, String... uids) {
final SharedPreferences.Editor editor = sharedPreferences.edit();
for (String uid : uids) {
sharedPreferences.edit().putInt(uid, POLICY_REJECT_METERED_BACKGROUND).apply();
editor.putInt(uid, POLICY_REJECT_METERED_BACKGROUND);
}
editor.apply();
}
}