Merge "Visit Uris related to Notification style extras" into udc-qpr-dev am:... (13374e78) · Commits · e / os / android_frameworks_base

core/java/android/app/Notification.java

+8 −9

Original line number	Diff line number	Diff line
		@@ -2900,11 +2900,6 @@ public class Notification implements Parcelable
		}
		}

		final Person person = extras.getParcelable(EXTRA_MESSAGING_PERSON, Person.class);
		if (person != null) {
		person.visitUris(visitor);
		}

		final RemoteInputHistoryItem[] history = extras.getParcelableArray(
		Notification.EXTRA_REMOTE_INPUT_HISTORY_ITEMS,
		RemoteInputHistoryItem.class);
		@@ -2916,9 +2911,14 @@ public class Notification implements Parcelable
		}
		}
		}

		// Extras for MessagingStyle. We visit them even if not isStyle(MessagingStyle), since
		// Notification Listeners might use directly (without the isStyle check).
		final Person person = extras.getParcelable(EXTRA_MESSAGING_PERSON, Person.class);
		if (person != null) {
		person.visitUris(visitor);
		}

		if (isStyle(MessagingStyle.class) && extras != null) {
		final Parcelable[] messages = extras.getParcelableArray(EXTRA_MESSAGES,
		Parcelable.class);
		if (!ArrayUtils.isEmpty(messages)) {
		@@ -2938,9 +2938,8 @@ public class Notification implements Parcelable
		}

		visitIconUri(visitor, extras.getParcelable(EXTRA_CONVERSATION_ICON, Icon.class));
		}

		if (isStyle(CallStyle.class) & extras != null) {
		// Extras for CallStyle (same reason for visiting without checking isStyle).
		Person callPerson = extras.getParcelable(EXTRA_CALL_PERSON, Person.class);
		if (callPerson != null) {
		callPerson.visitUris(visitor);

services/tests/uiservicestests/src/com/android/server/notification/NotificationManagerServiceTest.java

+43 −0

Original line number	Diff line number	Diff line
		@@ -6043,6 +6043,49 @@ public class NotificationManagerServiceTest extends UiServiceTestCase {
		verify(visitor, times(1)).accept(eq(verificationIcon.getUri()));
		}

		@Test
		public void testVisitUris_styleExtrasWithoutStyle() {
		Notification notification = new Notification.Builder(mContext, "a")
		.setSmallIcon(android.R.drawable.sym_def_app_icon)
		.build();

		Notification.MessagingStyle messagingStyle = new Notification.MessagingStyle(
		personWithIcon("content://user"))
		.addHistoricMessage(new Notification.MessagingStyle.Message("Heyhey!",
		System.currentTimeMillis(),
		personWithIcon("content://historicalMessenger")))
		.addMessage(new Notification.MessagingStyle.Message("Are you there",
		System.currentTimeMillis(),
		personWithIcon("content://messenger")))
		.setShortcutIcon(
		Icon.createWithContentUri("content://conversationShortcut"));
		messagingStyle.addExtras(notification.extras); // Instead of Builder.setStyle(style).

		Notification.CallStyle callStyle = Notification.CallStyle.forOngoingCall(
		personWithIcon("content://caller"),
		PendingIntent.getActivity(mContext, 0, new Intent(),
		PendingIntent.FLAG_IMMUTABLE))
		.setVerificationIcon(Icon.createWithContentUri("content://callVerification"));
		callStyle.addExtras(notification.extras); // Same.

		Consumer<Uri> visitor = (Consumer<Uri>) spy(Consumer.class);
		notification.visitUris(visitor);

		verify(visitor).accept(eq(Uri.parse("content://user")));
		verify(visitor).accept(eq(Uri.parse("content://historicalMessenger")));
		verify(visitor).accept(eq(Uri.parse("content://messenger")));
		verify(visitor).accept(eq(Uri.parse("content://conversationShortcut")));
		verify(visitor).accept(eq(Uri.parse("content://caller")));
		verify(visitor).accept(eq(Uri.parse("content://callVerification")));
		}

		private static Person personWithIcon(String iconUri) {
		return new Person.Builder()
		.setName("Mr " + iconUri)
		.setIcon(Icon.createWithContentUri(iconUri))
		.build();
		}

		@Test
		public void testVisitUris_wearableExtender() {
		Icon actionIcon = Icon.createWithContentUri("content://media/action");