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

Commit 40da2b12 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

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



Change-Id: Ic8f00e654240e87633b8d5963b178a6d13ba7f34
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 9788232a ebe89c5c

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,