Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 0cfb8be8 authored by Jing Ji's avatar Jing Ji Committed by Automerger Merge Worker
Browse files

Merge "Add support to persist app background restriction settings." into tm-dev am: 4935cbc6 

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/17354264



Change-Id: Id3c5917b61a4db5cd2453cb835f7642a98dc217d
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents e61b8a34 4935cbc6

services/core/java/com/android/server/am/AppRestrictionController.java

+451 −28

File changed.

Preview size limit exceeded, changes collapsed.

services/tests/mockingservicestests/src/com/android/server/am/BackgroundRestrictionTest.java

+108 −0
Original line number Diff line number Diff line
@@ -36,11 +36,15 @@ import static android.app.AppOpsManager.OP_CAMERA; 
import static android.app.AppOpsManager.OP_FINE_LOCATION;

import static android.app.AppOpsManager.OP_NONE;

import static android.app.AppOpsManager.OP_RECORD_AUDIO;

import static android.app.usage.UsageStatsManager.REASON_MAIN_DEFAULT;

import static android.app.usage.UsageStatsManager.REASON_MAIN_FORCED_BY_SYSTEM;

import static android.app.usage.UsageStatsManager.REASON_MAIN_FORCED_BY_USER;

import static android.app.usage.UsageStatsManager.REASON_MAIN_MASK;

import static android.app.usage.UsageStatsManager.REASON_MAIN_USAGE;

import static android.app.usage.UsageStatsManager.REASON_SUB_DEFAULT_UNDEFINED;

import static android.app.usage.UsageStatsManager.REASON_SUB_FORCED_SYSTEM_FLAG_ABUSE;

import static android.app.usage.UsageStatsManager.REASON_SUB_FORCED_USER_FLAG_INTERACTION;

import static android.app.usage.UsageStatsManager.REASON_SUB_MASK;

import static android.app.usage.UsageStatsManager.REASON_SUB_USAGE_USER_INTERACTION;

import static android.app.usage.UsageStatsManager.STANDBY_BUCKET_ACTIVE;

import static android.app.usage.UsageStatsManager.STANDBY_BUCKET_EXEMPTED;
@@ -93,6 +97,7 @@ import static org.mockito.Mockito.timeout; 
import static org.mockito.Mockito.times;

import static org.mockito.Mockito.verify;



import android.annotation.UserIdInt;

import android.app.ActivityManagerInternal;

import android.app.ActivityManagerInternal.AppBackgroundRestrictionListener;

import android.app.ActivityManagerInternal.BindServiceEventListener;
@@ -146,6 +151,7 @@ import com.android.server.am.AppFGSTracker.AppFGSPolicy; 
import com.android.server.am.AppMediaSessionTracker.AppMediaSessionPolicy;

import com.android.server.am.AppRestrictionController.ConstantsObserver;

import com.android.server.am.AppRestrictionController.NotificationHelper;

import com.android.server.am.AppRestrictionController.RestrictionSettings;

import com.android.server.am.AppRestrictionController.UidBatteryUsageProvider;

import com.android.server.am.BaseAppStateTimeEvents.BaseTimeEvent;

import com.android.server.apphibernation.AppHibernationManagerInternal;
@@ -165,6 +171,7 @@ import org.mockito.Mock; 
import org.mockito.MockitoAnnotations;

import org.mockito.verification.VerificationMode;



import java.io.File;

import java.time.Duration;

import java.util.ArrayList;

import java.util.Arrays;
@@ -356,6 +363,7 @@ public final class BackgroundRestrictionTest { 
        verify(mAppBindServiceEventsTracker.mInjector.getActivityManagerInternal())

                .addBindServiceEventListener(mBindServiceEventListenerCap.capture());

        mBindServiceEventListener = mBindServiceEventListenerCap.getValue();

        waitForIdleHandler(mBgRestrictionController.getBackgroundHandler());

    }



    @After
