Add a UrlManagerTest

chrome/android/javatests/src/org/chromium/chrome/browser/physicalweb/UrlManagerTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/physicalweb/UrlManagerTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/physicalweb/UrlManagerTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/physicalweb/UrlManagerTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..f3b5a2c5c8fb9b7ada49cbd0587abef041cd3437
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/physicalweb/UrlManagerTest.java
@@ -0,0 +1,115 @@
+// 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.content.Context;
+import android.content.SharedPreferences;
+import android.preference.PreferenceManager;
+import android.test.InstrumentationTestCase;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import org.chromium.chrome.test.util.browser.notifications.MockNotificationManagerProxy;
+import org.chromium.chrome.test.util.browser.notifications.MockNotificationManagerProxy.NotificationEntry;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+import java.util.Set;
+
+/**
+ * Tests for the UrlManager class.
+ */
+public class UrlManagerTest extends InstrumentationTestCase {
+    static final String URL1 = "https://example.com/";
+    static final String TITLE1 = "Example";
+    static final String DESC1 = "Example Website";
+    static final String PREF_PHYSICAL_WEB = "physical_web";
+    static final int PHYSICAL_WEB_OFF = 0;
+    static final int PHYSICAL_WEB_ON = 1;
+    static final int PHYSICAL_WEB_ONBOARDING = 2;
+    UrlManager mUrlManager = null;
+    MockPwsClient mMockPwsClient = null;
+    MockNotificationManagerProxy mMockNotificationManagerProxy = null;
+    SharedPreferences mSharedPreferences = null;
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        Context context = getInstrumentation().getTargetContext().getApplicationContext();
+        mSharedPreferences = PreferenceManager.getDefaultSharedPreferences(context);
+        mSharedPreferences.edit().putInt(PREF_PHYSICAL_WEB, PHYSICAL_WEB_ON).apply();
+        mUrlManager = new UrlManager(context);
+        mMockPwsClient = new MockPwsClient();
+        mUrlManager.overridePwsClientForTesting(mMockPwsClient);
+        mMockNotificationManagerProxy = new MockNotificationManagerProxy();
+        mUrlManager.overrideNotificationManagerForTesting(mMockNotificationManagerProxy);
+    }
+
+    private void setOnboarding() {
+        mSharedPreferences.edit().putInt(PREF_PHYSICAL_WEB, PHYSICAL_WEB_ONBOARDING).apply();
+    }
+
+    @SmallTest
+    public void testAddUrlWhileOnboardingMakesNotification() throws Exception {
+        setOnboarding();
+        ArrayList<PwsResult> results = new ArrayList<>();
+        results.add(new PwsResult(URL1, URL1, null, TITLE1, DESC1));
+        mMockPwsClient.addPwsResults(results);
+        mUrlManager.addUrl(URL1);
+
+        // Make sure that a resolution was *not* attempted.
+        List<Collection<String>> resolveCalls = mMockPwsClient.getResolveCalls();
+        assertEquals(0, resolveCalls.size());
+
+        // Make sure that we have no resolved URLs.
+        Set<String> urls = mUrlManager.getUrls();
+        assertEquals(0, urls.size());
+
+        // Make sure that a notification was shown.
+        List<NotificationEntry> notifications = mMockNotificationManagerProxy.getNotifications();
+        assertEquals(1, notifications.size());
+    }
+
+    @SmallTest
+    public void testAddUrlNoResolutionDoesNothing() throws Exception {
+        mMockPwsClient.addPwsResults(new ArrayList<PwsResult>());
+        mUrlManager.addUrl(URL1);
+
+        // Make sure that a resolution was attempted.
+        List<Collection<String>> resolveCalls = mMockPwsClient.getResolveCalls();
+        assertEquals(1, resolveCalls.size());
+
+        // Make sure that we have no resolved URLs.
+        Set<String> urls = mUrlManager.getUrls();
+        assertEquals(0, urls.size());
+        // Make sure that we do have unresolved URLs.
+        urls = mUrlManager.getUrls(true);
+        assertEquals(1, urls.size());
+
+        // Make sure that a notification was not shown.
+        List<NotificationEntry> notifications = mMockNotificationManagerProxy.getNotifications();
+        assertEquals(0, notifications.size());
+    }
+
+    @SmallTest
+    public void testAddUrlWithResolutionMakesNotification() throws Exception {
+        ArrayList<PwsResult> results = new ArrayList<>();
+        results.add(new PwsResult(URL1, URL1, null, TITLE1, DESC1));
+        mMockPwsClient.addPwsResults(results);
+        mUrlManager.addUrl(URL1);
+
+        // Make sure that a resolution was attempted.
+        List<Collection<String>> resolveCalls = mMockPwsClient.getResolveCalls();
+        assertEquals(1, resolveCalls.size());
+
+        // Make sure that we have our resolved URLs.
+        Set<String> urls = mUrlManager.getUrls();
+        assertEquals(1, urls.size());
+
+        // Make sure that a notification was shown.
+        List<NotificationEntry> notifications = mMockNotificationManagerProxy.getNotifications();
+        assertEquals(1, notifications.size());
+    }
+}