Merge "Fix IndexOutOfBoundsException on VirtualDisplay removal"

import java.io.FileDescriptor;

import java.io.PrintWriter;

import java.util.ArrayList;

import java.util.List;

import java.util.Set;

final class VirtualDeviceImpl extends IVirtualDevice.Stub

    private final int mOwnerUid;

    private final InputController mInputController;

    @VisibleForTesting

    final List<Integer> mVirtualDisplayIds = new ArrayList<>();

    final Set<Integer> mVirtualDisplayIds = new ArraySet<>();

    private final OnDeviceCloseListener mListener;

    private final IBinder mAppToken;

    private final VirtualDeviceParams mParams;

import static org.mockito.Mockito.doCallRealMethod;

import static org.mockito.Mockito.doNothing;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

import static org.testng.Assert.assertThrows;

import android.Manifest;

import android.app.admin.DevicePolicyManager;

import android.companion.virtual.VirtualDeviceParams;

import android.content.Context;

import android.content.ContextWrapper;

    private DisplayManagerInternal mDisplayManagerInternalMock;

    @Mock

    private VirtualDeviceImpl.PendingTrampolineCallback mPendingTrampolineCallback;

    @Mock

    private DevicePolicyManager mDevicePolicyManagerMock;

    @Before

    public void setUp() {

        mContext = Mockito.spy(new ContextWrapper(InstrumentationRegistry.getTargetContext()));

        doNothing().when(mContext).enforceCallingOrSelfPermission(

                eq(Manifest.permission.CREATE_VIRTUAL_DEVICE), anyString());

        when(mContext.getSystemService(Context.DEVICE_POLICY_SERVICE)).thenReturn(

                mDevicePolicyManagerMock);

        mInputController = new InputController(new Object(), mNativeWrapperMock);

        mDeviceImpl = new VirtualDeviceImpl(mContext,

                /* association info */ null, new Binder(), /* uid */ 0, mInputController,

                new VirtualDeviceParams.Builder().build());

    }

    @Test

    public void onVirtualDisplayRemovedLocked_doesNotThrowException() {

        final int displayId = 2;

        mDeviceImpl.onVirtualDisplayCreatedLocked(displayId);

        // This call should not throw any exceptions.

        mDeviceImpl.onVirtualDisplayRemovedLocked(displayId);

    }

    @Test

    public void createVirtualKeyboard_noDisplay_failsSecurityException() {

        assertThrows(