Merge "Dismiss dialog after MediaOutputDialog tests"

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;

        when(mMediaDevice.getFeatures()).thenReturn(mFeatures);

    }

    @After

    public void tearDown() {

        mMediaOutputDialog.dismissDialog();

    }

    @Test

    public void getStopButtonVisibility_remoteDevice_returnVisible() {

        mFeatures.add(MediaRoute2Info.FEATURE_REMOTE_PLAYBACK);