Enable deep-linking from notifications for recently used web apps on the Android home screen.

chrome/android/junit/src/org/chromium/chrome/browser/webapps/WebappRegistryTest.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.webapps;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 
 import android.content.Context;
@@ skipping 31 matching lines @@
 
     private static final int INITIAL_TIME = 0;
 
     private SharedPreferences mSharedPreferences;
     private boolean mCallbackCalled;
 
     private class FetchCallback implements WebappRegistry.FetchCallback {
         Set<String> mExpected;
 
         FetchCallback(Set<String> expected) {
+            mCallbackCalled = false;
             mExpected = expected;
         }
 
         @Override
         public void onWebappIdsRetrieved(Set<String> actual) {
             mCallbackCalled = true;
             assertEquals(mExpected, actual);
         }
     }
 
     private class FetchStorageCallback implements WebappRegistry.FetchWebappDataStorageCallback {
         Intent mShortcutIntent;
+        boolean mMarkLaunched;
 
-        FetchStorageCallback(Intent shortcutIntent) {
+        FetchStorageCallback(Intent shortcutIntent, boolean markLaunched) {
             mCallbackCalled = false;
             mShortcutIntent = shortcutIntent;
+            mMarkLaunched = markLaunched;
         }
 
         @Override
         public void onWebappDataStorageRetrieved(WebappDataStorage storage) {
             mCallbackCalled = true;
             storage.updateFromShortcutIntent(mShortcutIntent);
+            storage.updateLastUsedTime();
+            if (mMarkLaunched) storage.setLaunched();
         }
     }
 
+    private class FetchStorageByUrlCallback
+            implements WebappRegistry.FetchWebappDataStorageCallback {
+        String mUrl;
+        String mScope;
+
+        FetchStorageByUrlCallback(String url, String scope) {
+            mCallbackCalled = false;
+            mUrl = url;
+            mScope = scope;
+        }
+
+        @Override
+        public void onWebappDataStorageRetrieved(WebappDataStorage storage) {
+            mCallbackCalled = true;

[gone, 2016/04/07 19:10:44]

This really should be a member variable of this callback class, at this point. 
It'll be too easy to accidentally have one class accidentally affect another
test otherwise.

[dominickn, 2016/04/08 01:10:46]

Done.

+            assertEquals(mUrl, storage.getUrl());
+            assertEquals(mScope, storage.getScope());
+        }
+    }
+
     private class CallbackRunner implements Runnable {
         public CallbackRunner() {
             mCallbackCalled = false;
         }
 
         @Override
         public void run() {
             mCallbackCalled = true;
         }
     }
@@ skipping 22 matching lines @@
         BackgroundShadowAsyncTask.runBackgroundTasks();
 
         Set<String> actual = mSharedPreferences.getStringSet(
                 WebappRegistry.KEY_WEBAPP_SET, Collections.<String>emptySet());
         assertEquals(1, actual.size());
         assertTrue(actual.contains("test"));
     }
 
     @Test
     @Feature({"Webapp"})
