| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.settings.media; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import static org.mockito.Mockito.doReturn; |
| import static org.mockito.Mockito.mock; |
| 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 android.content.BroadcastReceiver; |
| import android.content.ContentResolver; |
| import android.content.Context; |
| import android.content.Intent; |
| import android.media.AudioManager; |
| import android.media.MediaRoute2ProviderService; |
| import android.media.RoutingSessionInfo; |
| import android.net.Uri; |
| |
| import com.android.settings.slices.ShadowSliceBackgroundWorker; |
| import com.android.settings.testutils.shadow.ShadowAudioManager; |
| import com.android.settings.testutils.shadow.ShadowBluetoothAdapter; |
| import com.android.settings.testutils.shadow.ShadowBluetoothUtils; |
| import com.android.settingslib.bluetooth.BluetoothEventManager; |
| import com.android.settingslib.bluetooth.LocalBluetoothManager; |
| import com.android.settingslib.media.LocalMediaManager; |
| import com.android.settingslib.media.MediaDevice; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.Mock; |
| import org.mockito.MockitoAnnotations; |
| import org.robolectric.RobolectricTestRunner; |
| import org.robolectric.RuntimeEnvironment; |
| import org.robolectric.annotation.Config; |
| import org.robolectric.shadows.ShadowApplication; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| @RunWith(RobolectricTestRunner.class) |
| @Config(shadows = {ShadowAudioManager.class, ShadowBluetoothAdapter.class, |
| ShadowBluetoothUtils.class, ShadowSliceBackgroundWorker.class}) |
| public class MediaDeviceUpdateWorkerTest { |
| |
| private static final Uri URI = Uri.parse("content://com.android.settings.slices/test"); |
| private static final Uri URI1 = Uri.parse("content://com.android.settings.slices/action/" |
| + "media_output?media_package_name=com.music1"); |
| private static final Uri URI2 = Uri.parse("content://com.android.settings.slices/action/" |
| + "media_output?media_package_name=com.music2"); |
| private static final String TEST_DEVICE_PACKAGE_NAME1 = "com.music1"; |
| private static final String TEST_DEVICE_PACKAGE_NAME2 = "com.music2"; |
| private static final String TEST_DEVICE_1_ID = "test_device_1_id"; |
| private static final String TEST_DEVICE_2_ID = "test_device_2_id"; |
| private static final String TEST_DEVICE_3_ID = "test_device_3_id"; |
| |
| private final List<MediaDevice> mMediaDevices = new ArrayList<>(); |
| |
| @Mock |
| private LocalBluetoothManager mLocalBluetoothManager; |
| @Mock |
| private BluetoothEventManager mBluetoothEventManager; |
| |
| private MediaDeviceUpdateWorker mMediaDeviceUpdateWorker; |
| private ContentResolver mResolver; |
| private Context mContext; |
| private MediaDevice mMediaDevice1; |
| private MediaDevice mMediaDevice2; |
| private ShadowApplication mShadowApplication; |
| private AudioManager mAudioManager; |
| |
| @Before |
| public void setUp() { |
| MockitoAnnotations.initMocks(this); |
| mContext = spy(RuntimeEnvironment.application); |
| mMediaDeviceUpdateWorker = new MediaDeviceUpdateWorker(mContext, URI); |
| mResolver = mock(ContentResolver.class); |
| mShadowApplication = ShadowApplication.getInstance(); |
| mAudioManager = mContext.getSystemService(AudioManager.class); |
| |
| mMediaDevice1 = mock(MediaDevice.class); |
| when(mMediaDevice1.getId()).thenReturn(TEST_DEVICE_1_ID); |
| mMediaDevice2 = mock(MediaDevice.class); |
| when(mMediaDevice2.getId()).thenReturn(TEST_DEVICE_2_ID); |
| mMediaDevices.add(mMediaDevice1); |
| mMediaDevices.add(mMediaDevice2); |
| |
| doReturn(mResolver).when(mContext).getContentResolver(); |
| } |
| |
| @Test |
| public void onDeviceListUpdate_shouldNotifyChange() { |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(mMediaDevices); |
| |
| verify(mResolver).notifyChange(URI, null); |
| } |
| |
| @Test |
| public void onSelectedDeviceStateChanged_shouldNotifyChange() { |
| mMediaDeviceUpdateWorker.onSelectedDeviceStateChanged(mMediaDevice1, |
| LocalMediaManager.MediaDeviceState.STATE_CONNECTED); |
| |
| verify(mResolver).notifyChange(URI, null); |
| } |
| |
| @Test |
| public void onDeviceAttributesChanged_shouldNotifyChange() { |
| mMediaDeviceUpdateWorker.onDeviceAttributesChanged(); |
| |
| verify(mResolver).notifyChange(URI, null); |
| } |
| |
| @Test |
| public void onDeviceListUpdate_sameDeviceList_shouldBeEqual() { |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(mMediaDevices); |
| |
| final List<MediaDevice> newDevices = new ArrayList<>(); |
| newDevices.add(mMediaDevice1); |
| newDevices.add(mMediaDevice2); |
| |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(newDevices); |
| final List<MediaDevice> devices = |
| new ArrayList<>(mMediaDeviceUpdateWorker.getMediaDevices()); |
| |
| assertThat(devices.get(0).getId()).isEqualTo(newDevices.get(0).getId()); |
| assertThat(devices.get(1).getId()).isEqualTo(newDevices.get(1).getId()); |
| } |
| |
| @Test |
| public void onDeviceListUpdate_add1DeviceToDeviceList_shouldBeEqual() { |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(mMediaDevices); |
| |
| final List<MediaDevice> newDevices = new ArrayList<>(); |
| final MediaDevice device3 = mock(MediaDevice.class); |
| when(mMediaDevice2.getId()).thenReturn(TEST_DEVICE_3_ID); |
| newDevices.add(mMediaDevice1); |
| newDevices.add(mMediaDevice2); |
| newDevices.add(device3); |
| |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(newDevices); |
| final List<MediaDevice> devices = |
| new ArrayList<>(mMediaDeviceUpdateWorker.getMediaDevices()); |
| |
| assertThat(devices.size()).isEqualTo(newDevices.size()); |
| } |
| |
| @Test |
| public void onDeviceListUpdate_less1DeviceToDeviceList_shouldBeEqual() { |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(mMediaDevices); |
| |
| final List<MediaDevice> newDevices = new ArrayList<>(); |
| newDevices.add(mMediaDevice1); |
| |
| mMediaDeviceUpdateWorker.onDeviceListUpdate(newDevices); |
| final List<MediaDevice> devices = |
| new ArrayList<>(mMediaDeviceUpdateWorker.getMediaDevices()); |
| |
| assertThat(devices.size()).isEqualTo(newDevices.size()); |
| } |
| |
| @Test |
| public void onRequestFailed_shouldNotifyChange() { |
| mMediaDeviceUpdateWorker.onRequestFailed(MediaRoute2ProviderService.REASON_UNKNOWN_ERROR); |
| |
| verify(mResolver).notifyChange(URI, null /* observer */); |
| } |
| |
| @Test |
| public void onReceive_inCallState_shouldNotifyChange() { |
| mMediaDeviceUpdateWorker.mLocalMediaManager = mock(LocalMediaManager.class); |
| mAudioManager.setMode(AudioManager.MODE_IN_CALL); |
| |
| mMediaDeviceUpdateWorker.onSlicePinned(); |
| final Intent intent = new Intent(AudioManager.STREAM_DEVICES_CHANGED_ACTION); |
| for (BroadcastReceiver receiver : mShadowApplication.getReceiversForIntent(intent)) { |
| receiver.onReceive(mContext, intent); |
| } |
| |
| verify(mResolver).notifyChange(URI, null); |
| } |
| |
| @Test |
| public void onReceive_notInCallState_doNothing() { |
| mMediaDeviceUpdateWorker.mLocalMediaManager = mock(LocalMediaManager.class); |
| mAudioManager.setMode(AudioManager.MODE_NORMAL); |
| |
| mMediaDeviceUpdateWorker.onSlicePinned(); |
| final Intent intent = new Intent(AudioManager.STREAM_DEVICES_CHANGED_ACTION); |
| for (BroadcastReceiver receiver : mShadowApplication.getReceiversForIntent(intent)) { |
| receiver.onReceive(mContext, intent); |
| } |
| |
| verify(mResolver, never()).notifyChange(URI, null); |
| } |
| |
| @Test |
| public void getActiveRemoteMediaSession_verifyList() { |
| mMediaDeviceUpdateWorker.mLocalMediaManager = mock(LocalMediaManager.class); |
| final List<RoutingSessionInfo> routingSessionInfos = new ArrayList<>(); |
| final RoutingSessionInfo remoteSessionInfo = mock(RoutingSessionInfo.class); |
| final RoutingSessionInfo localSessionInfo = mock(RoutingSessionInfo.class); |
| when(remoteSessionInfo.isSystemSession()).thenReturn(false); |
| when(localSessionInfo.isSystemSession()).thenReturn(true); |
| routingSessionInfos.add(remoteSessionInfo); |
| routingSessionInfos.add(localSessionInfo); |
| when(mMediaDeviceUpdateWorker.mLocalMediaManager.getActiveMediaSession()).thenReturn( |
| routingSessionInfos); |
| |
| assertThat(mMediaDeviceUpdateWorker.getActiveRemoteMediaDevice()).containsExactly( |
| remoteSessionInfo); |
| } |
| |
| @Test |
| public void onSlicePinned_packageUpdated_checkPackageName() { |
| ShadowBluetoothUtils.sLocalBluetoothManager = mLocalBluetoothManager; |
| when(mLocalBluetoothManager.getEventManager()).thenReturn(mBluetoothEventManager); |
| |
| mMediaDeviceUpdateWorker = new MediaDeviceUpdateWorker(mContext, URI1); |
| mMediaDeviceUpdateWorker.mLocalMediaManager = mock(LocalMediaManager.class); |
| when(mMediaDeviceUpdateWorker.mLocalMediaManager.getPackageName()) |
| .thenReturn(TEST_DEVICE_PACKAGE_NAME1); |
| mMediaDeviceUpdateWorker.onSlicePinned(); |
| |
| assertThat(mMediaDeviceUpdateWorker.mLocalMediaManager.getPackageName()).matches( |
| TEST_DEVICE_PACKAGE_NAME1); |
| |
| mMediaDeviceUpdateWorker = new MediaDeviceUpdateWorker(mContext, URI2); |
| mMediaDeviceUpdateWorker.mLocalMediaManager = mock(LocalMediaManager.class); |
| when(mMediaDeviceUpdateWorker.mLocalMediaManager.getPackageName()) |
| .thenReturn(TEST_DEVICE_PACKAGE_NAME2); |
| mMediaDeviceUpdateWorker.onSlicePinned(); |
| |
| assertThat(mMediaDeviceUpdateWorker.mLocalMediaManager.getPackageName()).matches( |
| TEST_DEVICE_PACKAGE_NAME2); |
| } |
| } |