[Media, UI] Change MediaNotification style to MediaStyle

chrome/android/javatests/src/org/chromium/chrome/browser/media/ui/NotificationTitleUpdatedTest.java

 // Copyright 2015 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.chromium.base.test.util.Restriction.RESTRICTION_TYPE_NON_LOW_END_DEVICE;
 
 import android.app.Notification;
 import android.test.suitebuilder.annotation.SmallTest;
 import android.view.View;
 import android.widget.TextView;
 
 import org.chromium.base.ObserverList;
 import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.CommandLineFlags;
 import org.chromium.base.test.util.Restriction;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.ChromeActivity;
+import org.chromium.chrome.browser.ChromeFeatureList;
 import org.chromium.chrome.browser.tab.Tab;
 import org.chromium.chrome.test.ChromeActivityTestCaseBase;
 import org.chromium.chrome.test.util.ChromeRestriction;
 import org.chromium.chrome.test.util.browser.TabTitleObserver;
 import org.chromium.content.browser.test.util.Criteria;
 import org.chromium.content.browser.test.util.CriteriaHelper;
 import org.chromium.content.browser.test.util.JavaScriptUtils;
 import org.chromium.content_public.browser.WebContentsObserver;
 import org.chromium.content_public.common.MediaMetadata;
 
@@ skipping 10 matching lines @@
         super(ChromeActivity.class);
     }
 
     @Override
     protected void setUp() throws Exception {
         super.setUp();
         mTab = getActivity().getActivityTab();
         simulateUpdateTitle(mTab, "title1");
     }
 
