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

Commit e31c06b3 authored by Joanne Chung's avatar Joanne Chung
Browse files

Add test for package visibility rule in PackageMonitorCallbackHelper callback 

Bug: 289567218
Test: atest PackageMonitorCallbackHelperTest --iterations 100
Change-Id: I78d88412d4fb4344d87946f5c2c0f3f6fdb4691d
parent 26178c8e

services/tests/mockingservicestests/src/com/android/server/pm/PackageMonitorCallbackHelperTest.java

+50 −0
Original line number Diff line number Diff line
@@ -33,6 +33,8 @@ import android.os.Bundle; 
import android.os.Handler;

import android.os.IRemoteCallback;

import android.os.Looper;

import android.os.Process;

import android.util.SparseArray;



import org.junit.After;

import org.junit.Before;
@@ -244,6 +246,54 @@ public class PackageMonitorCallbackHelperTest { 
        verify(callback, after(WAIT_CALLBACK_CALLED_IN_MS).never()).sendResult(any());

    }



    @Test

    public void testRegisterPackageMonitorCallbackInAllowList_callbackShouldCalled()

            throws Exception {

        IRemoteCallback callback = createMockPackageMonitorCallback();

        SparseArray<int[]> broadcastAllowList = new SparseArray<>();

        broadcastAllowList.put(0, new int[] {Binder.getCallingUid()});



        mPackageMonitorCallbackHelper.registerPackageMonitorCallback(callback, 0 /* userId */,

                Binder.getCallingUid());

        mPackageMonitorCallbackHelper.notifyPackageMonitor(Intent.ACTION_PACKAGE_ADDED,

                FAKE_PACKAGE_NAME, createFakeBundle(), new int[]{0} /* userIds */,

                null /* instantUserIds */, broadcastAllowList);



        verify(callback, after(WAIT_CALLBACK_CALLED_IN_MS).times(1)).sendResult(any());

    }



    @Test

    public void testRegisterPackageMonitorCallbackNotInAllowList_callbackShouldNotCalled()

            throws Exception {

        IRemoteCallback callback = createMockPackageMonitorCallback();

        SparseArray<int[]> broadcastAllowList = new SparseArray<>();

        broadcastAllowList.put(0, new int[] {12345});



        mPackageMonitorCallbackHelper.registerPackageMonitorCallback(callback, 0 /* userId */,

                Binder.getCallingUid());

        mPackageMonitorCallbackHelper.notifyPackageMonitor(Intent.ACTION_PACKAGE_ADDED,

                FAKE_PACKAGE_NAME, createFakeBundle(), new int[]{0} /* userIds */,

                null /* instantUserIds */, broadcastAllowList);



        verify(callback, after(WAIT_CALLBACK_CALLED_IN_MS).never()).sendResult(any());

    }



    @Test

    public void testRegisterPackageMonitorCallbackNotInAllowListSystemUid_callbackShouldCalled()

            throws Exception {

        IRemoteCallback callback = createMockPackageMonitorCallback();

        SparseArray<int[]> broadcastAllowList = new SparseArray<>();

        broadcastAllowList.put(0, new int[] {12345});



        mPackageMonitorCallbackHelper.registerPackageMonitorCallback(callback, 0 /* userId */,

                Process.SYSTEM_UID);

        mPackageMonitorCallbackHelper.notifyPackageMonitor(Intent.ACTION_PACKAGE_ADDED,

                FAKE_PACKAGE_NAME, createFakeBundle(), new int[]{0} /* userIds */,

                null /* instantUserIds */, broadcastAllowList);



        verify(callback, after(WAIT_CALLBACK_CALLED_IN_MS).times(1)).sendResult(any());

    }



    private IRemoteCallback createMockPackageMonitorCallback() {

        return spy(new IRemoteCallback.Stub() {

            @Override