path: root/robolectric/src/test/java/org/robolectric/shadows/ShadowNotificationBuilderTestBase.java

package org.robolectric.shadows;

import static android.os.Build.VERSION_CODES.JELLY_BEAN_MR1;
import static android.os.Build.VERSION_CODES.JELLY_BEAN_MR2;
import static android.os.Build.VERSION_CODES.M;
import static android.os.Build.VERSION_CODES.N;
import static com.google.common.truth.Truth.assertThat;
import static org.robolectric.Shadows.shadowOf;

import android.app.Notification;
import android.app.PendingIntent;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.graphics.drawable.Icon;
import android.text.SpannableString;
import androidx.test.core.app.ApplicationProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.R;
import org.robolectric.annotation.Config;

@RunWith(AndroidJUnit4.class)
public abstract class ShadowNotificationBuilderTestBase {
  private final Notification.Builder builder =
      new Notification.Builder(ApplicationProvider.getApplicationContext());

  @Test
  public void build_setsContentTitleOnNotification() {
    Notification notification = builder.setContentTitle("Hello").build();
    assertThat(shadowOf(notification).getContentTitle().toString()).isEqualTo("Hello");
  }

  @Test
  public void build_whenSetOngoingNotSet_leavesSetOngoingAsFalse() {
    Notification notification = builder.build();
    assertThat(shadowOf(notification).isOngoing()).isFalse();
  }

  @Test
  public void build_whenSetOngoing_setsOngoingToTrue() {
    Notification notification = builder.setOngoing(true).build();
    assertThat(shadowOf(notification).isOngoing()).isTrue();
  }

  @Test
  @Config(minSdk = JELLY_BEAN_MR1)
  public void build_whenShowWhenNotSet_setsShowWhenOnNotificationToTrue() {
    Notification notification = builder.setWhen(100).setShowWhen(true).build();

    assertThat(shadowOf(notification).isWhenShown()).isTrue();
  }

  @Test
  @Config(minSdk = JELLY_BEAN_MR1)
  public void build_setShowWhenOnNotification() {
    Notification notification = builder.setShowWhen(false).build();

    assertThat(shadowOf(notification).isWhenShown()).isFalse();
  }

  @Test
  public void build_setsContentTextOnNotification() {
    Notification notification = builder.setContentText("Hello Text").build();

    assertThat(shadowOf(notification).getContentText().toString()).isEqualTo("Hello Text");
  }

  @Test
  public void build_setsTickerOnNotification() {
    Notification notification = builder.setTicker("My ticker").build();

    assertThat(notification.tickerText.toString()).isEqualTo("My ticker");
  }

  @Test
  public void build_setsContentInfoOnNotification() {
    builder.setContentInfo("11");
    Notification notification = builder.build();
    assertThat(shadowOf(notification).getContentInfo().toString()).isEqualTo("11");
  }

  @Test
  @Config(minSdk = M)
  public void build_setsIconOnNotification() {
    Notification notification = builder.setSmallIcon(R.drawable.an_image).build();

    assertThat(notification.getSmallIcon().getResId()).isEqualTo(R.drawable.an_image);
  }

  @Test
  public void build_setsWhenOnNotification() {
    Notification notification = builder.setWhen(11L).build();

    assertThat(notification.when).isEqualTo(11L);
  }

  @Test
  public void build_setsProgressOnNotification_true() {
    Notification notification = builder.setProgress(36, 57, true).build();
    // If indeterminate then max and progress values are ignored.
    assertThat(shadowOf(notification).isIndeterminate()).isTrue();
  }

  @Test
  public void build_setsProgressOnNotification_false() {
    Notification notification = builder.setProgress(50, 10, false).build();

    assertThat(shadowOf(notification).getMax()).isEqualTo(50);
    assertThat(shadowOf(notification).getProgress()).isEqualTo(10);
    assertThat(shadowOf(notification).isIndeterminate()).isFalse();
  }

  @Test
  @Config(minSdk = JELLY_BEAN_MR1)
  public void build_setsUsesChronometerOnNotification_true() {
    Notification notification = builder.setUsesChronometer(true).setWhen(10).setShowWhen(true).build();

    assertThat(shadowOf(notification).usesChronometer()).isTrue();
  }

  @Test
  @Config(minSdk = JELLY_BEAN_MR1)
  public void build_setsUsesChronometerOnNotification_false() {
    Notification notification = builder.setUsesChronometer(false).setWhen(10).setShowWhen(true).build();

    assertThat(shadowOf(notification).usesChronometer()).isFalse();
  }

