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

Commit 499a586b authored by Vinit Nayak's avatar Vinit Nayak
Browse files

Add mock leashes for dimLayer and stage roots for 2 app flex split 

Bug: 401375944
Flag: com.android.wm.shell.enable_flexible_two_app_split
Test: atest WMShellUnitTests:com.android.wm.shell.splitscreen.StageCoordinatorTests
Change-Id: I3536f5d141b18787a981ab044026903127db8074
parent 83bdb759

libs/WindowManager/Shell/tests/unittest/src/com/android/wm/shell/splitscreen/StageCoordinatorTests.java

+9 −1
Original line number Diff line number Diff line
@@ -211,11 +211,19 @@ public class StageCoordinatorTests extends ShellTestCase { 
        when(mSplitLayout.getDividerLeash()).thenReturn(dividerLeash);



        mRootTask = new TestRunningTaskInfoBuilder().build();

        SurfaceControl rootLeash = new SurfaceControl.Builder().setName("test").build();

        SurfaceControl rootLeash = new SurfaceControl.Builder().setName("splitRoot").build();

        mStageCoordinator.onTaskAppeared(mRootTask, rootLeash);



        mSideStage.mRootTaskInfo = new TestRunningTaskInfoBuilder().build();

        mMainStage.mRootTaskInfo = new TestRunningTaskInfoBuilder().build();

        SurfaceControl mainRootLeash = new SurfaceControl.Builder().setName("mainRoot").build();

        SurfaceControl sideRootLeash = new SurfaceControl.Builder().setName("sideRoot").build();

        mMainStage.mRootLeash = mainRootLeash;

        mSideStage.mRootLeash = sideRootLeash;

        SurfaceControl mainDimLayer = new SurfaceControl.Builder().setName("mainDim").build();

        SurfaceControl sideDimLayer = new SurfaceControl.Builder().setName("sideDim").build();

        mMainStage.mDimLayer = mainDimLayer;

        mSideStage.mDimLayer = sideDimLayer;

        doReturn(mock(SplitDecorManager.class)).when(mMainStage).getSplitDecorManager();

        doReturn(mock(SplitDecorManager.class)).when(mSideStage).getSplitDecorManager();