Ensure Bubbles get removed at the end of the test

                    .withEndActions(() -> {

                        View child = mManageMenu.getChildAt(0);

                        child.requestAccessibilityFocus();

                        // Update the AV's obscured touchable region for the new visibility state.

                        if (mExpandedBubble != null && mExpandedBubble.getExpandedView() != null) {

                            // Update the AV's obscured touchable region for the new state.

                            mExpandedBubble.getExpandedView().updateObscuredTouchableRegion();

                        }

                    })

                    .start();

import android.view.ViewTreeObserver;

import android.view.WindowManager;

import androidx.test.filters.FlakyTest;

import androidx.test.filters.SmallTest;

import com.android.internal.colorextraction.ColorExtractor;

import com.android.wm.shell.sysui.ShellController;

import com.android.wm.shell.sysui.ShellInit;

import org.junit.After;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.mockito.MockitoAnnotations;

import org.mockito.stubbing.Answer;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.List;

import java.util.Optional;

@FlakyTest

@SmallTest

@RunWith(AndroidTestingRunner.class)

@TestableLooper.RunWithLooper(setAsMainLooper = true)

                .addCallback(mKeyguardStateControllerCallbackCaptor.capture());

    }

    @After

    public void tearDown() {

        ArrayList<Bubble> bubbles = new ArrayList<>(mBubbleData.getBubbles());

        for (int i = 0; i < bubbles.size(); i++) {

            mBubbleController.removeBubble(bubbles.get(i).getKey(),

                    Bubbles.DISMISS_NO_LONGER_BUBBLE);

        }

    }

    @Test

    public void dreamingHidesBubbles() throws RemoteException {

        mBubbleController.updateBubble(mBubbleEntry);