-    public void testWebappRegistrationUpdatesLastUsed() throws Exception {
-        long before = System.currentTimeMillis();

[dominickn, 2016/04/06 13:07:01]

The before check started being flaky for me, so I just made it an after check.

-
+    public void testWebappRegistrationUpdatesLastUsedAndDoesNotMarkLaunched() throws Exception {
         WebappRegistry.registerWebapp(Robolectric.application, "test", null);
         BackgroundShadowAsyncTask.runBackgroundTasks();
         long after = System.currentTimeMillis();
 
         SharedPreferences webAppPrefs = Robolectric.application.getSharedPreferences(
                 WebappDataStorage.SHARED_PREFS_FILE_PREFIX + "test", Context.MODE_PRIVATE);
         long actual = webAppPrefs.getLong(WebappDataStorage.KEY_LAST_USED,
                 WebappDataStorage.LAST_USED_INVALID);
-        assertTrue("Timestamp is out of range", before <= actual && actual <= after);
+        assertTrue("Timestamp is out of range", actual <= after);
+
+        boolean launched = webAppPrefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, false);
+        assertTrue(!launched);

[gone, 2016/04/07 19:10:45]

can't just assertTrue(!webAppPref....) here?

[dominickn, 2016/04/08 01:10:46]

Done.

     }
 
     @Test
     @Feature({"Webapp"})
     public void testWebappIdsRetrieval() throws Exception {
         final Set<String> expected = addWebappsToRegistry("first", "second");
 
         WebappRegistry.getRegisteredWebappIds(Robolectric.application, new FetchCallback(expected));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
@@ skipping 175 matching lines @@
 
     @Test
     @Feature({"Webapp"})
     public void testClearWebappHistory() throws Exception {
         final String webapp1Url = "https://www.google.com";
         final String webapp2Url = "https://drive.google.com";
         Intent shortcutIntent1 = createShortcutIntent(webapp1Url);
         Intent shortcutIntent2 = createShortcutIntent(webapp2Url);
 
         WebappRegistry.registerWebapp(Robolectric.application, "webapp1",
-                new FetchStorageCallback(shortcutIntent1));
+                new FetchStorageCallback(shortcutIntent1, true));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         WebappRegistry.registerWebapp(Robolectric.application, "webapp2",
-                new FetchStorageCallback(shortcutIntent2));
+                new FetchStorageCallback(shortcutIntent2, false));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         SharedPreferences webapp1Prefs = Robolectric.application.getSharedPreferences(
                 WebappDataStorage.SHARED_PREFS_FILE_PREFIX + "webapp1", Context.MODE_PRIVATE);
         SharedPreferences webapp2Prefs = Robolectric.application.getSharedPreferences(
                 WebappDataStorage.SHARED_PREFS_FILE_PREFIX + "webapp2", Context.MODE_PRIVATE);
 
         WebappRegistry.clearWebappHistory(Robolectric.application, new CallbackRunner());
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         Set<String> actual = mSharedPreferences.getStringSet(
                 WebappRegistry.KEY_WEBAPP_SET, Collections.<String>emptySet());
         assertEquals(2, actual.size());
         assertTrue(actual.contains("webapp1"));
         assertTrue(actual.contains("webapp2"));
 
         // Verify that the last used time for both web apps is WebappDataStorage.LAST_USED_UNSET.
         long actualLastUsed = webapp1Prefs.getLong(
                 WebappDataStorage.KEY_LAST_USED, WebappDataStorage.LAST_USED_UNSET);
         assertEquals(WebappDataStorage.LAST_USED_UNSET, actualLastUsed);
         actualLastUsed = webapp2Prefs.getLong(
                 WebappDataStorage.KEY_LAST_USED, WebappDataStorage.LAST_USED_UNSET);
         assertEquals(WebappDataStorage.LAST_USED_UNSET, actualLastUsed);
 
+        // Verify that neither web app is marked as launched.
+        boolean actualLaunched = webapp1Prefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, false);
+        assertTrue(!actualLaunched);
+        actualLaunched = webapp2Prefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, false);
+        assertTrue(!actualLaunched);
+
         // Verify that the URL and scope for both web apps is WebappDataStorage.URL_INVALID.
         String actualScope = webapp1Prefs.getString(
                 WebappDataStorage.KEY_SCOPE, WebappDataStorage.URL_INVALID);
         assertEquals(WebappDataStorage.URL_INVALID, actualScope);
         String actualUrl = webapp1Prefs.getString(
                 WebappDataStorage.KEY_URL, WebappDataStorage.URL_INVALID);
         assertEquals(WebappDataStorage.URL_INVALID, actualUrl);
         actualScope = webapp2Prefs.getString(
                 WebappDataStorage.KEY_SCOPE, WebappDataStorage.URL_INVALID);
         assertEquals(WebappDataStorage.URL_INVALID, actualScope);
         actualUrl = webapp2Prefs.getString(
                 WebappDataStorage.KEY_URL, WebappDataStorage.URL_INVALID);
         assertEquals(WebappDataStorage.URL_INVALID, actualUrl);
     }
 
     @Test
     @Feature({"Webapp"})
     public void testGetAfterClearWebappHistory() throws Exception {
         WebappRegistry.registerWebapp(Robolectric.application, "webapp", null);
         BackgroundShadowAsyncTask.runBackgroundTasks();
 
         SharedPreferences webappPrefs = Robolectric.application.getSharedPreferences(
                 WebappDataStorage.SHARED_PREFS_FILE_PREFIX + "webapp", Context.MODE_PRIVATE);
         WebappRegistry.clearWebappHistory(Robolectric.application, new CallbackRunner());
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
-        // Open the webapp up to set the last used time.
+        // Open the webapp up to set the last used time and launched.
         WebappRegistry.getWebappDataStorage(Robolectric.application, "webapp",
-                new FetchStorageCallback(null));
+                new FetchStorageCallback(null, true));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         // Verify that the last used time is valid.
         long actualLastUsed = webappPrefs.getLong(
                 WebappDataStorage.KEY_LAST_USED, WebappDataStorage.LAST_USED_INVALID);
         assertTrue(WebappDataStorage.LAST_USED_INVALID != actualLastUsed);
         assertTrue(WebappDataStorage.LAST_USED_UNSET != actualLastUsed);
