[Android O] Refactor channel initialization

chrome/android/junit/src/org/chromium/chrome/browser/notifications/ChannelsInitializerTest.java

Index: chrome/android/junit/src/org/chromium/chrome/browser/notifications/ChannelsInitializerTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/notifications/ChannelsInitializerTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/notifications/ChannelsInitializerTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..1ecb37cf30c7dc2826cae037ac36411f87973d4a
--- /dev/null
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/notifications/ChannelsInitializerTest.java
@@ -0,0 +1,80 @@
+// 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.notifications;
+
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertThat;
+
+import android.app.NotificationManager;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.BlockJUnit4ClassRunner;
+
+import org.chromium.chrome.test.util.browser.notifications.MockNotificationManagerProxy;
+
+/**
+ * Unit tests for ChannelsInitializer.
+ */
+@RunWith(BlockJUnit4ClassRunner.class)
+public class ChannelsInitializerTest {
+    private ChannelsInitializer mChannelsInitializer;
+    private MockNotificationManagerProxy mMockNotificationManager;
+
+    @Before
+    public void setUp() throws Exception {
+        mMockNotificationManager = new MockNotificationManagerProxy();
+        mChannelsInitializer = new ChannelsInitializer(mMockNotificationManager);
+    }
+
+    @Test
+    public void testEnsureInitialized_browserChannel() throws Exception {
+        mChannelsInitializer.ensureInitialized(ChannelsInitializer.CHANNEL_ID_BROWSER);
+
+        assertThat(mMockNotificationManager.getChannels().size(), is(1));
+        ChannelsInitializer.Channel channel = mMockNotificationManager.getChannels().get(0);
+        assertThat(channel.mId, is(ChannelsInitializer.CHANNEL_ID_BROWSER));
+        assertThat(
+                channel.mNameResId, is(org.chromium.chrome.R.string.notification_category_browser));
+        assertThat(channel.mImportance, is(NotificationManager.IMPORTANCE_LOW));
+        assertThat(channel.mGroupId, is(ChannelsInitializer.CHANNEL_GROUP_ID_GENERAL));
+
+        assertThat(mMockNotificationManager.getNotificationChannelGroups().size(), is(1));
+        ChannelsInitializer.ChannelGroup group =
+                mMockNotificationManager.getNotificationChannelGroups().get(0);
+        assertThat(group.mId, is(ChannelsInitializer.CHANNEL_GROUP_ID_GENERAL));
+        assertThat(group.mNameResId,
+                is(org.chromium.chrome.R.string.notification_category_group_general));
+    }
+
+    @Test
+    public void testEnsureInitialized_sitesChannel() throws Exception {
+        mChannelsInitializer.ensureInitialized(ChannelsInitializer.CHANNEL_ID_SITES);
+
+        assertThat(mMockNotificationManager.getChannels().size(), is(1));
+
+        ChannelsInitializer.Channel channel = mMockNotificationManager.getChannels().get(0);
+        assertThat(channel.mId, is(ChannelsInitializer.CHANNEL_ID_SITES));
+        assertThat(
+                channel.mNameResId, is(org.chromium.chrome.R.string.notification_category_sites));
+        assertThat(channel.mImportance, is(NotificationManager.IMPORTANCE_DEFAULT));
+        assertThat(channel.mGroupId, is(ChannelsInitializer.CHANNEL_GROUP_ID_GENERAL));
+
+        assertThat(mMockNotificationManager.getNotificationChannelGroups().size(), is(1));
+        ChannelsInitializer.ChannelGroup group =
+                mMockNotificationManager.getNotificationChannelGroups().get(0);
+        assertThat(group.mId, is(ChannelsInitializer.CHANNEL_GROUP_ID_GENERAL));
+        assertThat(group.mNameResId,
+                is(org.chromium.chrome.R.string.notification_category_group_general));
+    }
+
+    @Test
+    public void testEnsureInitialized_multipleCalls() throws Exception {
+        mChannelsInitializer.ensureInitialized(ChannelsInitializer.CHANNEL_ID_SITES);
+        mChannelsInitializer.ensureInitialized(ChannelsInitializer.CHANNEL_ID_BROWSER);
+        assertThat(mMockNotificationManager.getChannels().size(), is(2));
+    }
+}