Merge "Add HomeTransitionObserverTest to Shell." into main

/*

 * Copyright (C) 2023 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.wm.shell.transition;

import static android.app.WindowConfiguration.ACTIVITY_TYPE_HOME;

import static android.app.WindowConfiguration.ACTIVITY_TYPE_UNDEFINED;

import static android.view.WindowManager.TRANSIT_OPEN;

import static android.view.WindowManager.TRANSIT_TO_BACK;

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

import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.times;

import static org.mockito.Mockito.when;

import android.app.ActivityManager;

import android.app.WindowConfiguration.ActivityType;

import android.content.Context;

import android.os.Handler;

import android.os.IBinder;

import android.os.Looper;

import android.os.RemoteException;

import android.view.SurfaceControl;

import android.window.TransitionInfo;

import android.window.TransitionInfo.TransitionMode;

import android.window.WindowOrganizer;

import androidx.test.ext.junit.runners.AndroidJUnit4;

import androidx.test.filters.SmallTest;

import androidx.test.platform.app.InstrumentationRegistry;

import com.android.wm.shell.ShellTestCase;

import com.android.wm.shell.TestShellExecutor;

import com.android.wm.shell.common.DisplayController;

import com.android.wm.shell.common.ShellExecutor;

import com.android.wm.shell.common.TransactionPool;

import com.android.wm.shell.sysui.ShellController;

import com.android.wm.shell.sysui.ShellInit;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import java.util.ArrayList;

import java.util.List;

/**

 * Tests for the home transition observer.

 */

@SmallTest

@RunWith(AndroidJUnit4.class)

public class HomeTransitionObserverTest extends ShellTestCase {

    private final WindowOrganizer mOrganizer = mock(WindowOrganizer.class);

    private final TransactionPool mTransactionPool = mock(TransactionPool.class);

    private final Context mContext =

            InstrumentationRegistry.getInstrumentation().getTargetContext();

    private final ShellExecutor mAnimExecutor = new TestShellExecutor();

    private final TestShellExecutor mMainExecutor = new TestShellExecutor();

    private final Handler mMainHandler = new Handler(Looper.getMainLooper());

    private final DisplayController mDisplayController = mock(DisplayController.class);

    private Transitions mTransition;

    @Before

    public void setUp() {

        mTransition = new Transitions(mContext, mock(ShellInit.class), mock(ShellController.class),

                mOrganizer, mTransactionPool, mDisplayController, mMainExecutor,

                mMainHandler, mAnimExecutor);

    }

    @Test

    public void testHomeActivityWithOpenModeNotifiesHomeIsVisible() throws RemoteException {

        IHomeTransitionListener listener = mock(IHomeTransitionListener.class);

        when(listener.asBinder()).thenReturn(mock(IBinder.class));

        HomeTransitionObserver observer = new HomeTransitionObserver(mContext, mMainExecutor,

                mTransition);

        observer.setHomeTransitionListener(listener);

        TransitionInfo info = mock(TransitionInfo.class);

        TransitionInfo.Change change = mock(TransitionInfo.Change.class);

        ActivityManager.RunningTaskInfo taskInfo = mock(ActivityManager.RunningTaskInfo.class);

        when(change.getTaskInfo()).thenReturn(taskInfo);

        when(info.getChanges()).thenReturn(new ArrayList<>(List.of(change)));

        setupTransitionInfo(taskInfo, change, ACTIVITY_TYPE_HOME, TRANSIT_OPEN);

        observer.onTransitionReady(mock(IBinder.class),

                info,

                mock(SurfaceControl.Transaction.class),

                mock(SurfaceControl.Transaction.class));

        verify(listener, times(1)).onHomeVisibilityChanged(true);

    }

    @Test

    public void testHomeActivityWithCloseModeNotifiesHomeIsNotVisible() throws RemoteException {

        IHomeTransitionListener listener = mock(IHomeTransitionListener.class);

        when(listener.asBinder()).thenReturn(mock(IBinder.class));

        HomeTransitionObserver observer = new HomeTransitionObserver(mContext, mMainExecutor,

                mTransition);

        observer.setHomeTransitionListener(listener);

        TransitionInfo info = mock(TransitionInfo.class);

        TransitionInfo.Change change = mock(TransitionInfo.Change.class);

        ActivityManager.RunningTaskInfo taskInfo = mock(ActivityManager.RunningTaskInfo.class);

        when(change.getTaskInfo()).thenReturn(taskInfo);

        when(info.getChanges()).thenReturn(new ArrayList<>(List.of(change)));

        setupTransitionInfo(taskInfo, change, ACTIVITY_TYPE_HOME, TRANSIT_TO_BACK);

        observer.onTransitionReady(mock(IBinder.class),

                info,

                mock(SurfaceControl.Transaction.class),

                mock(SurfaceControl.Transaction.class));

        verify(listener, times(1)).onHomeVisibilityChanged(false);

    }

    @Test

    public void testNonHomeActivityDoesNotTriggerCallback() throws RemoteException {

        IHomeTransitionListener listener = mock(IHomeTransitionListener.class);

        when(listener.asBinder()).thenReturn(mock(IBinder.class));

        HomeTransitionObserver observer = new HomeTransitionObserver(mContext, mMainExecutor,

                mTransition);

        observer.setHomeTransitionListener(listener);

        TransitionInfo info = mock(TransitionInfo.class);

        TransitionInfo.Change change = mock(TransitionInfo.Change.class);

        ActivityManager.RunningTaskInfo taskInfo = mock(ActivityManager.RunningTaskInfo.class);

        when(change.getTaskInfo()).thenReturn(taskInfo);

        when(info.getChanges()).thenReturn(new ArrayList<>(List.of(change)));

        setupTransitionInfo(taskInfo, change, ACTIVITY_TYPE_UNDEFINED, TRANSIT_TO_BACK);

        observer.onTransitionReady(mock(IBinder.class),

                info,

                mock(SurfaceControl.Transaction.class),

                mock(SurfaceControl.Transaction.class));

        verify(listener, times(0)).onHomeVisibilityChanged(anyBoolean());

    }

    /**

     * Helper class to initialize variables for the rest.

     */

    private void setupTransitionInfo(ActivityManager.RunningTaskInfo taskInfo,

            TransitionInfo.Change change,

            @ActivityType int activityType,

            @TransitionMode int mode) {

        when(taskInfo.getActivityType()).thenReturn(activityType);

        when(change.getMode()).thenReturn(mode);

    }

}