Merge "[LE Broadcast]Add check for null point exception" (7aede2a1) · Commits · e / os / android_frameworks_base

packages/SystemUI/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialog.java

+34 −11

Original line number	Diff line number	Diff line
		@@ -36,6 +36,7 @@ import android.widget.TextView;

		import androidx.core.graphics.drawable.IconCompat;

		import com.android.internal.annotations.VisibleForTesting;
		import com.android.settingslib.qrcode.QrCodeGenerator;
		import com.android.systemui.R;
		import com.android.systemui.broadcast.BroadcastSender;
		@@ -49,7 +50,7 @@ import com.google.zxing.WriterException;
		*/
		@SysUISingleton
		public class MediaOutputBroadcastDialog extends MediaOutputBaseDialog {
		private static final String TAG = "BroadcastDialog";
		private static final String TAG = "MediaOutputBroadcastDialog";

		private ViewStub mBroadcastInfoArea;
		private ImageView mBroadcastQrCodeView;
		@@ -217,7 +218,8 @@ public class MediaOutputBroadcastDialog extends MediaOutputBaseDialog {
		refreshUi();
		}

		private void refreshUi() {
		@VisibleForTesting
		void refreshUi() {
		setQrCodeView();

		mCurrentBroadcastName = getBroadcastMetadataInfo(METADATA_BROADCAST_NAME);
		@@ -256,7 +258,8 @@ public class MediaOutputBroadcastDialog extends MediaOutputBaseDialog {
		mIsPasswordHide = !mIsPasswordHide;
		}

		private void launchBroadcastUpdatedDialog(boolean isBroadcastCode, String editString) {
		@VisibleForTesting
		void launchBroadcastUpdatedDialog(boolean isBroadcastCode, String editString) {
		final View layout = LayoutInflater.from(mContext).inflate(
		R.layout.media_output_broadcast_update_dialog, null);
		final EditText editText = layout.requireViewById(R.id.broadcast_edit_text);
		@@ -286,7 +289,8 @@ public class MediaOutputBroadcastDialog extends MediaOutputBaseDialog {
		return mMediaOutputController.getBroadcastMetadata();
		}

		private void updateBroadcastInfo(boolean isBroadcastCode, String updatedString) {
		@VisibleForTesting
		void updateBroadcastInfo(boolean isBroadcastCode, String updatedString) {
		Button positiveBtn = mAlertDialog.getButton(AlertDialog.BUTTON_POSITIVE);
		if (positiveBtn != null) {
		positiveBtn.setEnabled(false);
		@@ -377,16 +381,35 @@ public class MediaOutputBroadcastDialog extends MediaOutputBaseDialog {
		}

		private void handleUpdateFailedUi() {
		final Button positiveBtn = mAlertDialog.getButton(AlertDialog.BUTTON_POSITIVE);
		mBroadcastErrorMessage.setVisibility(View.VISIBLE);
		if (mRetryCount < MAX_BROADCAST_INFO_UPDATE) {
		if (positiveBtn != null) {
		positiveBtn.setEnabled(true);
		if (mAlertDialog == null) {
		Log.d(TAG, "handleUpdateFailedUi: mAlertDialog is null");
		return;
		}
		mBroadcastErrorMessage.setText(R.string.media_output_broadcast_update_error);
		int errorMessageStringId = -1;
		boolean enablePositiveBtn = false;
		if (mRetryCount < MAX_BROADCAST_INFO_UPDATE) {
		enablePositiveBtn = true;
		errorMessageStringId = R.string.media_output_broadcast_update_error;
		} else {
		mRetryCount = 0;
		mBroadcastErrorMessage.setText(R.string.media_output_broadcast_last_update_error);
		errorMessageStringId = R.string.media_output_broadcast_last_update_error;
		}

		// update UI
		final Button positiveBtn = mAlertDialog.getButton(AlertDialog.BUTTON_POSITIVE);
		if (positiveBtn != null && enablePositiveBtn) {
		positiveBtn.setEnabled(true);
		}
		if (mBroadcastErrorMessage != null) {
		mBroadcastErrorMessage.setVisibility(View.VISIBLE);
		if (errorMessageStringId > 0) {
		mBroadcastErrorMessage.setText(errorMessageStringId);
		}
		}
		}

		@VisibleForTesting
		int getRetryCount() {
		return mRetryCount;
		}
		}

packages/SystemUI/tests/src/com/android/systemui/media/dialog/MediaOutputBroadcastDialogTest.java

0 → 100644

+185 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright (C) 2023 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.systemui.media.dialog;

		import static com.google.common.truth.Truth.assertThat;

		import static org.mockito.Mockito.any;
		import static org.mockito.Mockito.mock;
		import static org.mockito.Mockito.when;

		import android.app.KeyguardManager;
		import android.media.AudioManager;
		import android.media.session.MediaController;
		import android.media.session.MediaSessionManager;
		import android.media.session.PlaybackState;
		import android.os.PowerExemptionManager;
		import android.testing.AndroidTestingRunner;
		import android.testing.TestableLooper;
		import android.widget.ImageView;
		import android.widget.TextView;

		import androidx.test.filters.SmallTest;

		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcast;
		import com.android.settingslib.bluetooth.LocalBluetoothLeBroadcastMetadata;
		import com.android.settingslib.bluetooth.LocalBluetoothManager;
		import com.android.settingslib.bluetooth.LocalBluetoothProfileManager;
		import com.android.settingslib.media.LocalMediaManager;
		import com.android.settingslib.media.MediaDevice;
		import com.android.systemui.R;
		import com.android.systemui.SysuiTestCase;
		import com.android.systemui.animation.DialogLaunchAnimator;
		import com.android.systemui.broadcast.BroadcastSender;
		import com.android.systemui.flags.FeatureFlags;
		import com.android.systemui.flags.Flags;
		import com.android.systemui.media.nearby.NearbyMediaDevicesManager;
		import com.android.systemui.plugins.ActivityStarter;
		import com.android.systemui.statusbar.notification.collection.notifcollection.CommonNotifCollection;

		import org.junit.After;
		import org.junit.Before;
		import org.junit.Test;
		import org.junit.runner.RunWith;

		import java.nio.charset.StandardCharsets;
		import java.util.ArrayList;
		import java.util.List;
		import java.util.Optional;

		@SmallTest
		@RunWith(AndroidTestingRunner.class)
		@TestableLooper.RunWithLooper
		public class MediaOutputBroadcastDialogTest extends SysuiTestCase {

		private static final String TEST_PACKAGE = "test_package";
		private static final String BROADCAST_NAME_TEST = "Broadcast_name_test";
		private static final String BROADCAST_CODE_TEST = "112233";
		private static final String BROADCAST_CODE_UPDATE_TEST = "11223344";

		// Mock
		private final MediaSessionManager mMediaSessionManager = mock(MediaSessionManager.class);
		private MediaController mMediaController = mock(MediaController.class);
		private PlaybackState mPlaybackState = mock(PlaybackState.class);
		private final LocalBluetoothManager mLocalBluetoothManager = mock(LocalBluetoothManager.class);
		private final LocalBluetoothProfileManager mLocalBluetoothProfileManager = mock(
		LocalBluetoothProfileManager.class);
		private final LocalBluetoothLeBroadcast mLocalBluetoothLeBroadcast = mock(
		LocalBluetoothLeBroadcast.class);
		private final ActivityStarter mStarter = mock(ActivityStarter.class);
		private final BroadcastSender mBroadcastSender = mock(BroadcastSender.class);
		private final LocalMediaManager mLocalMediaManager = mock(LocalMediaManager.class);
		private final MediaDevice mMediaDevice = mock(MediaDevice.class);
		private final CommonNotifCollection mNotifCollection = mock(CommonNotifCollection.class);
		private final DialogLaunchAnimator mDialogLaunchAnimator = mock(DialogLaunchAnimator.class);
		private final NearbyMediaDevicesManager mNearbyMediaDevicesManager = mock(
		NearbyMediaDevicesManager.class);
		private final AudioManager mAudioManager = mock(AudioManager.class);
		private PowerExemptionManager mPowerExemptionManager = mock(PowerExemptionManager.class);
		private KeyguardManager mKeyguardManager = mock(KeyguardManager.class);
		private final LocalBluetoothLeBroadcastMetadata mLocalBluetoothLeBroadcastMetadata =
		mock(LocalBluetoothLeBroadcastMetadata.class);
		private FeatureFlags mFlags = mock(FeatureFlags.class);

		private List<MediaController> mMediaControllers = new ArrayList<>();
		private MediaOutputBroadcastDialog mMediaOutputBroadcastDialog;
		private MediaOutputController mMediaOutputController;
		private final List<String> mFeatures = new ArrayList<>();

		@Before
		public void setUp() {
		when(mLocalBluetoothManager.getProfileManager()).thenReturn(mLocalBluetoothProfileManager);
		when(mLocalBluetoothProfileManager.getLeAudioBroadcastProfile())
		.thenReturn(mLocalBluetoothLeBroadcast);
		when(mMediaController.getPlaybackState()).thenReturn(mPlaybackState);
		when(mPlaybackState.getState()).thenReturn(PlaybackState.STATE_NONE);
		when(mMediaController.getPackageName()).thenReturn(TEST_PACKAGE);
		mMediaControllers.add(mMediaController);
		when(mMediaSessionManager.getActiveSessions(any())).thenReturn(mMediaControllers);
		when(mLocalBluetoothLeBroadcast.getLocalBluetoothLeBroadcastMetaData()).thenReturn(
		mLocalBluetoothLeBroadcastMetadata);
		when(mLocalBluetoothLeBroadcastMetadata.convertToQrCodeString())
		.thenReturn("metadata_test_convert");
		when(mLocalBluetoothLeBroadcast.getProgramInfo()).thenReturn(BROADCAST_NAME_TEST);
		when(mLocalBluetoothLeBroadcast.getBroadcastCode())
		.thenReturn(BROADCAST_CODE_TEST.getBytes(StandardCharsets.UTF_8));

		mMediaOutputController = new MediaOutputController(mContext, TEST_PACKAGE,
		mMediaSessionManager, mLocalBluetoothManager, mStarter,
		mNotifCollection, mDialogLaunchAnimator,
		Optional.of(mNearbyMediaDevicesManager), mAudioManager, mPowerExemptionManager,
		mKeyguardManager, mFlags);
		when(mFlags.isEnabled(Flags.OUTPUT_SWITCHER_ADVANCED_LAYOUT)).thenReturn(false);

		mMediaOutputController.mLocalMediaManager = mLocalMediaManager;
		mMediaOutputBroadcastDialog = new MediaOutputBroadcastDialog(mContext, false,
		mBroadcastSender, mMediaOutputController);
		mMediaOutputBroadcastDialog.show();

		when(mLocalMediaManager.getCurrentConnectedDevice()).thenReturn(mMediaDevice);
		when(mMediaDevice.getFeatures()).thenReturn(mFeatures);
		}

		@After
		public void tearDown() {
		mMediaOutputBroadcastDialog.dismissDialog();
		}

		@Test
		public void refreshUi_checkBroadcastQrCodeView() {
		mMediaOutputBroadcastDialog.refreshUi();
		final ImageView broadcastQrCodeView = mMediaOutputBroadcastDialog.mDialogView
		.requireViewById(R.id.qrcode_view);

		assertThat(broadcastQrCodeView.getDrawable()).isNotNull();
		}

		@Test
		public void refreshUi_checkBroadcastName() {
		mMediaOutputBroadcastDialog.refreshUi();
		final TextView broadcastName = mMediaOutputBroadcastDialog.mDialogView
		.requireViewById(R.id.broadcast_name_summary);

		assertThat(broadcastName.getText().toString()).isEqualTo(BROADCAST_NAME_TEST);
		}

		@Test
		public void refreshUi_checkBroadcastCode() {
		mMediaOutputBroadcastDialog.refreshUi();
		final TextView broadcastCode = mMediaOutputBroadcastDialog.mDialogView
		.requireViewById(R.id.broadcast_code_summary);

		assertThat(broadcastCode.getText().toString()).isEqualTo(BROADCAST_CODE_TEST);
		}

		@Test
		public void updateBroadcastInfo_stopBroadcastFailed_handleFailedUI() {
		mMediaOutputBroadcastDialog.launchBroadcastUpdatedDialog(true, BROADCAST_CODE_UPDATE_TEST);
		when(mLocalBluetoothProfileManager.getLeAudioBroadcastProfile())
		.thenReturn(null);

		mMediaOutputBroadcastDialog.updateBroadcastInfo(true, BROADCAST_CODE_UPDATE_TEST);
		assertThat(mMediaOutputBroadcastDialog.getRetryCount()).isEqualTo(1);

		mMediaOutputBroadcastDialog.updateBroadcastInfo(true, BROADCAST_CODE_UPDATE_TEST);
		assertThat(mMediaOutputBroadcastDialog.getRetryCount()).isEqualTo(2);

		// It will be the MAX Retry Count = 3
		mMediaOutputBroadcastDialog.updateBroadcastInfo(true, BROADCAST_CODE_UPDATE_TEST);
		assertThat(mMediaOutputBroadcastDialog.getRetryCount()).isEqualTo(0);
		}
		}