Add a UrlManagerTest

chrome/android/java/src/org/chromium/chrome/browser/physicalweb/UrlManager.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.physicalweb;
 
 import android.app.Notification;
+import android.app.NotificationManager;
 import android.app.PendingIntent;
 import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
 import android.content.res.Resources;
 import android.graphics.Bitmap;
 import android.graphics.BitmapFactory;
 import android.os.SystemClock;
 import android.support.v4.app.NotificationCompat;
-import android.support.v4.app.NotificationManagerCompat;
 
 import org.chromium.base.Log;
+import org.chromium.base.VisibleForTesting;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.notifications.NotificationConstants;
+import org.chromium.chrome.browser.notifications.NotificationManagerProxy;
+import org.chromium.chrome.browser.notifications.NotificationManagerProxyImpl;
 
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.Set;
 
 /**
  * This class stores URLs which are discovered by scanning for Physical Web beacons, and updates a
  * Notification as the set changes.
  *
  * There are two sets of URLs maintained:
  * - Those which are currently nearby, as tracked by calls to addUrl/removeUrl
  * - Those which have ever resolved through the Physical Web Service (e.g. are known to produce
  *     good results).
  *
  * Whenever either list changes, we update the Physical Web Notification, based on the intersection
  * of currently-nearby and known-resolved URLs.
  */
 class UrlManager {
     private static final String TAG = "PhysicalWeb";
     private static final String PREFS_NAME = "org.chromium.chrome.browser.physicalweb.URL_CACHE";
     private static final String PREFS_VERSION_KEY = "version";
     private static final String PREFS_NEARBY_URLS_KEY = "nearby_urls";
     private static final String PREFS_RESOLVED_URLS_KEY = "resolved_urls";
     private static final String DEPRECATED_PREFS_URLS_KEY = "urls";
     private static final int PREFS_VERSION = 2;
     private static UrlManager sInstance = null;
     private final Context mContext;
-    private final NotificationManagerCompat mNotificationManager;
-    private final PwsClient mPwsClient;
+    private NotificationManagerProxy mNotificationManager;
+    private PwsClient mPwsClient;
 
     /**
      * Construct the UrlManager.
      * @param context An instance of android.content.Context
      */
     public UrlManager(Context context) {
         mContext = context;
-        mNotificationManager = NotificationManagerCompat.from(context);
-        mPwsClient = new PwsClient();
+        mNotificationManager = new NotificationManagerProxyImpl(
+                (NotificationManager) context.getSystemService(Context.NOTIFICATION_SERVICE));
+        mPwsClient = new PwsClientImpl();
         initSharedPreferences();
     }
 
     /**
      * Get a singleton instance of this class.
      * @param context An instance of android.content.Context.
      * @return A singleton instance of this class.
      */
     public static UrlManager getInstance(Context context) {
         if (sInstance == null) {
             sInstance = new UrlManager(context);
         }
         return sInstance;
     }
 
     /**
      * Add a URL to the store of URLs.
      * This method additionally updates the Physical Web notification.
      * @param url The URL to add.
      */
+    @VisibleForTesting
     public void addUrl(String url) {
         Log.d(TAG, "URL found: " + url);
         boolean isOnboarding = PhysicalWeb.isOnboarding(mContext);
         Set<String> nearbyUrls = getCachedNearbyUrls();
 
         boolean wasUrlListEmpty = (isOnboarding && nearbyUrls.isEmpty())
                 || (!isOnboarding && getUrls().isEmpty());
 
         nearbyUrls.add(url);
         putCachedNearbyUrls(nearbyUrls);
@@ skipping 243 matching lines @@
                 .setVisibility(NotificationCompat.VISIBILITY_PUBLIC)
                 .setAutoCancel(true)
                 .build();
         mNotificationManager.notify(NotificationConstants.NOTIFICATION_ID_PHYSICAL_WEB,
                                     notification);
     }
 
     private void clearNotification() {
         mNotificationManager.cancel(NotificationConstants.NOTIFICATION_ID_PHYSICAL_WEB);
     }
+
+    @VisibleForTesting
+    void overridePwsClientForTesting(PwsClient pwsClient) {
+        mPwsClient = pwsClient;
+    }
+
+    @VisibleForTesting
+    void overrideNotificationManagerForTesting(
+            NotificationManagerProxy notificationManager) {
+        mNotificationManager = notificationManager;
+    }
 }