[Android NTP] Move suggestion sections into a separate node.

chrome/android/java/src/org/chromium/chrome/browser/ntp/cards/SectionList.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.browser.ntp.cards;
 
-import android.annotation.SuppressLint;
-import android.graphics.Canvas;
-import android.support.annotation.StringRes;
-import android.support.v7.widget.RecyclerView;
-import android.support.v7.widget.RecyclerView.Adapter;
-import android.support.v7.widget.RecyclerView.ViewHolder;
-import android.support.v7.widget.helper.ItemTouchHelper;
-import android.view.View;
-import android.view.ViewGroup;
-
 import org.chromium.base.Log;
-import org.chromium.base.VisibleForTesting;
-import org.chromium.chrome.R;
-import org.chromium.chrome.browser.ntp.NewTabPage.DestructionObserver;
 import org.chromium.chrome.browser.ntp.NewTabPageView.NewTabPageManager;
-import org.chromium.chrome.browser.ntp.UiConfig;
 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.SectionHeaderViewHolder;
 import org.chromium.chrome.browser.ntp.snippets.SnippetArticle;
-import org.chromium.chrome.browser.ntp.snippets.SnippetArticleViewHolder;
 import org.chromium.chrome.browser.ntp.snippets.SnippetsBridge;
 import org.chromium.chrome.browser.ntp.snippets.SnippetsConfig;
 import org.chromium.chrome.browser.ntp.snippets.SuggestionsSource;
 import org.chromium.chrome.browser.offlinepages.OfflinePageBridge;
 
 import java.util.ArrayList;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 
 /**
- * A class that handles merging above the fold elements and below the fold cards into an adapter
- * that will be used to back the NTP RecyclerView. The first element in the adapter should always be
- * the above-the-fold view (containing the logo, search box, and most visited tiles) and subsequent
- * elements will be the cards shown to the user
+ * A node in the tree containing a list of all suggestions sections. It listens to changes in the
+ * suggestions source and updates the corresponding sections.
  */
