[Media>UI] Migrate media notification tests to JUnit tests

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

Index: chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationActionsUpdatedTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationActionsUpdatedTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationActionsUpdatedTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..182b2fac7c9f8f1bb998e6d14e5280a6154a7af9
--- /dev/null
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/media/ui/MediaNotificationActionsUpdatedTest.java
@@ -0,0 +1,97 @@
+// 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 org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.annotation.Config;
+
+import org.chromium.base.BaseChromiumApplication;
+import org.chromium.blink.mojom.MediaSessionAction;
+import org.chromium.chrome.R;
+import org.chromium.testing.local.LocalRobolectricTestRunner;
+
+import java.util.HashSet;
+import java.util.Set;
+
+/**
+ * Test of media notifications to see whether the control buttons update when MediaSessionActions
+ * change or the tab navigates.
+ */
+@RunWith(LocalRobolectricTestRunner.class)
+@Config(manifest = Config.NONE, application = BaseChromiumApplication.class,
+        shadows = {MediaNotificationTestShadowResources.class,
+                MediaNotificationTestShadowNotificationManager.class})
+public class MediaNotificationActionsUpdatedTest extends MediaNotificationManagerTestBase {
+    private static final int TAB_ID_1 = 1;
+    private static final int TAB_ID_2 = 2;
+    private static final int HIDE_NOTIFICATION_DELAY_MILLIS =
+            MediaSessionTabHelper.HIDE_NOTIFICATION_DELAY_MILLIS;
+
+    private MediaNotificationTestTabHolder mTabHolder;
+
+    @Before
+    @Override
+    public void setUp() {
+        super.setUp();
+
+        mTabHolder = new MediaNotificationTestTabHolder(TAB_ID_1, "about:blank", "title1");
+    }
+
+    @Test
+    public void testActionsDefaultToNull() {
+        mTabHolder.simulateMediaSessionStateChanged(true, false);
+        assertEquals(new HashSet<Integer>(), getDisplayedActions());
+    }
+
+    @Test
+    public void testActionPropagateProperly() {
+        mTabHolder.simulateMediaSessionActionsChanged(buildActions());
+        mTabHolder.simulateMediaSessionStateChanged(true, false);
+        assertEquals(buildActions(), getDisplayedActions());
+    }
+
+    @Test
+    public void testActionsResetAfterNavigation() {
+        mTabHolder.simulateNavigation("https://example.com/", false);
+        mTabHolder.simulateMediaSessionActionsChanged(buildActions());
+        mTabHolder.simulateMediaSessionStateChanged(true, false);
+        assertEquals(buildActions(), getDisplayedActions());
+
+        mTabHolder.simulateNavigation("https://example1.com/", false);
+        assertEquals(new HashSet<Integer>(), getDisplayedActions());
+    }
+
+    @Test
+    public void testActionsPersistAfterSamePageNavigation() {
+        mTabHolder.simulateNavigation("https://example.com/", false);
+        mTabHolder.simulateMediaSessionActionsChanged(buildActions());
+        mTabHolder.simulateMediaSessionStateChanged(true, false);
+        assertEquals(buildActions(), getDisplayedActions());
+
+        mTabHolder.simulateNavigation("https://example.com/foo.html", true);
+        assertEquals(buildActions(), getDisplayedActions());
+    }
+
+    @Override
+    int getNotificationId() {
+        return R.id.media_playback_notification;
+    }
+
+    private Set<Integer> getDisplayedActions() {
+        return getManager().mMediaNotificationInfo.mediaSessionActions;
+    }
+
+    private Set<Integer> buildActions() {
+        HashSet<Integer> actions = new HashSet<>();
+
+        actions.add(MediaSessionAction.PLAY);
+        actions.add(MediaSessionAction.PAUSE);
+        return actions;
+    }
+}