Fixup tests for Android N in chrome.browser.notifications

chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java b/chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java
new file mode 100644
index 0000000000000000000000000000000000000000..d631a34025676bc5140e38d8941b659ca9f372cd
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java
@@ -0,0 +1,80 @@
+// Copyright 2016 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.notifications;
+
+import android.annotation.SuppressLint;
+import android.annotation.TargetApi;
+import android.app.Notification;
+import android.content.Context;
+import android.graphics.Bitmap;
+import android.graphics.BitmapFactory;
+import android.graphics.drawable.BitmapDrawable;
+import android.graphics.drawable.Icon;
+import android.os.Build;
+import android.os.Bundle;
+
+/**
+ * Utils for Android notification tests.
+ */
+public class NotificationTestUtil {
+    @SuppressWarnings("deprecation") // for Notification.icon
+    public static Bitmap getSmallIconFromNotification(Context context, Notification notification) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+            return getBitmapFromIcon(context, notification.getSmallIcon());
+        } else {
+            return BitmapFactory.decodeResource(context.getResources(), notification.icon);
+        }
+    }
+
+    @SuppressWarnings("deprecation") // for Notification.largeIcon
+    public static Bitmap getLargeIconFromNotification(Context context, Notification notification) {
+        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+            return getBitmapFromIcon(context, notification.getLargeIcon());
+        } else {
+            return notification.largeIcon;
+        }
+    }
+
+    @TargetApi(Build.VERSION_CODES.M) // for Icon.loadDrawable()
+    public static Bitmap getBitmapFromIcon(Context context, Icon icon) {
+        assert Build.VERSION.SDK_INT >= Build.VERSION_CODES.M;
+        return ((BitmapDrawable) icon.loadDrawable(context)).getBitmap();
+    }
+
+    @SuppressLint("NewApi") // Notification.actions is hidden in Jellybean
+    static Notification.Action[] getActions(Notification notification) {
+        return notification.actions;
+    }
+
+    @SuppressLint("NewApi") // Notification.Action is hidden in Jellybean
+    public static CharSequence getActionTitle(Notification.Action action) {
+        return action.title;
+    }
+
+    @SuppressLint("NewApi") // Notification.extras is hidden in Jellybean
+    static Bundle getExtras(Notification notification) {
+        return notification.extras;
+    }
+
+    @SuppressLint("InlinedApi") // Notification.EXTRA_TITLE is hidden in Jellybean
+    static String getExtraTitle(Notification notification) {
+        return getExtras(notification).getString(Notification.EXTRA_TITLE);
+    }
+
+    @SuppressLint("InlinedApi") // Notification.EXTRA_TEXT is hidden in Jellybean
+    static String getExtraText(Notification notification) {
+        return getExtras(notification).getString(Notification.EXTRA_TEXT);
+    }
+
+    @SuppressLint("InlinedApi") // Notification.EXTRA_SUB_TEXT is hidden in Jellybean
+    static String getExtraSubText(Notification notification) {
+        return getExtras(notification).getString(Notification.EXTRA_SUB_TEXT);
+    }
+
+    @SuppressLint("InlinedApi") // Notification.EXTRA_PICTURE is hidden in Jellybean
+    static Bitmap getExtraPicture(Notification notification) {
+        return (Bitmap) getExtras(notification).get(Notification.EXTRA_PICTURE);
+    }
+}