Index: chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/notifications/MockNotificationManagerProxy.java |
diff --git a/chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/notifications/MockNotificationManagerProxy.java b/chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/notifications/MockNotificationManagerProxy.java |
index 44a86857cdefb68867b196b0ee9d85c084554e3c..0e50e6902e035d066e6e29a1220cca3f117e08d2 100644 |
--- a/chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/notifications/MockNotificationManagerProxy.java |
+++ b/chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/notifications/MockNotificationManagerProxy.java |
@@ -10,6 +10,7 @@ import org.chromium.chrome.browser.notifications.ChannelsInitializer; |
import org.chromium.chrome.browser.notifications.NotificationManagerProxy; |
import java.util.ArrayList; |
+import java.util.Iterator; |
import java.util.LinkedHashMap; |
import java.util.List; |
import java.util.Map; |
@@ -114,6 +115,23 @@ public class MockNotificationManagerProxy implements NotificationManagerProxy { |
} |
@Override |
+ public List<String> getNotificationChannelIds() { |
+ List<String> channelIds = new ArrayList<>(); |
+ for (ChannelsInitializer.Channel channel : mChannels) { |
+ channelIds.add(channel.mId); |
+ } |
+ return channelIds; |
+ } |
+ |
+ @Override |
+ public void deleteNotificationChannel(@ChannelsInitializer.ChannelId String id) { |
+ for (Iterator<ChannelsInitializer.Channel> it = mChannels.iterator(); it.hasNext();) { |
+ ChannelsInitializer.Channel channel = it.next(); |
+ if (id.equals(channel.mId)) it.remove(); |
+ } |
+ } |
+ |
+ @Override |
public void notify(int id, Notification notification) { |
notify(null /* tag */, id, notification); |
} |