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

Commit 2d957bb0 authored by Joanne Chung's avatar Joanne Chung Committed by Android (Google) Code Review
Browse files

Merge "Add test for package visibility rule in PackageMonitorCallbackHelper callback" into main

parents efc969f9 e31c06b3

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