+
+        // Verify that the app is marked as launched.
+        boolean actualLaunched = webappPrefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, false);
+        assertTrue(actualLaunched);
     }
 
     @Test
     @Feature({"Webapp"})
     public void testUpdateAfterClearWebappHistory() throws Exception {
-        final String webappUrl =  "http://www.google.com";
+        final String webappUrl = "http://www.google.com";
+        final String webappScope = "http://www.google.com/";
         final Intent shortcutIntent = createShortcutIntent(webappUrl);
         WebappRegistry.registerWebapp(Robolectric.application, "webapp",
-                new FetchStorageCallback(shortcutIntent));
+                new FetchStorageCallback(shortcutIntent, false));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         SharedPreferences webappPrefs = Robolectric.application.getSharedPreferences(
                 WebappDataStorage.SHARED_PREFS_FILE_PREFIX + "webapp", Context.MODE_PRIVATE);
 
+        // Verify that the app is not marked as launched.
+        boolean actualLaunched = webappPrefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, true);
+        assertTrue(actualLaunched);
+
         // Verify that the URL and scope match the original in the intent.
-        String actualURL = webappPrefs.getString(
+        String actualUrl = webappPrefs.getString(
                 WebappDataStorage.KEY_URL, WebappDataStorage.URL_INVALID);
-        assertEquals(webappUrl, actualURL);
+        assertEquals(webappUrl, actualUrl);
         String actualScope = webappPrefs.getString(
                 WebappDataStorage.KEY_SCOPE, WebappDataStorage.URL_INVALID);
-        assertEquals(webappUrl, actualScope);
+        assertEquals(webappScope, actualScope);
 
         WebappRegistry.clearWebappHistory(Robolectric.application, new CallbackRunner());
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
         // Update the webapp from the intent again.
         WebappRegistry.getWebappDataStorage(Robolectric.application, "webapp",
-                new FetchStorageCallback(shortcutIntent));
+                new FetchStorageCallback(shortcutIntent, true));
         BackgroundShadowAsyncTask.runBackgroundTasks();
         Robolectric.runUiThreadTasks();
         assertTrue(mCallbackCalled);
 
+        // Verify that the app is marked as launched.
+        actualLaunched = webappPrefs.getBoolean(WebappDataStorage.KEY_LAUNCHED, true);
+        assertTrue(actualLaunched);
+
         // Verify that the URL and scope match the original in the intent.