@@ -2788,6 +2796,96 @@ public final class BackgroundRestrictionTest { 
                new double[] {10.0d, 1.0d, 12.0d, 1.0d, 18.0d, 1.0d})));

    }



    @SuppressWarnings("GuardedBy")

    @Test

    public void testPersistRestrictionSettings() throws Exception {

        final RestrictionSettings settings = mBgRestrictionController.mRestrictionSettings;

        final String testPkg0 = TEST_PACKAGE_BASE + 0;

        final String testPkg1 = TEST_PACKAGE_BASE + 1;

        final String testPkg2 = TEST_PACKAGE_BASE + 2;

        final String testPkg3 = TEST_PACKAGE_BASE + 3;

        final int testUid0 = UserHandle.getUid(TEST_USER0, TEST_PACKAGE_APPID_BASE + 0);

        final int testUid1 = UserHandle.getUid(TEST_USER0, TEST_PACKAGE_APPID_BASE + 1);

        final int testUid2 = UserHandle.getUid(TEST_USER1, TEST_PACKAGE_APPID_BASE + 2);

        final int testUid3 = UserHandle.getUid(TEST_USER1, TEST_PACKAGE_APPID_BASE + 3);

        settings.reset();

        setRestrictionSettings(settings, testPkg0, testUid0,

                RESTRICTION_LEVEL_ADAPTIVE_BUCKET,

                10_000L, REASON_MAIN_DEFAULT | REASON_SUB_DEFAULT_UNDEFINED,

                new long[] {0L, 0L}, false);

        setRestrictionSettings(settings, testPkg1, testUid1,

                RESTRICTION_LEVEL_RESTRICTED_BUCKET,

                20_000L, REASON_MAIN_FORCED_BY_SYSTEM | REASON_SUB_FORCED_SYSTEM_FLAG_ABUSE,

                new long[] {10_000L, 0L}, false);

        setRestrictionSettings(settings, testPkg2, testUid2,

                RESTRICTION_LEVEL_BACKGROUND_RESTRICTED,

                25_000L, REASON_MAIN_FORCED_BY_USER | REASON_SUB_FORCED_USER_FLAG_INTERACTION,

                new long[] {0L, 15_000L}, false);

        setRestrictionSettings(settings, testPkg3, testUid3,

                RESTRICTION_LEVEL_RESTRICTED_BUCKET,

                30_000L, REASON_MAIN_DEFAULT | REASON_SUB_DEFAULT_UNDEFINED,

                new long[] {0L, 0L}, false);

        RestrictionSettings test = (RestrictionSettings) settings.clone();



        // Verify our clone works correctly.

        assertTrue(settings.equals(test));



        // Reset the test object.

        test.resetToDefault();



        // Save the original data into xml.

        settings.persistToXml(TEST_USER0);

        settings.persistToXml(TEST_USER1);



        // Load it to our test object.

        test.loadFromXml(false);

        // Verify we restored it correctly.

        assertTrue(settings.equals(test));



        // Remove one package.

        settings.removePackage(testPkg3, testUid3);

        // Verify it.

        verifyLoadedSettings(settings);



        // Add it back.

        setRestrictionSettings(settings, testPkg3, testUid3,

                RESTRICTION_LEVEL_RESTRICTED_BUCKET,

                30_000L, REASON_MAIN_DEFAULT | REASON_SUB_DEFAULT_UNDEFINED,

                new long[] {0L, 1_000L}, true);

        // Verify it.

        verifyLoadedSettings(settings);



        // Remove one user.

        settings.removeUser(TEST_USER1);

        // Verify it.

        verifyLoadedSettings(settings);

    }



    private void verifyLoadedSettings(RestrictionSettings settings) throws Exception {

        // Make a new copy and reset it.

        RestrictionSettings test = (RestrictionSettings) settings.clone();

        test.resetToDefault();



        // Wait for the idleness so the data is persisted.

        waitForIdleHandler(mBgRestrictionController.getBackgroundHandler());

        // Load it to our test object.

        test.loadFromXml(false);

        // Verify we restored it correctly.

        assertTrue(settings.equals(test));

    }



    @SuppressWarnings("GuardedBy")

    private void setRestrictionSettings(RestrictionSettings settings, String pkgName, int uid,

            int level, long levelTs, int reason, long[] notificationTime, boolean persist) {

        mCurrentTimeMillis = levelTs;

        settings.update(pkgName, uid, level, reason & REASON_MAIN_MASK, reason & REASON_SUB_MASK);

        final RestrictionSettings.PkgSettings pkgSettings = settings.getRestrictionSettingsLocked(

                uid, pkgName);

        for (int i = 0; i < notificationTime.length; i++) {

            pkgSettings.setLastNotificationTime(i, notificationTime[i], persist);

        }

    }



    private LinkedList<UidStateEventWithBattery> createUidStateEventWithBatteryList(

            boolean[] isStart, long[] timestamps, double[] batteryUsage) {

        final LinkedList<UidStateEventWithBattery> result = new LinkedList<>();
@@ -2948,6 +3046,16 @@ public final class BackgroundRestrictionTest { 
        void scheduleInitTrackers(Handler handler, Runnable initializers) {

            initializers.run();

        }



        @Override

        File getDataSystemDeDirectory(@UserIdInt int userId) {

            return new File(mContext.getFilesDir(), Integer.toString(userId));

        }



        @Override

        long currentTimeMillis() {

            return mCurrentTimeMillis;

        }

    }



    private class TestBaseTrackerInjector<T extends BaseAppStatePolicy>