[Suggestions] Remove TreeNode.getSuggestionAt() in favor of a visitor.

chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/InnerNodeTest.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 static org.hamcrest.CoreMatchers.is;
 import static org.junit.Assert.assertThat;
 import static org.junit.Assert.fail;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.reset;
 import static org.mockito.Mockito.spy;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.verifyNoMoreInteractions;
 import static org.mockito.Mockito.verifyZeroInteractions;
 import static org.mockito.Mockito.when;
 
 import android.support.annotation.Nullable;
 
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 import org.robolectric.annotation.Config;
 
 import org.chromium.chrome.browser.ntp.cards.NewTabPageViewHolder.PartialBindCallback;
-import org.chromium.chrome.browser.ntp.snippets.SnippetArticle;
 import org.chromium.testing.local.LocalRobolectricTestRunner;
 
 import java.util.ArrayList;
 import java.util.List;
 
 /**
  * JUnit tests for {@link InnerNode}.
  */
 @RunWith(LocalRobolectricTestRunner.class)
 @Config(manifest = Config.NONE)
@@ skipping 44 matching lines @@
         mInnerNode.onBindViewHolder(holder, 6);
         mInnerNode.onBindViewHolder(holder, 11);
 
         verify(mChildren.get(0)).onBindViewHolder(holder, 0);
         verify(mChildren.get(2)).onBindViewHolder(holder, 2);
         verify(mChildren.get(4)).onBindViewHolder(holder, 0);
         verify(mChildren.get(6)).onBindViewHolder(holder, 0);
     }
 
     @Test
-    public void testGetSuggestion() {
-        SnippetArticle article1 = mock(SnippetArticle.class);
-        SnippetArticle article2 = mock(SnippetArticle.class);
-        SnippetArticle article3 = mock(SnippetArticle.class);
-        SnippetArticle article4 = mock(SnippetArticle.class);
-
-        when(mChildren.get(0).getSuggestionAt(0)).thenReturn(article1);
-        when(mChildren.get(2).getSuggestionAt(2)).thenReturn(article2);
-        when(mChildren.get(4).getSuggestionAt(0)).thenReturn(article3);
-        when(mChildren.get(6).getSuggestionAt(0)).thenReturn(article4);
-
-        assertThat(mInnerNode.getSuggestionAt(0), is(article1));
-        assertThat(mInnerNode.getSuggestionAt(5), is(article2));
-        assertThat(mInnerNode.getSuggestionAt(6), is(article3));
-        assertThat(mInnerNode.getSuggestionAt(11), is(article4));
-    }
-
-    @Test
     public void testAddChild() {
         final int itemCountBefore = mInnerNode.getItemCount();
 
         TreeNode child = makeDummyNode(23);
         mInnerNode.addChild(child);
 
         // The child should have been initialized and the parent hierarchy notified about it.
         verify(child).setParent(eq(mInnerNode));
         verify(mParent).onItemRangeInserted(mInnerNode, itemCountBefore, 23);
 
@@ skipping 117 matching lines @@
             mNextExpectedItemCount = null;
         }
     }
 
     private static TreeNode makeDummyNode(int itemCount) {
         TreeNode node = mock(TreeNode.class);
         when(node.getItemCount()).thenReturn(itemCount);
         return node;
     }
 }