-    @SmallTest
-    public void testSessionStatePlaying() throws InterruptedException {
+    private void testSessionStatePlayingInternal() throws InterruptedException {
         simulateMediaSessionStateChanged(mTab, true, false);
         assertTitleMatches("title1");
         simulateUpdateTitle(mTab, "title2");
         assertTitleMatches("title2");
     }
 
-    @SmallTest
-    public void testSessionStatePaused() throws InterruptedException {
+    private void testSessionStatePausedInternal() throws InterruptedException {
         simulateMediaSessionStateChanged(mTab, true, true);
         assertTitleMatches("title1");
         simulateUpdateTitle(mTab, "title2");
         assertTitleMatches("title2");
     }
 
-    @SmallTest
-    public void testSessionStateUncontrollable() throws InterruptedException {
+    private void testSessionStateUncontrollableInternal() throws InterruptedException {
         simulateMediaSessionStateChanged(mTab, true, false);
         assertTitleMatches("title1");
         simulateMediaSessionStateChanged(mTab, false, false);
         simulateUpdateTitle(mTab, "title2");
     }
 
-    @SmallTest
-    public void testMediaMetadaSetsTitle() throws InterruptedException {
+    private void testMediaMetadataSetsTitleInternal() throws InterruptedException {
         simulateMediaSessionStateChanged(mTab, true, false, new MediaMetadata("title2", "", ""));
         assertTitleMatches("title2");
     }
 
-    @SmallTest
-    public void testMediaMetadaOverridesTitle() throws InterruptedException {
+    private void testMediaMetadataOverridesTitleInternal() throws InterruptedException {
         simulateMediaSessionStateChanged(mTab, true, false, new MediaMetadata("title2", "", ""));
         assertTitleMatches("title2");
 
         simulateUpdateTitle(mTab, "title3");
         assertTitleMatches("title2");
     }
 
     /**
      * Test if a notification accepts the title update from another tab, using the following steps:
      *   1. set the title of mTab, start the media session, a notification should show up;
      *   2. stop the media session of mTab, the notification shall hide;
      *   3. create newTab, set the title of mTab, start the media session of mTab,
      *      a notification should show up;
      *   4. change the title of newTab and then mTab to different names,
      *      the notification should have the title of newTab.
      */
-    @SmallTest
-    @Restriction({ChromeRestriction.RESTRICTION_TYPE_PHONE, RESTRICTION_TYPE_NON_LOW_END_DEVICE})
-    public void testMultipleTabs() throws Throwable {
+    private void testMultipleTabsInternal() throws Throwable {
         simulateMediaSessionStateChanged(mTab, true, false);
         assertTitleMatches("title1");
         simulateMediaSessionStateChanged(mTab, false, false);
 
         Tab newTab = loadUrlInNewTab("about:blank");
         assertNotNull(newTab);
 
         simulateMediaSessionStateChanged(newTab, true, false);
         simulateUpdateTitle(newTab, "title3");
         simulateUpdateTitle(mTab, "title2");
         assertTitleMatches("title3");
     }
 
+    @SmallTest
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testSessionStatePlaying_MediaStyleNotification() throws InterruptedException {
+        testSessionStatePlayingInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testSessionStatePaused_MediaStyleNotification() throws InterruptedException {
+        testSessionStatePausedInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testSessionStateUncontrollable_MediaStyleNotification()
+            throws InterruptedException {
+        testSessionStateUncontrollableInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testMediaMetadataSetsTitle_MediaStyleNotification() throws InterruptedException {
+        testMediaMetadataSetsTitleInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testMediaMetadataOverridesTitle_MediaStyleNotification()
+            throws InterruptedException {
+        testMediaMetadataOverridesTitleInternal();
+    }
+
+    @SmallTest
+    @Restriction({ChromeRestriction.RESTRICTION_TYPE_PHONE, RESTRICTION_TYPE_NON_LOW_END_DEVICE})
+    @CommandLineFlags.Add("enable-features=MediaStyleNotification")
+    public void testMultipleTabs_MediaStyleNotification() throws Throwable {
+        testMultipleTabsInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testSessionStatePlaying_CustomNotification() throws InterruptedException {
+        testSessionStatePlayingInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testSessionStatePaused_CustomNotification() throws InterruptedException {
+        testSessionStatePausedInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testSessionStateUncontrollable_CustomNotification() throws InterruptedException {
+        testSessionStateUncontrollableInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testMediaMetadataSetsTitle_CustomNotification() throws InterruptedException {
+        testMediaMetadataSetsTitleInternal();
+    }
+
+    @SmallTest
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testMediaMetadataOverridesTitle_CustomNotification() throws InterruptedException {
+        testMediaMetadataOverridesTitleInternal();
+    }
+
+    @SmallTest
+    @Restriction({ChromeRestriction.RESTRICTION_TYPE_PHONE, RESTRICTION_TYPE_NON_LOW_END_DEVICE})
+    @CommandLineFlags.Add("disable-features=MediaStyleNotification")
+    public void testMultipleTabs_CustomNotification() throws Throwable {
+        testMultipleTabsInternal();
+    }
+
     @Override
     public void startMainActivity() throws InterruptedException {
         startMainActivityOnBlankPage();
     }
 
     private void simulateMediaSessionStateChanged(
             final Tab tab, final boolean isControllable, final boolean isSuspended) {
         simulateMediaSessionStateChanged(
                 tab, isControllable, isSuspended, new MediaMetadata("", "", ""));
     }
@@ skipping 39 matching lines @@
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
                 @Override
                 public void run() {
                     Notification notification =
                             MediaNotificationManager
                                     .getNotificationBuilderForTesting(NOTIFICATION_ID)
                                     .build();
 
                     View contentView = notification.contentView.apply(
                             getActivity().getApplicationContext(), null);
-                    TextView titleView = (TextView) contentView.findViewById(R.id.title);
-                    assertEquals(title, titleView.getText());
+                    String observedText = null;
+                    if (ChromeFeatureList.isEnabled(ChromeFeatureList.MEDIA_STYLE_NOTIFICATION)) {
+                        TextView view = (TextView) contentView.findViewById(android.R.id.title);
+                        if (view == null) {
+                            // Case where NotificationCompat does not use the native Notification.
+                            // The TextView id will be in Chrome's namespace.
+                            view = (TextView) contentView.findViewById(R.id.title);
+                        }
+                        observedText = view.getText().toString();
+                    } else {
+                        observedText = ((TextView) contentView.findViewById(R.id.title))
+                                               .getText()
+                                               .toString();
+                    }
+                    assertEquals(title, observedText);
                 }
             });
     }
 }