-public class NewTabPageAdapter
-        extends Adapter<NewTabPageViewHolder> implements SuggestionsSource.Observer, NodeParent {
+public class SectionList extends InnerNode implements SuggestionsSource.Observer {
     private static final String TAG = "Ntp";
 
-    private final NewTabPageManager mNewTabPageManager;
-    private final View mAboveTheFoldView;
-    private final UiConfig mUiConfig;
-    private final ItemTouchCallbacks mItemTouchCallbacks = new ItemTouchCallbacks();
-    private final OfflinePageBridge mOfflinePageBridge;
-    private NewTabPageRecyclerView mRecyclerView;
-
-    /**
-     * List of all child nodes (which can themselves contain multiple child nodes).
-     */
-    private final List<TreeNode> mChildren = new ArrayList<>();
-    private final AboveTheFoldItem mAboveTheFold = new AboveTheFoldItem();
-    private final SignInPromo mSigninPromo;
-    private final AllDismissedItem mAllDismissed;
-    private final Footer mFooter;
-    private final SpacingItem mBottomSpacer = new SpacingItem();
-    private final InnerNode mRoot;
-
     /** Maps suggestion categories to sections, with stable iteration ordering. */
     private final Map<Integer, SuggestionsSection> mSections = new LinkedHashMap<>();
+    private final List<TreeNode> mChildren = new ArrayList<>();
+    private final NewTabPageManager mNewTabPageManager;
+    private final OfflinePageBridge mOfflinePageBridge;
 
-    private class ItemTouchCallbacks extends ItemTouchHelper.Callback {
-        @Override
-        public void onSwiped(ViewHolder viewHolder, int direction) {
-            mRecyclerView.onItemDismissStarted(viewHolder);
-            NewTabPageAdapter.this.dismissItem(viewHolder.getAdapterPosition());
-        }
+    public SectionList(NodeParent parent, NewTabPageManager newTabPageManager,
+            OfflinePageBridge offlinePageBridge) {
+        super(parent);
+        mNewTabPageManager = newTabPageManager;
+        mNewTabPageManager.getSuggestionsSource().setObserver(this);

[dgn, 2016/12/13 14:54:47]

setObserver will hook into the native code and might end up with
onNewSuggestions being called. I think it should move at the end of init() to
make sure everything is properly setup before we start receiving events.

+        mOfflinePageBridge = offlinePageBridge;
+    }
 
-        @Override
-        public void clearView(RecyclerView recyclerView, ViewHolder viewHolder) {
-            // clearView() is called when an interaction with the item is finished, which does
-            // not mean that the user went all the way and dismissed the item before releasing it.
-            // We need to check that the item has been removed.
-            if (viewHolder.getAdapterPosition() == RecyclerView.NO_POSITION) {
-                mRecyclerView.onItemDismissFinished(viewHolder);
-            }
+    @Override
+    public void init() {
+        super.init();
+        resetSections(/* alwaysAllowEmptySections = */ false);
+    }
 
-            super.clearView(recyclerView, viewHolder);
-        }
-
-        @Override
-        public boolean onMove(RecyclerView recyclerView, ViewHolder viewHolder, ViewHolder target) {
-            assert false; // Drag and drop not supported, the method will never be called.
-            return false;
-        }
-
-        @Override
-        public int getMovementFlags(RecyclerView recyclerView, ViewHolder viewHolder) {
-            assert viewHolder instanceof NewTabPageViewHolder;
-
-            int swipeFlags = 0;
-            if (((NewTabPageViewHolder) viewHolder).isDismissable()) {
-                swipeFlags = ItemTouchHelper.START | ItemTouchHelper.END;
-            }
-
-            return makeMovementFlags(0 /* dragFlags */, swipeFlags);
-        }
-
-        @Override
-        public void onChildDraw(Canvas c, RecyclerView recyclerView, ViewHolder viewHolder,
-                float dX, float dY, int actionState, boolean isCurrentlyActive) {
-            assert viewHolder instanceof NewTabPageViewHolder;
-
-            // The item has already been removed. We have nothing more to do.
-            // In some cases a removed children may call this method when unrelated items are
-            // interacted with, but this check also covers the case.
-            // See https://crbug.com/664466, b/32900699
-            if (viewHolder.getAdapterPosition() == RecyclerView.NO_POSITION) return;
-
-            // We use our own implementation of the dismissal animation, so we don't call the
-            // parent implementation. (by default it changes the translation-X and elevation)
-            mRecyclerView.updateViewStateForDismiss(dX, (NewTabPageViewHolder) viewHolder);
-
-            // If there is another item that should be animated at the same time, do the same to it.
-            NewTabPageViewHolder siblingViewHolder = getDismissSibling(viewHolder);
-            if (siblingViewHolder != null) {
-                mRecyclerView.updateViewStateForDismiss(dX, siblingViewHolder);
-            }
-        }
+    @Override
+    protected List<TreeNode> getChildren() {
+        return mChildren;
     }
 
     /**
-     * Creates the adapter that will manage all the cards to display on the NTP.
-     *
-     * @param manager the NewTabPageManager to use to interact with the rest of the system.
-     * @param aboveTheFoldView the layout encapsulating all the above-the-fold elements
-     *                         (logo, search box, most visited tiles)
-     * @param uiConfig the NTP UI configuration, to be passed to created views.
-     * @param offlinePageBridge the OfflinePageBridge used to determine if articles are available
-     *                              offline.
-     *
-     */
-    public NewTabPageAdapter(NewTabPageManager manager, View aboveTheFoldView, UiConfig uiConfig,
-            OfflinePageBridge offlinePageBridge) {
-        mNewTabPageManager = manager;
-        mAboveTheFoldView = aboveTheFoldView;
-        mUiConfig = uiConfig;
-        mOfflinePageBridge = offlinePageBridge;
-        mRoot = new InnerNode(this) {
-            @Override
-            protected List<TreeNode> getChildren() {
-                return mChildren;
-            }
-
-            @Override
-            public void onItemRangeChanged(TreeNode child, int index, int count) {
-                if (mChildren.isEmpty()) return; // The sections have not been initialised yet.
-                super.onItemRangeChanged(child, index, count);
-            }
-
-            @Override
-            public void onItemRangeInserted(TreeNode child, int index, int count) {
-                if (mChildren.isEmpty()) return; // The sections have not been initialised yet.
-                super.onItemRangeInserted(child, index, count);
-            }
-
-            @Override
-            public void onItemRangeRemoved(TreeNode child, int index, int count) {
-                if (mChildren.isEmpty()) return; // The sections have not been initialised yet.
-                super.onItemRangeRemoved(child, index, count);
-            }
-        };
-
-        mSigninPromo = new SignInPromo(mRoot);
-        mAllDismissed = new AllDismissedItem(mRoot);
-        mFooter = new Footer(mRoot);
-        DestructionObserver signInObserver = mSigninPromo.getObserver();
-        if (signInObserver != null) mNewTabPageManager.addDestructionObserver(signInObserver);
-
-        resetSections(/*alwaysAllowEmptySections=*/false);
-        mNewTabPageManager.getSuggestionsSource().setObserver(this);
-    }
-
-    /**
      * Resets the sections, reloading the whole new tab page content.
      * @param alwaysAllowEmptySections Whether sections are always allowed to be displayed when
      *     they are empty, even when they are normally not.
      */
     public void resetSections(boolean alwaysAllowEmptySections) {
         mSections.clear();
         mChildren.clear();
 
         SuggestionsSource suggestionsSource = mNewTabPageManager.getSuggestionsSource();
         int[] categories = suggestionsSource.getCategories();
         int[] suggestionsPerCategory = new int[categories.length];
         int i = 0;
         for (int category : categories) {
             int categoryStatus = suggestionsSource.getCategoryStatus(category);
             if (categoryStatus == CategoryStatus.LOADING_ERROR
                     || categoryStatus == CategoryStatus.NOT_PROVIDED
                     || categoryStatus == CategoryStatus.CATEGORY_EXPLICITLY_DISABLED)
                 continue;
 
             suggestionsPerCategory[i++] =
                     resetSection(category, categoryStatus, alwaysAllowEmptySections);
         }
 
         mNewTabPageManager.trackSnippetsPageImpression(categories, suggestionsPerCategory);
-
-        updateChildren();
     }
 
     /**
      * Resets the section for {@code category}. Removes the section if there are no suggestions for
      * it and it is not allowed to be empty. Otherwise, creates the section if it is not present
      * yet. Sets the available suggestions on the section.
      * @param category The category for which the section must be reset.
      * @param categoryStatus The category status.
      * @param alwaysAllowEmptySections Whether sections are always allowed to be displayed when
      *     they are empty, even when they are normally not.
      * @return The number of suggestions for the section.
      */
     private int resetSection(@CategoryInt int category, @CategoryStatusEnum int categoryStatus,
             boolean alwaysAllowEmptySections) {
         SuggestionsSource suggestionsSource = mNewTabPageManager.getSuggestionsSource();
         List<SnippetArticle> suggestions = suggestionsSource.getSuggestionsForCategory(category);
         SuggestionsCategoryInfo info = suggestionsSource.getCategoryInfo(category);
 
+        SuggestionsSection section = mSections.get(category);
+
         // Do not show an empty section if not allowed.
         if (suggestions.isEmpty() && !info.showIfEmpty() && !alwaysAllowEmptySections) {
-            mSections.remove(category);
+            if (section != null) removeSection(section);
             return 0;
         }
 
         // Create the section if needed.
-        SuggestionsSection section = mSections.get(category);
         if (section == null) {
-            section = new SuggestionsSection(mRoot, info, mNewTabPageManager, mOfflinePageBridge);
+            section = new SuggestionsSection(this, mNewTabPageManager, mOfflinePageBridge, info);
             mSections.put(category, section);
+            mChildren.add(section);
+            didAddChild(section);
         }
 
         // Add the new suggestions.
         setSuggestions(category, suggestions, categoryStatus);
 
         return suggestions.size();
     }
 
-    /** Returns callbacks to configure the interactions with the RecyclerView's items. */
-    public ItemTouchHelper.Callback getItemTouchCallbacks() {
-        return mItemTouchCallbacks;
-    }
-
     @Override
     public void onNewSuggestions(@CategoryInt int category) {
         @CategoryStatusEnum
         int status = mNewTabPageManager.getSuggestionsSource().getCategoryStatus(category);
 
         if (!canLoadSuggestions(category, status)) return;
 
         // We never want to refresh the suggestions if we already have some content.
         if (mSections.get(category).hasSuggestions()) return;
 
@@ skipping 30 matching lines @@
                 // The section provider has gone away. Keep open UIs as they are.
                 return;
 
             case CategoryStatus.CATEGORY_EXPLICITLY_DISABLED:
             case CategoryStatus.LOADING_ERROR:
                 // Need to remove the entire section from the UI immediately.
                 removeSection(mSections.get(category));
                 return;
 
             case CategoryStatus.SIGNED_OUT:
-                // TODO(dgn): We currently can only reach this through an old variation parameter.
+                resetSection(category, status, /* alwaysAllowEmptySections = */ false);
+                return;
+
             default:
                 mSections.get(category).setStatus(status);
                 return;
         }
     }
 
     @Override
     public void onSuggestionInvalidated(@CategoryInt int category, String idWithinCategory) {
         if (!mSections.containsKey(category)) return;
         mSections.get(category).removeSuggestionById(idWithinCategory);
     }
 
     @Override
     public void onFullRefreshRequired() {
-        resetSections(/*alwaysAllowEmptySections=*/false);
-    }
-
-    @Override
-    @ItemViewType
-    public int getItemViewType(int position) {
-        return mRoot.getItemViewType(position);
-    }
-
-    @Override
-    public NewTabPageViewHolder onCreateViewHolder(ViewGroup parent, int viewType) {
-        assert parent == mRecyclerView;
-
-        switch (viewType) {
-            case ItemViewType.ABOVE_THE_FOLD:
-                return new NewTabPageViewHolder(mAboveTheFoldView);
-
-            case ItemViewType.HEADER:
-                return new SectionHeaderViewHolder(mRecyclerView, mUiConfig);
-
-            case ItemViewType.SNIPPET:
-                return new SnippetArticleViewHolder(mRecyclerView, mNewTabPageManager, mUiConfig);
-
-            case ItemViewType.SPACING:
-                return new NewTabPageViewHolder(SpacingItem.createView(parent));
-
-            case ItemViewType.STATUS:
-                return new StatusCardViewHolder(mRecyclerView, mNewTabPageManager, mUiConfig);
-
-            case ItemViewType.PROGRESS:
-                return new ProgressViewHolder(mRecyclerView);
-
-            case ItemViewType.ACTION:
-                return new ActionItem.ViewHolder(mRecyclerView, mNewTabPageManager, mUiConfig);
-
-            case ItemViewType.PROMO:
-                return new SignInPromo.ViewHolder(mRecyclerView, mNewTabPageManager, mUiConfig);
-
-            case ItemViewType.FOOTER:
-                return new Footer.ViewHolder(mRecyclerView, mNewTabPageManager);
-
-            case ItemViewType.ALL_DISMISSED:
-                return new AllDismissedItem.ViewHolder(mRecyclerView, mNewTabPageManager, this);
-        }
-
-        assert false : viewType;
-        return null;
-    }
-
-    @Override
-    public void onBindViewHolder(NewTabPageViewHolder holder, final int position) {
-        mRoot.onBindViewHolder(holder, position);
-    }
-
-    @Override
-    public int getItemCount() {
-        return mRoot.getItemCount();
-    }
-
-    public int getAboveTheFoldPosition() {
-        return getChildPositionOffset(mAboveTheFold);
-    }
-
-    public int getFirstHeaderPosition() {
-        return getFirstPositionForType(ItemViewType.HEADER);
-    }
-
-    public int getFirstCardPosition() {
-        for (int i = 0; i < getItemCount(); ++i) {
-            if (CardViewHolder.isCard(getItemViewType(i))) return i;
-        }
-        return RecyclerView.NO_POSITION;
-    }
-
-    int getBottomSpacerPosition() {
-        return getChildPositionOffset(mBottomSpacer);
-    }
-
-    int getLastContentItemPosition() {
-        return getChildPositionOffset(hasAllBeenDismissed() ? mAllDismissed : mFooter);
+        resetSections(/* alwaysAllowEmptySections = */false);
     }
 
     private void setSuggestions(@CategoryInt int category, List<SnippetArticle> suggestions,
             @CategoryStatusEnum int status) {
         // Count the number of suggestions before this category.
         int globalPositionOffset = 0;
         for (Map.Entry<Integer, SuggestionsSection> entry : mSections.entrySet()) {
             if (entry.getKey() == category) break;
             globalPositionOffset += entry.getValue().getSuggestionsCount();
         }
         // Assign global indices to the new suggestions.
         for (SnippetArticle suggestion : suggestions) {
             suggestion.mGlobalPosition = globalPositionOffset + suggestion.mPosition;
         }
 
         mSections.get(category).addSuggestions(suggestions, status);
     }
 
-    private void updateChildren() {
-        mChildren.clear();
-        mChildren.add(mAboveTheFold);
-        mChildren.addAll(mSections.values());
-        mChildren.add(mSigninPromo);
-        mChildren.add(mAllDismissed);
-        mChildren.add(mFooter);
-        mChildren.add(mBottomSpacer);
-
-        updateAllDismissedVisibility();
-
-        // TODO(mvanouwerkerk): Notify about the subset of changed items. At least |mAboveTheFold|
-        // has not changed when refreshing from the all dismissed state.
-        notifyDataSetChanged();
-    }
-
-    private void updateAllDismissedVisibility() {
-        boolean showAllDismissed = hasAllBeenDismissed();
-        mAllDismissed.setVisible(showAllDismissed);
-        mFooter.setVisible(!showAllDismissed);
-    }
-
-    private void removeSection(SuggestionsSection section) {
-        mSections.remove(section.getCategory());
-        int startPos = getChildPositionOffset(section);
-        mChildren.remove(section);
-        notifyItemRangeRemoved(startPos, section.getItemCount());
-
-        updateAllDismissedVisibility();
-
-        notifyItemChanged(getBottomSpacerPosition());
-    }
-
-    @Override
-    public void onItemRangeChanged(TreeNode child, int itemPosition, int itemCount) {
-        assert child == mRoot;
-        notifyItemRangeChanged(itemPosition, itemCount);
-    }
-
-    @Override
-    public void onItemRangeInserted(TreeNode child, int itemPosition, int itemCount) {
-        assert child == mRoot;
-        notifyItemRangeInserted(itemPosition, itemCount);
-        notifyItemChanged(getItemCount() - 1); // Refresh the spacer too.
-
-        updateAllDismissedVisibility();
-    }
-
-    @Override
-    public void onItemRangeRemoved(TreeNode child, int itemPosition, int itemCount) {
-        assert child == mRoot;
-        notifyItemRangeRemoved(itemPosition, itemCount);
-        notifyItemChanged(getItemCount() - 1); // Refresh the spacer too.
-
-        updateAllDismissedVisibility();
-    }
-
-    @Override
-    public void onAttachedToRecyclerView(RecyclerView recyclerView) {
-        super.onAttachedToRecyclerView(recyclerView);
-
-        // We are assuming for now that the adapter is used with a single RecyclerView.
-        // Getting the reference as we are doing here is going to be broken if that changes.
-        assert mRecyclerView == null;
-
-        // FindBugs chokes on the cast below when not checked, raising BC_UNCONFIRMED_CAST
-        assert recyclerView instanceof NewTabPageRecyclerView;
-
-        mRecyclerView = (NewTabPageRecyclerView) recyclerView;
-    }
-
-    /**
-     * Dismisses the item at the provided adapter position. Can also cause the dismissal of other
-     * items or even entire sections.
-     */
-    // TODO(crbug.com/635567): Fix this properly.
-    @SuppressLint("SwitchIntDef")
-    public void dismissItem(int position) {
-        int itemViewType = getItemViewType(position);
-
-        // TODO(dgn): Polymorphism is supposed to allow to avoid that kind of stuff.
-        switch (itemViewType) {
-            case ItemViewType.STATUS:
-            case ItemViewType.ACTION:
-                dismissSection(getSuggestionsSection(position));
-                return;
-
-            case ItemViewType.SNIPPET:
-                dismissSuggestion(position);
-                return;
-
-            case ItemViewType.PROMO:
-                dismissPromo();
-                return;
-
-            default:
-                Log.wtf(TAG, "Unsupported dismissal of item of type %d", itemViewType);
-                return;
-        }
-    }
-
-    private void dismissSection(SuggestionsSection section) {
-        assert SnippetsConfig.isSectionDismissalEnabled();
-
-        announceItemRemoved(section.getHeaderText());
-
-        mNewTabPageManager.getSuggestionsSource().dismissCategory(section.getCategory());
-        removeSection(section);
-    }
-
-    private void dismissSuggestion(int position) {
-        SnippetArticle suggestion = mRoot.getSuggestionAt(position);
-        SuggestionsSource suggestionsSource = mNewTabPageManager.getSuggestionsSource();
-        if (suggestionsSource == null) {
-            // It is possible for this method to be called after the NewTabPage has had destroy()
-            // called. This can happen when NewTabPageRecyclerView.dismissWithAnimation() is called
-            // and the animation ends after the user has navigated away. In this case we cannot
-            // inform the native side that the snippet has been dismissed (http://crbug.com/649299).
-            return;
-        }
-
-        announceItemRemoved(suggestion.mTitle);
-
-        suggestionsSource.dismissSuggestion(suggestion);
-        SuggestionsSection section = getSuggestionsSection(position);
-        section.removeSuggestion(suggestion);
-    }
-
-    private void dismissPromo() {
-        announceItemRemoved(mSigninPromo.getHeader());
-        mSigninPromo.dismiss();
-    }
-
-    /**
-     * Returns another view holder that should be dismissed at the same time as the provided one.
-     */
-    public NewTabPageViewHolder getDismissSibling(ViewHolder viewHolder) {
-        int swipePos = viewHolder.getAdapterPosition();
-        int siblingPosDelta = mRoot.getDismissSiblingPosDelta(swipePos);
-        if (siblingPosDelta == 0) return null;
-
-        return (NewTabPageViewHolder) mRecyclerView.findViewHolderForAdapterPosition(
-                siblingPosDelta + swipePos);
-    }
-
-    private boolean hasAllBeenDismissed() {
-        return mSections.isEmpty() && !mSigninPromo.isVisible();
-    }
-
     private boolean canLoadSuggestions(@CategoryInt int category, @CategoryStatusEnum int status) {
         // We never want to add suggestions from unknown categories.
         if (!mSections.containsKey(category)) return false;
 
         // The status may have changed while the suggestions were loading, perhaps they should not
         // be displayed any more.
         if (!SnippetsBridge.isCategoryEnabled(status)) {
             Log.w(TAG, "Received suggestions for a disabled category (id=%d, status=%d)", category,
                     status);
             return false;
         }
 
         return true;
     }
 
     /**
-     * @param itemPosition The position of an item in the adapter.
-     * @return Returns the {@link SuggestionsSection} that contains the item at
-     *     {@code itemPosition}, or null if the item is not part of one.
+     * Dismisses a section.
+     * @param section The section to be dismissed.
      */
-    private SuggestionsSection getSuggestionsSection(int itemPosition) {
-        TreeNode child = mRoot.getChildForPosition(itemPosition);
-        if (!(child instanceof SuggestionsSection)) return null;
-        return (SuggestionsSection) child;
+    public void dismissSection(SuggestionsSection section) {
+        assert SnippetsConfig.isSectionDismissalEnabled();
+
+        mNewTabPageManager.getSuggestionsSource().dismissCategory(section.getCategory());
+        removeSection(section);
     }
 
-    private int getChildPositionOffset(TreeNode child) {
-        return mRoot.getStartingOffsetForChild(child);
+    private void removeSection(SuggestionsSection section) {
+        mSections.remove(section.getCategory());
+        willRemoveChild(section);
+        mChildren.remove(section);
     }
 
-    @VisibleForTesting
-    SnippetArticle getSuggestionAt(int position) {
-        return mRoot.getSuggestionAt(position);
+    /**
+     * Restores any sections that have been dismissed and triggers a new fetch.
+     */
+    public void restoreDismissedSections() {
+        mNewTabPageManager.getSuggestionsSource().restoreDismissedCategories();
+        resetSections(/* allowEmptySections = */ true);
+        mNewTabPageManager.getSuggestionsSource().fetchRemoteSuggestions();
     }
 
-    @VisibleForTesting
-    int getFirstPositionForType(@ItemViewType int viewType) {
-        int count = getItemCount();
-        for (int i = 0; i < count; i++) {
-            if (getItemViewType(i) == viewType) return i;
-        }
-        return RecyclerView.NO_POSITION;
+    /**
+     * @return Whether the list of sections is empty.
+     */
+    public boolean isEmpty() {
+        return mSections.isEmpty();
     }
 
-    SuggestionsSection getSectionForTesting(@CategoryInt int category) {
-        return mSections.get(category);
-    }
-
-    InnerNode getRootForTesting() {
-        return mRoot;
-    }
-
-    private void announceItemRemoved(String itemTitle) {
-        // In tests the RecyclerView can be null.
-        if (mRecyclerView == null) return;
-
-        mRecyclerView.announceForAccessibility(mRecyclerView.getResources().getString(
-                R.string.ntp_accessibility_item_removed, itemTitle));
-    }
-
-    private void announceItemRemoved(@StringRes int stringToAnnounce) {
-        // In tests the RecyclerView can be null.
-        if (mRecyclerView == null) return;
-
-        announceItemRemoved(mRecyclerView.getResources().getString(stringToAnnounce));
+    SuggestionsSection getSectionForTesting(@CategoryInt int categoryId) {
+        return mSections.get(categoryId);
     }
 }