From 2d044ee0a3e9e3c4a1fe09e55717bfbc861c1897 Mon Sep 17 00:00:00 2001
From: SongFerngWang <songferngwang@google.com>
Date: Wed, 7 Jun 2023 17:26:26 +0800
Subject: [PATCH] Fix the blocking test case issue

Bug: 284074742
Test: run ABTD for v2/android-virtual-infra/test_mapping/presubmit-large-avd
Change-Id: If7c89d3a8219f126d2a3f95e9447f27542680a31
---
 .../media/dialog/MediaOutputBroadcastDialogTest.java         | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialogTest.java b/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialogTest.java
index 9dba9b5b3c3e7..f8971fd7c99c7 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialogTest.java
+++ b/packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialogTest.java
@@ -136,7 +136,10 @@ public class MediaOutputBroadcastDialogTest extends SysuiTestCase {
 
     @After
     public void tearDown() {
-        mMediaOutputBroadcastDialog.dismissDialog();
+        if (mMediaOutputBroadcastDialog.mAlertDialog != null){
+            mMediaOutputBroadcastDialog.mAlertDialog.dismiss();
+        }
+        mMediaOutputBroadcastDialog.dismiss();
     }
 
     @Test