Add support for testing InboxStyle notifications.

Adding new getTextLines method in the ShadowNotification that allows to get text lines set by the `InboxStyle.addLine()`. PiperOrigin-RevId: 590002758
robolectric · Dec 12, 2023 · 00e219a · 00e219a
1 parent d023d24
commit 00e219a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/robolectric/src/test/java/org/robolectric/shadows/ShadowNotificationBuilderTestBase.java b/robolectric/src/test/java/org/robolectric/shadows/ShadowNotificationBuilderTestBase.java
@@ -247,4 +247,22 @@ public void withBigPictureStyle() {
 
     assertThat(shadowOf(notification).getBigPicture().sameAs(bigPicture)).isTrue();
   }
+
+  @Test
+  public void withInboxStyle() {
+    Notification notification =
+        builder
+            .setStyle(
+                new Notification.InboxStyle(builder)
+                    .addLine("Line1")
+                    .addLine("Line2")
+                    .setBigContentTitle("Title")
+                    .setSummaryText("Summary"))
+            .build();
+
+    assertThat(shadowOf(notification).getBigContentTitle().toString()).isEqualTo("Title");
+    assertThat(shadowOf(notification).getBigContentText().toString()).isEqualTo("Summary");
+    assertThat(shadowOf(notification).getBigPicture()).isNull();
+    assertThat(shadowOf(notification).getTextLines()).containsExactly("Line1", "Line2");
+  }
 }
diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowNotification.java b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowNotification.java
@@ -14,8 +14,10 @@
 import android.widget.ImageView;
 import android.widget.ProgressBar;
 import android.widget.TextView;
+import com.google.common.collect.ImmutableList;
 import java.io.ByteArrayOutputStream;
 import java.io.PrintStream;
+import java.util.List;
 import org.robolectric.RuntimeEnvironment;
 import org.robolectric.annotation.Implements;
 import org.robolectric.annotation.RealObject;
@@ -76,6 +78,15 @@ public CharSequence getBigContentText() {
     }
   }
 
+  public List<CharSequence> getTextLines() {
+    CharSequence[] linesArray =
+        realNotification.extras.getCharSequenceArray(Notification.EXTRA_TEXT_LINES);
+    if (linesArray == null) {
+      return ImmutableList.of();
+    }
+    return ImmutableList.copyOf(linesArray);
+  }
+
   public Bitmap getBigPicture() {
     if (RuntimeEnvironment.getApiLevel() >= Build.VERSION_CODES.N) {
       return realNotification.extras.getParcelable(Notification.EXTRA_PICTURE);