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

Commit f90c69f4 authored by shafik's avatar shafik
Browse files

Test Rollback Manager can handle time change 

Test flow:
	* Install app A with rollback enabled
	* Wait half of the expiration time
	* Install app B with rollback enabled
	* Move time forward by expiration time
	* Wait half of the expiration time
	- At this point, an expiration check should run and expire app A
	but not app B
	* Assert app A rollback is expired
	* Assert app B rollback is not expired
	* Wait half of the expiration time
	* Assert app B rollback is expired

Note: installing app A before app B is done solely to schedule an
expiration check before app B rollback has expired

Test: atest RollbackTest#testTimeChangeDoesNotAffectLifetime
Test: run the same test without the registering the time change handler
in RollbackManagerServiceImpl - test fails
Bug: 124095332

Change-Id: I4bc2d449e4195e059fa43e7568783683a60e2197
parent 55805af7

tests/RollbackTest/RollbackTest/src/com/android/tests/rollback/RollbackTest.java

+77 −0
Original line number Diff line number Diff line
@@ -379,6 +379,83 @@ public class RollbackTest { 
            // Check that the data has expired after the expiration time (with a buffer of 1 second)

            Thread.sleep(expirationTime / 2);

            assertNull(getUniqueRollbackInfoForPackage(rm.getAvailableRollbacks(), TEST_APP_A));



        } finally {

            DeviceConfig.setProperty(DeviceConfig.Rollback.BOOT_NAMESPACE,

                    DeviceConfig.Rollback.ROLLBACK_LIFETIME_IN_MILLIS,

                    Long.toString(defaultExpirationTime), false /* makeDefault*/);

            RollbackTestUtils.dropShellPermissionIdentity();

        }

    }



    /**

     * Test that changing time on device does not affect the duration of time that we keep

     * rollback available

     */

    @Test

    public void testTimeChangeDoesNotAffectLifetime() throws Exception {

        long expirationTime = TimeUnit.SECONDS.toMillis(30);

        long defaultExpirationTime = TimeUnit.HOURS.toMillis(48);

        RollbackManager rm = RollbackTestUtils.getRollbackManager();



        try {

            RollbackTestUtils.adoptShellPermissionIdentity(

                    Manifest.permission.INSTALL_PACKAGES,

                    Manifest.permission.DELETE_PACKAGES,

                    Manifest.permission.MANAGE_ROLLBACKS,

                    Manifest.permission.WRITE_DEVICE_CONFIG,

                    Manifest.permission.SET_TIME);



            DeviceConfig.setProperty(DeviceConfig.Rollback.BOOT_NAMESPACE,

                    DeviceConfig.Rollback.ROLLBACK_LIFETIME_IN_MILLIS,

                    Long.toString(expirationTime), false /* makeDefault*/);



            // Pull the new expiration time from DeviceConfig

            rm.reloadPersistedData();



            // Install app A with rollback enabled

            RollbackTestUtils.uninstall(TEST_APP_A);

            RollbackTestUtils.install("RollbackTestAppAv1.apk", false);

            RollbackTestUtils.install("RollbackTestAppAv2.apk", true);

            assertEquals(2, RollbackTestUtils.getInstalledVersion(TEST_APP_A));



            Thread.sleep(expirationTime / 2);



            // Install app B with rollback enabled

            RollbackTestUtils.uninstall(TEST_APP_B);

            RollbackTestUtils.install("RollbackTestAppBv1.apk", false);

            RollbackTestUtils.install("RollbackTestAppBv2.apk", true);

            assertEquals(2, RollbackTestUtils.getInstalledVersion(TEST_APP_B));

            // 1 second buffer

            Thread.sleep(1000);



            try {

                // Change the time

                RollbackTestUtils.forwardTimeBy(expirationTime);



                // 1 second buffer to allow Rollback Manager to handle time change before loading

                // persisted data

                Thread.sleep(1000);



                // Load timestamps from storage

                rm.reloadPersistedData();



                // Wait until rollback for app A has expired

                // This will trigger an expiration run that should expire app A but not B

                Thread.sleep(expirationTime / 2);

                assertNull(getUniqueRollbackInfoForPackage(rm.getAvailableRollbacks(), TEST_APP_A));



                // Rollback for app B should not be expired

                RollbackInfo rollback = getUniqueRollbackInfoForPackage(

                        rm.getAvailableRollbacks(), TEST_APP_B);

                assertRollbackInfoEquals(TEST_APP_B, 2, 1, rollback);



                // Wait until rollback for app B has expired

                Thread.sleep(expirationTime / 2);

                assertNull(getUniqueRollbackInfoForPackage(rm.getAvailableRollbacks(), TEST_APP_B));

            } finally {

                RollbackTestUtils.forwardTimeBy(-expirationTime);

            }

        } finally {

            DeviceConfig.setProperty(DeviceConfig.Rollback.BOOT_NAMESPACE,

                    DeviceConfig.Rollback.ROLLBACK_LIFETIME_IN_MILLIS,

tests/RollbackTest/RollbackTest/src/com/android/tests/rollback/RollbackTestUtils.java

+12 −0
Original line number Diff line number Diff line
@@ -21,6 +21,7 @@ import static org.junit.Assert.assertNotNull; 
import static org.junit.Assert.assertNull;

import static org.junit.Assert.fail;



import android.app.AlarmManager;

import android.content.BroadcastReceiver;

import android.content.ComponentName;

import android.content.Context;
@@ -63,6 +64,17 @@ class RollbackTestUtils { 
        return rm;

    }



    private static void setTime(long millis) {

        Context context = InstrumentationRegistry.getContext();

        AlarmManager am = (AlarmManager) context.getSystemService(Context.ALARM_SERVICE);

        am.setTime(millis);

    }



    static void forwardTimeBy(long offsetMillis) {

        setTime(System.currentTimeMillis() + offsetMillis);

        Log.i(TAG, "Forwarded time on device by " + offsetMillis + " millis");

    }



    /**

     * Returns the version of the given package installed on device.

     * Returns -1 if the package is not currently installed.