  @Test @Config(maxSdk = M)
  public void build_handlesNullContentTitle() {
    Notification notification = builder.setContentTitle(null).build();

    assertThat(shadowOf(notification).getContentTitle().toString()).isEmpty();
  }

  @Test @Config(minSdk = N)
  public void build_handlesNullContentTitle_atLeastN() {
    Notification notification = builder.setContentTitle(null).build();

    assertThat(shadowOf(notification).getContentTitle()).isNull();
  }

  @Test @Config(maxSdk = M)
  public void build_handlesNullContentText() {
    Notification notification = builder.setContentText(null).build();

    assertThat(shadowOf(notification).getContentText().toString()).isEmpty();
  }

  @Test @Config(minSdk = N)
  public void build_handlesNullContentText_atLeastN() {
    Notification notification = builder.setContentText(null).build();

    assertThat(shadowOf(notification).getContentText()).isNull();
  }

  @Test
  public void build_handlesNullTicker() {
    Notification notification = builder.setTicker(null).build();

    assertThat(notification.tickerText).isNull();
  }

  @Test @Config(maxSdk = M)
  public void build_handlesNullContentInfo() {
    Notification notification = builder.setContentInfo(null).build();

    assertThat(shadowOf(notification).getContentInfo().toString()).isEmpty();
  }

  @Test @Config(minSdk = N)
  public void build_handlesNullContentInfo_atLeastN() {
    Notification notification = builder.setContentInfo(null).build();

    assertThat(shadowOf(notification).getContentInfo()).isNull();
  }

  @Test
  @Config(maxSdk = M)
  public void build_handlesNonStringContentText() {
    Notification notification = builder.setContentText(new SpannableString("Hello")).build();

    assertThat(shadowOf(notification).getContentText().toString()).isEqualTo("Hello");
  }

  @Test
  @Config(minSdk = N)
  public void build_handlesNonStringContentText_atLeastN() {
    Notification notification = builder.setContentText(new SpannableString("Hello")).build();

    assertThat(shadowOf(notification).getContentText().toString()).isEqualTo("Hello");
  }

  @Test
  @Config(maxSdk = M)
  public void build_handlesNonStringContentTitle() {
    Notification notification = builder.setContentTitle(new SpannableString("My title")).build();

    assertThat(shadowOf(notification).getContentTitle().toString()).isEqualTo("My title");
  }

  @Test
  @Config(minSdk = N)
  public void build_handlesNonStringContentTitle_atLeastN() {
    Notification notification = builder.setContentTitle(new SpannableString("My title")).build();

    assertThat(shadowOf(notification).getContentTitle().toString()).isEqualTo("My title");
  }

  @Test
  @Config(minSdk = JELLY_BEAN_MR2)
  public void build_addsActionToNotification() throws Exception {
    PendingIntent action =
        PendingIntent.getBroadcast(ApplicationProvider.getApplicationContext(), 0, null, 0);
    Notification notification = builder.addAction(0, "Action", action).build();

    assertThat(notification.actions[0].actionIntent).isEqualTo(action);
  }

  @Test
  public void withBigTextStyle() {
    Notification notification = builder.setStyle(new Notification.BigTextStyle(builder)
        .bigText("BigText")
        .setBigContentTitle("Title")
        .setSummaryText("Summary"))
        .build();

    assertThat(shadowOf(notification).getBigText().toString()).isEqualTo("BigText");
    assertThat(shadowOf(notification).getBigContentTitle().toString()).isEqualTo("Title");
    assertThat(shadowOf(notification).getBigContentText().toString()).isEqualTo("Summary");
    assertThat(shadowOf(notification).getBigPicture()).isNull();
  }

  @Test
  @Config(minSdk = M)
  public void withBigPictureStyle() {
    Bitmap bigPicture =
        BitmapFactory.decodeResource(
            ApplicationProvider.getApplicationContext().getResources(), R.drawable.an_image);

    Icon bigLargeIcon = Icon.createWithBitmap(bigPicture);
    Notification notification = builder.setStyle(new Notification.BigPictureStyle(builder)
        .bigPicture(bigPicture)
        .bigLargeIcon(bigLargeIcon))
        .build();

    assertThat(shadowOf(notification).getBigPicture().sameAs(bigPicture)).isTrue();
  }
}