diff --git a/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputDialogTest.java b/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputDialogTest.java index ca328fbe44fb5..9ebb587d3d858 100644 --- a/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputDialogTest.java +++ b/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputDialogTest.java @@ -36,6 +36,7 @@ import com.android.systemui.SysuiTestCase; import com.android.systemui.plugins.ActivityStarter; import com.android.systemui.statusbar.phone.ShadeController; +import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; @@ -73,6 +74,11 @@ public class MediaOutputDialogTest extends SysuiTestCase { when(mMediaDevice.getFeatures()).thenReturn(mFeatures); } + @After + public void tearDown() { + mMediaOutputDialog.dismissDialog(); + } + @Test public void getStopButtonVisibility_remoteDevice_returnVisible() { mFeatures.add(MediaRoute2Info.FEATURE_REMOTE_PLAYBACK);