[Content suggestions] Add a function to the service API to fetch favicons

chrome/test/android/javatests/src/org/chromium/chrome/test/util/browser/suggestions/FakeSuggestionsSource.java

 // Copyright 2016 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.test.util.browser.suggestions;
 
 import android.graphics.Bitmap;
 
 import org.chromium.base.Callback;
+import org.chromium.base.ThreadUtils;
 import org.chromium.chrome.browser.ntp.cards.SuggestionsCategoryInfo;
 import org.chromium.chrome.browser.ntp.snippets.CategoryInt;
 import org.chromium.chrome.browser.ntp.snippets.CategoryStatus;
 import org.chromium.chrome.browser.ntp.snippets.CategoryStatus.CategoryStatusEnum;
 import org.chromium.chrome.browser.ntp.snippets.SnippetArticle;
 import org.chromium.chrome.browser.ntp.snippets.SnippetsBridge;
 import org.chromium.chrome.browser.ntp.snippets.SuggestionsSource;
 
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 
 /**
  * A fake Suggestions source for use in unit and instrumentation tests.
  */
 public class FakeSuggestionsSource implements SuggestionsSource {
     private SuggestionsSource.Observer mObserver;
     private final List<Integer> mCategories = new ArrayList<>();
     private final Map<Integer, List<SnippetArticle>> mSuggestions = new HashMap<>();
     private final Map<Integer, Integer> mCategoryStatus = new LinkedHashMap<>();
     private final Map<Integer, SuggestionsCategoryInfo> mCategoryInfo = new HashMap<>();
+
+    // Maps within-category ids to their fake bitmaps.
     private final Map<String, Bitmap> mThumbnails = new HashMap<>();
+    private final Map<String, Bitmap> mFavicons = new HashMap<>();
 
     private final List<Integer> mDismissedCategories = new ArrayList<>();
     private final Map<Integer, List<SnippetArticle>> mDismissedCategorySuggestions =
             new HashMap<>();
     private final Map<Integer, Integer> mDismissedCategoryStatus = new LinkedHashMap<>();
     private final Map<Integer, SuggestionsCategoryInfo> mDismissedCategoryInfo = new HashMap<>();
 
     /**
      * Sets the status to be returned for a given category.
      */
@@ skipping 18 matching lines @@
     }
 
     /**
      * Sets the metadata to be returned for a given category.
      */
     public void setInfoForCategory(@CategoryInt int category, SuggestionsCategoryInfo info) {
         mCategoryInfo.put(category, info);
     }
 
     /**
-     * Sets the bitmap to be returned when the thumbnail is requested for a snippet with that id.
+     * Sets the bitmap to be returned when the thumbnail is requested for a suggestion with that
+     * (within-category) id.
      */
     public void setThumbnailForId(String id, Bitmap bitmap) {
         mThumbnails.put(id, bitmap);
     }
 
     /**
+     * Sets the bitmap to be returned when the favicon is requested for a suggestion with that
+     * (within-category) id.
+     */
+    public void setFaviconForId(String id, Bitmap bitmap) {
+        mFavicons.put(id, bitmap);
+    }
+
+    /**
      * Removes the given suggestion from the source and notifies any observer that it has been
      * invalidated.
      */
     public void fireSuggestionInvalidated(@CategoryInt int category, String idWithinCategory) {
         for (SnippetArticle suggestion : mSuggestions.get(category)) {
             if (suggestion.mIdWithinCategory.equals(idWithinCategory)) {
                 mSuggestions.get(category).remove(suggestion);
                 break;
             }
         }
@@ skipping 41 matching lines @@
         for (int category : mDismissedCategories) {
             mSuggestions.put(category, mDismissedCategorySuggestions.remove(category));
             mCategoryStatus.put(category, mDismissedCategoryStatus.remove(category));
             mCategoryInfo.put(category, mDismissedCategoryInfo.remove(category));
             mCategories.add(category);
         }
         mDismissedCategories.clear();
     }
 
     @Override
-    public void fetchSuggestionImage(SnippetArticle suggestion, Callback<Bitmap> callback) {
+    public void fetchSuggestionImage(
+            final SnippetArticle suggestion, final Callback<Bitmap> callback) {
         if (mThumbnails.containsKey(suggestion.mIdWithinCategory)) {
-            callback.onResult(mThumbnails.get(suggestion.mIdWithinCategory));
+            ThreadUtils.postOnUiThread(new Runnable() {
+                @Override
+                public void run() {
+                    callback.onResult(mThumbnails.get(suggestion.mIdWithinCategory));
+                }
+            });
         }
     }
 
+    @Override
+    public void fetchSuggestionFavicon(final SnippetArticle suggestion, int minimumSizePx,
+            int desiredSizePx, final Callback<Bitmap> callback) {
+        if (mFavicons.containsKey(suggestion.mIdWithinCategory)) {
+            ThreadUtils.postOnUiThread(new Runnable() {
+                @Override
+                public void run() {
+                    callback.onResult(mFavicons.get(suggestion.mIdWithinCategory));
+                }
+            });
+        }
+    }
+
     @Override
     public void fetchSuggestions(@CategoryInt int category, String[] displayedSuggestionIds) {
         throw new UnsupportedOperationException();
     }
 
     @Override
     public void setObserver(Observer observer) {
         mObserver = observer;
     }
 
@@ skipping 18 matching lines @@
 
     @Override
     public List<SnippetArticle> getSuggestionsForCategory(int category) {
         if (!SnippetsBridge.isCategoryStatusAvailable(mCategoryStatus.get(category))) {
             return Collections.emptyList();
         }
         List<SnippetArticle> result = mSuggestions.get(category);
         return result == null ? Collections.<SnippetArticle>emptyList() : new ArrayList<>(result);
     }
 }