Update setDenylist() input structure
- Change the input type from list of String to Integer
- Update test case
Bug: 306329984
Test: make SettingsRoboTests
Change-Id: I018ef49523810577b17375628e45c24270a3bdf4
diff --git a/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManager.java b/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManager.java
index 7eae7eb..e9e1218 100644
--- a/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManager.java
+++ b/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManager.java
@@ -20,18 +20,15 @@
import static android.net.NetworkPolicyManager.POLICY_REJECT_METERED_BACKGROUND;
import static com.android.settings.Utils.SETTINGS_PACKAGE_NAME;
-import static com.android.settings.fuelgauge.BatteryUtils.UID_ZERO;
import android.content.Context;
import android.content.SharedPreferences;
-import android.content.pm.PackageManager;
import android.net.NetworkPolicyManager;
import android.util.ArraySet;
import android.util.Log;
import androidx.annotation.VisibleForTesting;
-import java.util.List;
import java.util.Set;
/** A class to dynamically manage per apps {@link NetworkPolicyManager} POLICY_ flags. */
@@ -95,20 +92,7 @@
}
/** Suggest a list of package to set as POLICY_REJECT. */
- public void setDenylist(List<String> packageNameList) {
- final Set<Integer> denylistTargetUids = new ArraySet<>(packageNameList.size());
- for (String packageName : packageNameList) {
- try {
- final int uid = mContext.getPackageManager().getPackageUid(packageName, 0);
- if (uid == UID_ZERO) {
- continue;
- }
- denylistTargetUids.add(uid);
- } catch (PackageManager.NameNotFoundException e) {
- Log.e(TAG, "Unknown package name: " + packageName, e);
- }
- }
-
+ public void setDenylist(Set<Integer> denylistTargetUids) {
final Set<Integer> manualDenylistUids = getDenylistAllUids(getManualDenylistPref());
denylistTargetUids.removeAll(manualDenylistUids);
@@ -126,16 +110,14 @@
uid -> editor.putInt(String.valueOf(uid), POLICY_REJECT_METERED_BACKGROUND));
editor.apply();
- // Set new added UIDs into REJECT policy.
synchronized (mLock) {
+ // Set new added UIDs into REJECT policy.
for (int uid : denylistTargetUids) {
if (!lastDynamicDenylistUids.contains(uid)) {
mNetworkPolicyManager.setUidPolicy(uid, POLICY_REJECT_METERED_BACKGROUND);
}
}
- }
- // Unset removed UIDs back to NONE policy.
- synchronized (mLock) {
+ // Unset removed UIDs back to NONE policy.
for (int uid : lastDynamicDenylistUids) {
if (!denylistTargetUids.contains(uid)) {
mNetworkPolicyManager.setUidPolicy(uid, POLICY_NONE);
diff --git a/tests/robotests/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManagerTest.java b/tests/robotests/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManagerTest.java
index bfa7cfa..3202610 100644
--- a/tests/robotests/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManagerTest.java
+++ b/tests/robotests/src/com/android/settings/fuelgauge/datasaver/DynamicDenylistManagerTest.java
@@ -27,7 +27,6 @@
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 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 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 @@
@Mock
private NetworkPolicyManager mNetworkPolicyManager;
- @Mock
- private PackageManager mPackageManager;
@Before
public void setUp() {
@@ -90,7 +88,7 @@
@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 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 @@
}
@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 @@
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 @@
@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 @@
@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 @@
@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 @@
@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 @@
@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 @@
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 @@
@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 @@
@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 @@
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 @@
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();
}
}