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

Commit 33a7977b authored by Sharon Su's avatar Sharon Su Committed by Automerger Merge Worker
Browse files

Merge "Ensure service info is updated so that checking for remote service uid... 

Merge "Ensure service info is updated so that checking for remote service uid returns the correct value." into tm-dev am: 31b9a4cb am: ebe89c5c am: 40da2b12

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



Change-Id: Ie0d1b8afe9f2afc2bd97077b8b3759a385408438
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents cc52a265 40da2b12

services/tests/mockingservicestests/Android.bp

+1 −0
Original line number Diff line number Diff line
@@ -61,6 +61,7 @@ android_test { 
        "services.devicepolicy",

        "services.net",

        "services.usage",

        "services.wallpapereffectsgeneration",

        "servicestests-core-utils",

        "servicestests-utils-mockito-extended",

        "testables",

services/tests/mockingservicestests/src/com/android/server/wallpapereffectsgeneration/WallpaperEffectsGenerationPerUserServiceTest.java

0 → 100644
+102 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2022 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.server.wallpapereffectsgeneration;



import static com.android.dx.mockito.inline.extended.ExtendedMockito.spyOn;



import static org.junit.Assert.assertFalse;

import static org.junit.Assert.assertNotNull;

import static org.junit.Assert.assertNull;

import static org.junit.Assert.assertTrue;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.ArgumentMatchers.anyLong;

import static org.mockito.Mockito.spy;

import static org.mockito.Mockito.when;



import android.Manifest;

import android.annotation.SuppressLint;

import android.app.AppGlobals;

import android.app.wallpapereffectsgeneration.CinematicEffectRequest;

import android.app.wallpapereffectsgeneration.ICinematicEffectListener;

import android.content.Context;

import android.content.pm.ApplicationInfo;

import android.content.pm.IPackageManager;

import android.content.pm.PackageManager;

import android.content.pm.ServiceInfo;

import android.graphics.Bitmap;



import androidx.test.core.app.ApplicationProvider;

import androidx.test.filters.SmallTest;



import org.junit.Before;

import org.junit.Test;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;



@SmallTest

@SuppressLint("GuardedBy")

public class WallpaperEffectsGenerationPerUserServiceTest {

    private static final int BIND_SERVICE_UID = 123;

    @Mock

    PackageManager mMockPackageManager;

    @Mock

    ICinematicEffectListener mListener;



    private Context mContext;

    private IPackageManager mIPackageManager;

    private WallpaperEffectsGenerationPerUserService mService;

    private ServiceInfo mServiceInfo;



    @Before

    public void setUp() throws Exception {

        MockitoAnnotations.initMocks(this);

        mContext = spy(ApplicationProvider.getApplicationContext());

        mIPackageManager = AppGlobals.getPackageManager();

        spyOn(mIPackageManager);

        when(mContext.getPackageManager()).thenReturn(mMockPackageManager);

        when(mMockPackageManager.checkPermission(any(), any()))

                .thenReturn(PackageManager.PERMISSION_GRANTED);

        ApplicationInfo mockApplicationInfo = new ApplicationInfo();

        mockApplicationInfo.uid = BIND_SERVICE_UID;

        mServiceInfo = new ServiceInfo();

        mServiceInfo.permission = Manifest.permission.BIND_WALLPAPER_EFFECTS_GENERATION_SERVICE;

        mServiceInfo.applicationInfo = mockApplicationInfo;

        when(mIPackageManager.getServiceInfo(any(), anyLong(), anyInt()))

                .thenReturn(mServiceInfo);



        WallpaperEffectsGenerationManagerService managerService =

                new WallpaperEffectsGenerationManagerService(mContext);

        mService = new WallpaperEffectsGenerationPerUserService(

                managerService, new Object(), mContext.getUserId());

    }



    @Test

    public void testIsCallingUidAllowed_returnFalseWhenServiceNotBound() {

        assertNull(mService.getServiceInfo());

        assertFalse(mService.isCallingUidAllowed(BIND_SERVICE_UID));

    }



    @Test

    public void testIsCallingUidAllowed_uidSetCorrectlyAfterFirstCall() {

        CinematicEffectRequest request = new CinematicEffectRequest(

                "test-id", Bitmap.createBitmap(32, 48, Bitmap.Config.ARGB_8888));

        mService.onGenerateCinematicEffectLocked(request, mListener);

        assertNotNull(mService.getServiceInfo());

        assertTrue(mService.isCallingUidAllowed(BIND_SERVICE_UID));

    }

}

services/wallpapereffectsgeneration/java/com/android/server/wallpapereffectsgeneration/WallpaperEffectsGenerationPerUserService.java

+2 −3
Original line number Diff line number Diff line
@@ -233,14 +233,13 @@ public class WallpaperEffectsGenerationPerUserService extends 
    @Nullable

    private RemoteWallpaperEffectsGenerationService ensureRemoteServiceLocked() {

        if (mRemoteService == null) {

            final String serviceName = getComponentNameLocked();

            if (serviceName == null) {

            final ComponentName serviceComponent = updateServiceInfoLocked();

            if (serviceComponent == null) {

                if (mMaster.verbose) {

                    Slog.v(TAG, "ensureRemoteServiceLocked(): not set");

                }

                return null;

            }

            ComponentName serviceComponent = ComponentName.unflattenFromString(serviceName);



            mRemoteService = new RemoteWallpaperEffectsGenerationService(getContext(),

                    serviceComponent, mUserId, this,