Adding JUnit tests for MediaNotificationManager

chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationManagerNotificationTest.java

Index: chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationManagerNotificationTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationManagerNotificationTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationManagerNotificationTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..e6090bbdcde211d929bdee1ab4862ada6809b313
--- /dev/null
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationManagerNotificationTest.java
@@ -0,0 +1,216 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.chrome.browser.media.ui;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+
+import android.app.Notification;
+import android.content.Intent;
+import android.graphics.Bitmap;
+import android.os.Build;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import org.robolectric.RuntimeEnvironment;
+import org.robolectric.Shadows;
+import org.robolectric.annotation.Config;
+import org.robolectric.shadows.ShadowNotification;
+
+import org.chromium.base.BaseChromiumApplication;
+import org.chromium.content_public.common.MediaMetadata;
+import org.chromium.testing.local.LocalRobolectricTestRunner;
+
+/**
+ * JUnit tests for checking MediaNotificationManager presents correct notification to Android
+ * NotificationManager.
+ */
+@RunWith(LocalRobolectricTestRunner.class)
+@Config(manifest = Config.NONE, application = BaseChromiumApplication.class,
+        shadows = MediaNotificationTestShadowResources.class)
+public class MediaNotificationManagerNotificationTest extends MediaNotificationManagerTestBase {
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMetadata_PreN_NonEmptyArtistAndAlbum() {
+        MediaNotificationManager.sOverrideIsRunningNForTesting = false;
+
+        mMediaNotificationInfoBuilder.setMetadata(new MediaMetadata("title", "artist", "album"));
+        mMediaNotificationInfoBuilder.setOrigin("https://example.com/");
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        ShadowNotification shadowNotification = Shadows.shadowOf(notification);
+
+        assertEquals("title", shadowNotification.getContentTitle());
+        assertEquals("artist - album", shadowNotification.getContentText());
+        if (hasNApis()) {
+            assertEquals("https://example.com/",
+                    notification.extras.getString(Notification.EXTRA_SUB_TEXT));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMetadata_PreN_EmptyArtistAndAlbum() {
+        MediaNotificationManager.sOverrideIsRunningNForTesting = false;
+
+        mMediaNotificationInfoBuilder.setMetadata(new MediaMetadata("title", "", ""));
+        mMediaNotificationInfoBuilder.setOrigin("https://example.com/");
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        ShadowNotification shadowNotification = Shadows.shadowOf(notification);
+
+        assertEquals(info.metadata.getTitle(), shadowNotification.getContentTitle());
+        assertEquals(info.origin, shadowNotification.getContentText());
+        if (hasNApis()) {
+            assertEquals(null, notification.extras.getString(Notification.EXTRA_SUB_TEXT));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMetadata_AtLeastN_EmptyArtistAndAlbum() {
+        MediaNotificationManager.sOverrideIsRunningNForTesting = true;
+
+        mMediaNotificationInfoBuilder.setMetadata(new MediaMetadata("title", "", ""));
+        mMediaNotificationInfoBuilder.setOrigin("https://example.com/");
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        ShadowNotification shadowNotification = Shadows.shadowOf(notification);
+
+        assertEquals(info.metadata.getTitle(), shadowNotification.getContentTitle());
+        assertEquals("", shadowNotification.getContentText());
+        if (hasNApis()) {
+            assertEquals(info.origin, notification.extras.getString(Notification.EXTRA_SUB_TEXT));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectLargeIcon_WithLargeIcon() {
+        Bitmap largeIcon = Bitmap.createBitmap(100, 100, Bitmap.Config.ARGB_8888);
+        mMediaNotificationInfoBuilder.setNotificationLargeIcon(largeIcon);
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        if (hasNApis()) {
+            assertTrue(largeIcon.sameAs(iconToBitmap(
+                    notification.extras.getParcelable(Notification.EXTRA_LARGE_ICON))));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectLargeIcon_WithoutLargeIcon_AtLeastN() {
+        mMediaNotificationInfoBuilder.setNotificationLargeIcon(null);
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        if (hasNApis()) {
+            assertNull(notification.extras.getParcelable(Notification.EXTRA_LARGE_ICON));
+        }
+        assertNull(getManager().mDefaultNotificationLargeIcon);
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectLargeIcon_WithoutLargeIcon_PreN() {
+        MediaNotificationManager.sOverrideIsRunningNForTesting = false;
+        assertNull(getManager().mDefaultNotificationLargeIcon);
+
+        mMediaNotificationInfoBuilder.setNotificationLargeIcon(null);
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        assertNotNull(getManager().mDefaultNotificationLargeIcon);
+        if (hasNApis()) {
+            assertTrue(getManager().mDefaultNotificationLargeIcon.sameAs(iconToBitmap(
+                    notification.extras.getParcelable(Notification.EXTRA_LARGE_ICON))));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectLargeIcon_DontSupportPlayPause() {
+        Bitmap largeIcon = Bitmap.createBitmap(100, 100, Bitmap.Config.ARGB_8888);
+        mMediaNotificationInfoBuilder.setNotificationLargeIcon(largeIcon).setActions(0);
+
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        if (hasNApis()) {
+            assertNull(notification.extras.getParcelable(Notification.EXTRA_LARGE_ICON));
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMiscInfo() {
+        mMediaNotificationInfoBuilder.setNotificationSmallIcon(1 /* resId */)
+                .setActions(0)
+                .setContentIntent(new Intent());
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        ShadowNotification shadowNotification = Shadows.shadowOf(notification);
+
+        assertFalse(shadowNotification.isWhenShown());
+        assertFalse(shadowNotification.isOngoing());
+        if (hasNApis()) {
+            assertNotNull(notification.getSmallIcon());
+            assertFalse((notification.flags & Notification.FLAG_AUTO_CANCEL) != 0);
+            assertTrue((notification.flags & Notification.FLAG_LOCAL_ONLY) != 0);
+            assertEquals(NOTIFICATION_GROUP_NAME, notification.getGroup());
+            assertTrue(notification.isGroupSummary());
+            assertNull(notification.deleteIntent);
+            assertNotNull(notification.contentIntent);
+            assertEquals(Notification.VISIBILITY_PRIVATE, notification.visibility);
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMiscInfo_SupportsSwipeAway() {
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        ShadowNotification shadowNotification = Shadows.shadowOf(notification);
+
+        assertTrue(shadowNotification.isOngoing());
+        if (hasNApis()) {
+            assertNotNull(notification.deleteIntent);
+        }
+    }
+
+    @Test
+    public void updateNotificationBuilderDisplaysCorrectMiscInfo_Private() {
+        mMediaNotificationInfoBuilder.setPrivate(false);
+        MediaNotificationInfo info = mMediaNotificationInfoBuilder.build();
+        Notification notification = updateNotificationBuilderAndBuild(info);
+
+        if (hasNApis()) {
+            assertEquals(Notification.VISIBILITY_PUBLIC, notification.visibility);
+        }
+    }
+
+    private Notification updateNotificationBuilderAndBuild(MediaNotificationInfo info) {
+        getManager().mMediaNotificationInfo = info;
+
+        // This is the fake implementation to ensure |mMediaSession| is non-null.
+        //
+        // TODO(zqzhang): move around code so that updateNotification() doesn't need a MediaSession.
+        getManager().updateMediaSession();
+        getManager().updateNotificationBuilder();
+
+        return getManager().mNotificationBuilder.build();
+    }
+
+    private boolean hasNApis() {
+        return RuntimeEnvironment.getApiLevel() >= Build.VERSION_CODES.N;
+    }
+}