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

Commit b8065d3c authored by Michael Mikhail's avatar Michael Mikhail
Browse files

Fix test to run on scene container flag. 

Flag: com.android.systemui.scene_container
Bug: 410078024
Test: atest SystemUiRoboTests:MediaOutputAdapterLegacyTest
Change-Id: I94715f8f799174a0af1e3c29c4c46742e28c06c7
parent 75ce70f8

packages/SystemUI/multivalentTests/src/com/android/systemui/media/dialog/MediaOutputAdapterLegacyTest.java

+5 −3
Original line number Diff line number Diff line
@@ -32,6 +32,7 @@ import static org.mockito.Mockito.never; 
import static org.mockito.Mockito.spy;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

import static org.mockito.kotlin.StubberKt.doNothing;



import android.graphics.drawable.Icon;

import android.platform.test.annotations.DisableFlags;
@@ -715,15 +716,16 @@ public class MediaOutputAdapterLegacyTest extends SysuiTestCase { 
        assertThat(mMediaDevice2.getState()).isEqualTo(

                LocalMediaManager.MediaDeviceState.STATE_DISCONNECTED);

        when(mMediaDevice2.getSelectionBehavior()).thenReturn(SELECTION_BEHAVIOR_TRANSFER);

        mMediaOutputAdapter = new MediaOutputAdapterLegacy(mMediaSwitchingController,

                mContext.getMainExecutor(), ThreadUtils.getBackgroundExecutor());

        mMediaOutputAdapter = new MediaOutputAdapterLegacy(mMediaSwitchingController, mMainExecutor,

                mBackgroundExecutor);

        mMediaOutputAdapter.updateItems();

        mViewHolder = (MediaOutputAdapterLegacy.MediaDeviceViewHolderLegacy) mMediaOutputAdapter

                .onCreateViewHolder(new LinearLayout(mContext), 0);

        MediaOutputAdapterLegacy.MediaDeviceViewHolderLegacy spyMediaDeviceViewHolder = spy(

                mViewHolder);

        mMediaOutputAdapter.getItemCount();

        doNothing().when(spyMediaDeviceViewHolder).showCustomEndSessionDialog(mMediaDevice2);



        mMediaOutputAdapter.getItemCount();

        mMediaOutputAdapter.onBindViewHolder(spyMediaDeviceViewHolder, 0);

        mMediaOutputAdapter.onBindViewHolder(spyMediaDeviceViewHolder, 1);

        spyMediaDeviceViewHolder.mContainerLayout.performClick();