Merge "Add test for setting windowingMode in startTask()" into main

        for (int taskId : taskIds) {

            ActivityManager.RunningTaskInfo task = mTaskOrganizer.getRunningTaskInfo(taskId);

            if (task != null) {

                wct.setWindowingMode(task.token, WINDOWING_MODE_UNDEFINED)

                        .setBounds(task.token, null);

                wct.setWindowingMode(task.getToken(), WINDOWING_MODE_UNDEFINED)

                        .setBounds(task.getToken(), null);

            }

        }

    }

package com.android.wm.shell.splitscreen;

import static android.app.ActivityTaskManager.INVALID_TASK_ID;

import static android.app.WindowConfiguration.WINDOWING_MODE_UNDEFINED;

import static android.view.Display.DEFAULT_DISPLAY;

import static com.android.wm.shell.shared.split.SplitScreenConstants.SPLIT_INDEX_UNDEFINED;

import static org.junit.Assert.assertEquals;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.ArgumentMatchers.eq;

import static org.mockito.ArgumentMatchers.notNull;

import static org.mockito.Mockito.atLeastOnce;

import android.graphics.Rect;

import android.os.Bundle;

import android.os.Handler;

import android.os.IBinder;

import android.os.Looper;

import android.view.SurfaceControl;

import android.window.RemoteTransition;

import android.window.WindowContainerToken;

import android.window.WindowContainerTransaction;

import androidx.test.annotation.UiThreadTest;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.ArgumentCaptor;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

                .startFullscreenTransition(any(), any());

    }

    @Test

    public void startTask_ensureWindowingModeCleared() {

        SplitScreenTransitions splitScreenTransitions =

                spy(mStageCoordinator.getSplitTransitions());

        mStageCoordinator.setSplitTransitions(splitScreenTransitions);

        ArgumentCaptor<WindowContainerTransaction> wctCaptor =

                ArgumentCaptor.forClass(WindowContainerTransaction.class);

        int taskId = 18;

        IBinder binder = mock(IBinder.class);

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

        WindowContainerToken mockToken = mock(WindowContainerToken.class);

        when(mockToken.asBinder()).thenReturn(binder);

        when(rti.getToken()).thenReturn(mockToken);

        when(mTaskOrganizer.getRunningTaskInfo(taskId)).thenReturn(rti);

        mStageCoordinator.startTask(taskId, SPLIT_POSITION_TOP_OR_LEFT, null /*options*/,

                null, SPLIT_INDEX_UNDEFINED);

        verify(splitScreenTransitions).startEnterTransition(anyInt(),

                wctCaptor.capture(), any(), any(), anyInt(), anyBoolean());

        int windowingMode = wctCaptor.getValue().getChanges().get(binder).getWindowingMode();

        assertEquals(windowingMode, WINDOWING_MODE_UNDEFINED);

    }

    private Transitions createTestTransitions() {

        ShellInit shellInit = new ShellInit(mMainExecutor);

        final Transitions t = new Transitions(mContext, shellInit, mock(ShellController.class),