Merge "[Audiosharing] Fix tests for hysteresis mode" into main (fba28fab) · Commits · e / os / android_packages_apps_Settings

tests/robotests/src/com/android/settings/bluetooth/AvailableMediaBluetoothDeviceUpdaterTest.java

+55 −13

Original line number	Diff line number	Diff line
		@@ -40,6 +40,7 @@ import com.android.settings.testutils.shadow.ShadowBluetoothAdapter;
		import com.android.settings.testutils.shadow.ShadowBluetoothUtils;
		import com.android.settingslib.bluetooth.CachedBluetoothDevice;
		import com.android.settingslib.bluetooth.CachedBluetoothDeviceManager;
		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcast;
		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant;
		import com.android.settingslib.bluetooth.LocalBluetoothManager;
		import com.android.settingslib.bluetooth.LocalBluetoothProfileManager;
		@@ -84,6 +85,7 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		@Mock private CachedBluetoothDeviceManager mCachedDeviceManager;
		@Mock private LocalBluetoothProfileManager mProfileManager;
		@Mock private LocalBluetoothLeBroadcastAssistant mAssistant;
		@Mock private LocalBluetoothLeBroadcast mBroadcast;
		@Mock private BluetoothLeBroadcastReceiveState mBroadcastReceiveState;

		private Context mContext;
		@@ -102,6 +104,7 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		ShadowBluetoothUtils.sLocalBluetoothManager = mLocalBtManager;
		mLocalBtManager = Utils.getLocalBtManager(mContext);
		when(mProfileManager.getLeAudioBroadcastAssistantProfile()).thenReturn(mAssistant);
		when(mProfileManager.getLeAudioBroadcastProfile()).thenReturn(mBroadcast);
		when(mLocalBtManager.getProfileManager()).thenReturn(mProfileManager);
		when(mLocalBtManager.getCachedDeviceManager()).thenReturn(mCachedDeviceManager);
		mShadowBluetoothAdapter = Shadow.extract(BluetoothAdapter.getDefaultAdapter());
		@@ -267,8 +270,7 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void
		onProfileConnectionStateChanged_leaConnected_notInCallSharingFlagOff_addsPreference() {
		public void onProfileConnectionStateChanged_leaConnected_notInCallSharingFlagOff_addPref() {
		mSetFlagsRule.disableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		@@ -288,8 +290,7 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void
		onProfileConnectionStateChanged_leaConnected_notInCallNotInSharing_addsPreference() {
		public void onProfileConnectionStateChanged_leaConnected_notInCallNotInSharing_addPref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		@@ -306,16 +307,13 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_inCallSharingFlagOff_addsPreference() {
		public void onProfileConnectionStateChanged_leaConnected_inCallSharingFlagOff_addPref() {
		mSetFlagsRule.disableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mAudioManager.setMode(AudioManager.MODE_IN_CALL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		.thenReturn(true);
		when(mCachedBluetoothDevice.isConnectedLeAudioDevice()).thenReturn(true);
		when(mAssistant.getAllSources(any())).thenReturn(ImmutableList.of(mBroadcastReceiveState));
		List<Long> bisSyncState = new ArrayList<>();
		bisSyncState.add(1L);
		when(mBroadcastReceiveState.getBisSyncState()).thenReturn(bisSyncState);
		when(mAssistant.getAllSources(any())).thenReturn(ImmutableList.of());

		mBluetoothDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		@@ -326,7 +324,7 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_inCallNotInSharing_addsPreference() {
		public void onProfileConnectionStateChanged_leaConnected_inCallNotInSharing_addPref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mAudioManager.setMode(AudioManager.MODE_IN_CALL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		@@ -343,9 +341,9 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void
		onProfileConnectionStateChanged_leaDeviceConnected_notInCallInSharing_removesPref() {
		public void onProfileConnectionStateChanged_leaConnected_notInCallInSharing_removePref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mSetFlagsRule.disableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		.thenReturn(true);
		@@ -365,8 +363,31 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaDeviceConnected_inCallInSharing_removesPref() {
		public void
		onProfileConnectionStateChanged_leaConnected_noInCallInSharing_hysteresis_removePref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mSetFlagsRule.enableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		.thenReturn(true);
		when(mCachedBluetoothDevice.isConnectedLeAudioDevice()).thenReturn(true);
		when(mCachedBluetoothDevice.isConnectedA2dpDevice()).thenReturn(true);
		when(mAssistant.getAllSources(any())).thenReturn(ImmutableList.of(mBroadcastReceiveState));
		when(mBroadcast.getLatestBroadcastId()).thenReturn(1);
		when(mBroadcastReceiveState.getBroadcastId()).thenReturn(1);

		mBluetoothDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);

		verify(mBluetoothDeviceUpdater).removePreference(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_inCallSharing_removePref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mSetFlagsRule.disableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		.thenReturn(true);
		@@ -385,6 +406,27 @@ public class AvailableMediaBluetoothDeviceUpdaterTest {
		verify(mBluetoothDeviceUpdater).removePreference(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_inCallSharing_hysteresis_removePref() {
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		mSetFlagsRule.enableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		mAudioManager.setMode(AudioManager.MODE_NORMAL);
		when(mBluetoothDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class)))
		.thenReturn(true);
		when(mCachedBluetoothDevice.isConnectedLeAudioDevice()).thenReturn(true);
		when(mCachedBluetoothDevice.isConnectedHfpDevice()).thenReturn(true);
		when(mAssistant.getAllSources(any())).thenReturn(ImmutableList.of(mBroadcastReceiveState));
		when(mBroadcast.getLatestBroadcastId()).thenReturn(1);
		when(mBroadcastReceiveState.getBroadcastId()).thenReturn(1);

		mBluetoothDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);

		verify(mBluetoothDeviceUpdater).removePreference(mCachedBluetoothDevice);
		}

		@Test
		public void
		onProfileConnectionStateChanged_deviceIsNotInList_notInCall_invokesRemovePreference() {

tests/robotests/src/com/android/settings/connecteddevice/audiosharing/AudioSharingBluetoothDeviceUpdaterTest.java

+130 −14

Original line number	Diff line number	Diff line
		@@ -51,6 +51,7 @@ import com.android.settings.testutils.shadow.ShadowBluetoothUtils;
		import com.android.settings.testutils.shadow.ShadowThreadUtils;
		import com.android.settingslib.bluetooth.CachedBluetoothDevice;
		import com.android.settingslib.bluetooth.CachedBluetoothDeviceManager;
		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcast;
		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastAssistant;
		import com.android.settingslib.bluetooth.LocalBluetoothManager;
		import com.android.settingslib.bluetooth.LocalBluetoothProfileManager;
		@@ -99,6 +100,7 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		@Mock private CachedBluetoothDeviceManager mCachedDeviceManager;
		@Mock private LocalBluetoothProfileManager mLocalBtProfileManager;
		@Mock private LocalBluetoothLeBroadcastAssistant mAssistant;
		@Mock private LocalBluetoothLeBroadcast mBroadcast;
		@Mock private BluetoothLeBroadcastReceiveState mState;

		private Context mContext;
		@@ -122,9 +124,7 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		when(mLocalBtManager.getCachedDeviceManager()).thenReturn(mCachedDeviceManager);
		when(mLocalBtManager.getProfileManager()).thenReturn(mLocalBtProfileManager);
		when(mLocalBtProfileManager.getLeAudioBroadcastAssistantProfile()).thenReturn(mAssistant);
		List<Long> bisSyncState = new ArrayList<>();
		bisSyncState.add(1L);
		when(mState.getBisSyncState()).thenReturn(bisSyncState);
		when(mLocalBtProfileManager.getLeAudioBroadcastProfile()).thenReturn(mBroadcast);
		Pair<Drawable, String> pairs = new Pair<>(mDrawable, TEST_DEVICE_NAME);
		doReturn(TEST_DEVICE_NAME).when(mCachedBluetoothDevice).getName();
		doReturn(mBluetoothDevice).when(mCachedBluetoothDevice).getDevice();
		@@ -152,7 +152,26 @@ public class AudioSharingBluetoothDeviceUpdaterTest {

		@Test
		public void onProfileConnectionStateChanged_leaDeviceConnected_flagOff_removesPref() {
		setupPreferenceMapWithDevice();
		setupPreferenceMapWithDevice(false);

		mSetFlagsRule.disableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		mDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);
		shadowOf(Looper.getMainLooper()).idle();

		verify(mDevicePreferenceCallback).onDeviceRemoved(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_flagOff_hysteresisMode_removesPref() {
		setupPreferenceMapWithDevice(true);

		mSetFlagsRule.disableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);
		@@ -170,8 +189,27 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaDeviceConnected_noSource_removesPref() {
		setupPreferenceMapWithDevice();
		public void onProfileConnectionStateChanged_leaConnected_noSource_removesPref() {
		setupPreferenceMapWithDevice(false);

		when(mAssistant.getAllSources(mBluetoothDevice)).thenReturn(ImmutableList.of());
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		mDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);
		shadowOf(Looper.getMainLooper()).idle();

		verify(mDevicePreferenceCallback).onDeviceRemoved(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_noSource_hysteresisMode_removesPref() {
		setupPreferenceMapWithDevice(true);

		when(mAssistant.getAllSources(mBluetoothDevice)).thenReturn(ImmutableList.of());
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);
		@@ -190,7 +228,7 @@ public class AudioSharingBluetoothDeviceUpdaterTest {

		@Test
		public void onProfileConnectionStateChanged_deviceIsNotInList_removesPref() {
		setupPreferenceMapWithDevice();
		setupPreferenceMapWithDevice(false);

		mCachedDevices.clear();
		when(mCachedDeviceManager.getCachedDevicesCopy()).thenReturn(mCachedDevices);
		@@ -209,8 +247,28 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaDeviceDisconnected_removesPref() {
		setupPreferenceMapWithDevice();
		public void onProfileConnectionStateChanged_deviceIsNotInList_hysteresisMode_removesPref() {
		setupPreferenceMapWithDevice(true);

		mCachedDevices.clear();
		when(mCachedDeviceManager.getCachedDevicesCopy()).thenReturn(mCachedDevices);
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		mDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);
		shadowOf(Looper.getMainLooper()).idle();

		verify(mDevicePreferenceCallback).onDeviceRemoved(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaDisconnected_removesPref() {
		setupPreferenceMapWithDevice(false);

		when(mDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class))).thenReturn(false);
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);
		@@ -228,8 +286,45 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaDeviceDisconnecting_removesPref() {
		setupPreferenceMapWithDevice();
		public void onProfileConnectionStateChanged_leaDisconnected_hysteresisMode_removesPref() {
		setupPreferenceMapWithDevice(true);

		when(mDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class))).thenReturn(false);
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		mDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_DISCONNECTED,
		BluetoothProfile.LE_AUDIO);
		shadowOf(Looper.getMainLooper()).idle();

		verify(mDevicePreferenceCallback).onDeviceRemoved(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaDisconnecting_removesPref() {
		setupPreferenceMapWithDevice(false);
		doReturn(false).when(mCachedBluetoothDevice).isConnectedLeAudioDevice();
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		mDeviceUpdater.onProfileConnectionStateChanged(
		mCachedBluetoothDevice,
		BluetoothProfile.STATE_CONNECTED,
		BluetoothProfile.LE_AUDIO);
		shadowOf(Looper.getMainLooper()).idle();

		verify(mDevicePreferenceCallback).onDeviceRemoved(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaDisconnecting_hysteresisMode_removesPref() {
		setupPreferenceMapWithDevice(true);
		doReturn(false).when(mCachedBluetoothDevice).isConnectedLeAudioDevice();
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);

		@@ -246,9 +341,20 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		}

		@Test
		public void onProfileConnectionStateChanged_leaDeviceConnected_hasSource_addsPreference() {
		public void onProfileConnectionStateChanged_leaConnected_hasSource_addsPref() {
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);
		setupPreferenceMapWithDevice();
		setupPreferenceMapWithDevice(false);

		verify(mDevicePreferenceCallback).onDeviceAdded(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		assertThat(((BluetoothDevicePreference) captor.getValue()).getBluetoothDevice())
		.isEqualTo(mCachedBluetoothDevice);
		}

		@Test
		public void onProfileConnectionStateChanged_leaConnected_hasSource_hysteresisMode_addsPref() {
		ArgumentCaptor<Preference> captor = ArgumentCaptor.forClass(Preference.class);
		setupPreferenceMapWithDevice(true);

		verify(mDevicePreferenceCallback).onDeviceAdded(captor.capture());
		assertThat(captor.getValue() instanceof BluetoothDevicePreference).isTrue();
		@@ -275,9 +381,19 @@ public class AudioSharingBluetoothDeviceUpdaterTest {
		verify(mDevicePreferenceCallback).onDeviceClick(preference);
		}

		private void setupPreferenceMapWithDevice() {
		private void setupPreferenceMapWithDevice(boolean hysteresisModeOn) {
		// Add device to preferenceMap
		mSetFlagsRule.enableFlags(Flags.FLAG_ENABLE_LE_AUDIO_SHARING);
		if (hysteresisModeOn) {
		mSetFlagsRule.enableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		when(mBroadcast.getLatestBroadcastId()).thenReturn(1);
		when(mState.getBroadcastId()).thenReturn(1);
		} else {
		mSetFlagsRule.disableFlags(Flags.FLAG_AUDIO_SHARING_HYSTERESIS_MODE_FIX);
		List<Long> bisSyncState = new ArrayList<>();
		bisSyncState.add(1L);
		when(mState.getBisSyncState()).thenReturn(bisSyncState);
		}
		when(mAssistant.getAllSources(mBluetoothDevice)).thenReturn(ImmutableList.of(mState));
		when(mDeviceUpdater.isDeviceConnected(any(CachedBluetoothDevice.class))).thenReturn(true);
		doReturn(true).when(mCachedBluetoothDevice).isConnectedLeAudioDevice();