-        actualURL = webappPrefs.getString(
+        actualUrl = webappPrefs.getString(
                 WebappDataStorage.KEY_URL, WebappDataStorage.URL_INVALID);
-        assertEquals(webappUrl, actualURL);
+        assertEquals(webappUrl, actualUrl);
         actualScope = webappPrefs.getString(
                 WebappDataStorage.KEY_SCOPE, WebappDataStorage.URL_INVALID);
-        assertEquals(webappUrl, actualScope);
+        assertEquals(webappScope, actualScope);
+    }
+
+    @Test
+    @Feature({"Webapp"})
+    public void testGetWebappDataStorageForUrl() throws Exception {

[gone, 2016/04/07 19:10:45]

Can you add comments here?  I'm not sure what I'm looking at.

[dominickn, 2016/04/08 01:10:46]

Done.

+        final String webapp1Url = "https://www.google.com/";
+        final String webapp2Url = "https://drive.google.com/";
+        final String webapp3Url = "https://www.google.com/drive/index.html";
+        final String webapp4Url = "https://www.google.com/drive/docs/index.html";
+
+        final String webapp3Scope = "https://www.google.com/drive/";
+        final String webapp4Scope = "https://www.google.com/drive/docs/";
+
+        final String test1Url = "https://www.google.com/index.html";
+        final String test2Url = "https://www.google.com/drive/recent.html";
+        final String test3Url = "https://www.google.com/drive/docs/recent.html";
+        final String test4Url = "https://www.google.com/drive/docs/recent/index.html";
+        final String test5Url = "https://maps.google.com/";
+
+        Intent shortcutIntent1 = createShortcutIntent(webapp1Url);
+        Intent shortcutIntent2 = createShortcutIntent(webapp2Url);
+        Intent shortcutIntent3 = createShortcutIntent(webapp3Url);
+        Intent shortcutIntent4 = createShortcutIntent(webapp4Url);
+
+        WebappRegistry.registerWebapp(Robolectric.application, "webapp1",
+                new FetchStorageCallback(shortcutIntent1, true));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.registerWebapp(Robolectric.application, "webapp2",
+                new FetchStorageCallback(shortcutIntent2, false));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.registerWebapp(Robolectric.application, "webapp3",
+                new FetchStorageCallback(shortcutIntent3, true));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.registerWebapp(Robolectric.application, "webapp4",
+                new FetchStorageCallback(shortcutIntent4, false));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.getWebappDataStorageForUrl(Robolectric.application, test1Url,
+                new FetchStorageByUrlCallback(webapp1Url, webapp1Url));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.getWebappDataStorageForUrl(Robolectric.application, test2Url,
+                new FetchStorageByUrlCallback(webapp3Url, webapp3Scope));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.getWebappDataStorageForUrl(Robolectric.application, test3Url,
+                new FetchStorageByUrlCallback(webapp4Url, webapp4Scope));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        WebappRegistry.getWebappDataStorageForUrl(Robolectric.application, test4Url,
+                new FetchStorageByUrlCallback(webapp4Url, webapp4Scope));
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
+
+        mCallbackCalled = false;
+        WebappRegistry.getWebappDataStorageForUrl(Robolectric.application, test5Url,
+                new WebappRegistry.FetchWebappDataStorageCallback() {
+                    @Override
+                    public void onWebappDataStorageRetrieved(WebappDataStorage storage) {
+                        assertEquals(null, storage);
+                        mCallbackCalled = true;
+                    }
+                }
+        );
+        BackgroundShadowAsyncTask.runBackgroundTasks();
+        Robolectric.runUiThreadTasks();
+        assertTrue(mCallbackCalled);
     }
 
     private Set<String> addWebappsToRegistry(String... webapps) {
         final Set<String> expected = new HashSet<String>(Arrays.asList(webapps));
         mSharedPreferences.edit()
                 .putStringSet(WebappRegistry.KEY_WEBAPP_SET, expected)
                 .commit();
         return expected;
     }
 
     private Set<String> getRegisteredWebapps() {
         return mSharedPreferences.getStringSet(
                 WebappRegistry.KEY_WEBAPP_SET, Collections.<String>emptySet());
     }
 
     private Intent createShortcutIntent(String url) {
-        return ShortcutHelper.createWebappShortcutIntent("id", "action", url, url, "name",
-                "shortName", null, ShortcutHelper.WEBAPP_SHORTCUT_VERSION, 0, 0, 0, false);
+        return ShortcutHelper.createWebappShortcutIntent("id", "action", url,
+                ShortcutHelper.getScopeFromUrl(url), "name", "shortName", null,
+                ShortcutHelper.WEBAPP_SHORTCUT_VERSION, 0, 0, 0, false);
     }
 }