| Index: chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SectionListTest.java
|
| diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SectionListTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SectionListTest.java
|
| index 0993d08bae28d38f8d2f54eb3a37c4c5bb6c60c8..8145e3e1aa73527c54a59ef5f40365fbc79136bf 100644
|
| --- a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SectionListTest.java
|
| +++ b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/SectionListTest.java
|
| @@ -22,7 +22,6 @@
|
|
|
| import org.chromium.base.Callback;
|
| import org.chromium.base.test.util.Feature;
|
| -import org.chromium.chrome.browser.ntp.NewTabPageView.NewTabPageManager;
|
| import org.chromium.chrome.browser.ntp.cards.ContentSuggestionsTestUtils.CategoryInfoBuilder;
|
| import org.chromium.chrome.browser.ntp.snippets.FakeSuggestionsSource;
|
| import org.chromium.chrome.browser.ntp.snippets.KnownCategories;
|
| @@ -30,6 +29,7 @@
|
| import org.chromium.chrome.browser.offlinepages.OfflinePageBridge;
|
| import org.chromium.chrome.browser.suggestions.SuggestionsMetricsReporter;
|
| import org.chromium.chrome.browser.suggestions.SuggestionsMetricsReporter.SuggestionRanker;
|
| +import org.chromium.chrome.browser.suggestions.SuggestionsUiDelegate;
|
| import org.chromium.testing.local.LocalRobolectricTestRunner;
|
|
|
| import java.util.List;
|
| @@ -41,7 +41,7 @@
|
| @Config(manifest = Config.NONE)
|
| public class SectionListTest {
|
| @Mock
|
| - private NewTabPageManager mNtpManager;
|
| + private SuggestionsUiDelegate mUiDelegate;
|
| @Mock
|
| private OfflinePageBridge mOfflinePageBridge;
|
| @Mock
|
| @@ -53,8 +53,8 @@ public void setUp() {
|
| MockitoAnnotations.initMocks(this);
|
| mSuggestionSource = new FakeSuggestionsSource();
|
|
|
| - when(mNtpManager.getSuggestionsSource()).thenReturn(mSuggestionSource);
|
| - when(mNtpManager.getSuggestionsMetricsReporter()).thenReturn(mMetricsReporter);
|
| + when(mUiDelegate.getSuggestionsSource()).thenReturn(mSuggestionSource);
|
| + when(mUiDelegate.getMetricsReporter()).thenReturn(mMetricsReporter);
|
| }
|
|
|
| @Test
|
| @@ -81,7 +81,7 @@ public void testGetSuggestionRank() {
|
| List<SnippetArticle> bookmarks =
|
| registerCategory(mSuggestionSource, KnownCategories.BOOKMARKS, 4);
|
|
|
| - SectionList sectionList = new SectionList(mNtpManager, mOfflinePageBridge);
|
| + SectionList sectionList = new SectionList(mUiDelegate, mOfflinePageBridge);
|
|
|
| assertThat(articles.get(0).getGlobalRank(), equalTo(0));
|
| assertThat(articles.get(0).getSectionRank(), equalTo(0));
|
| @@ -146,7 +146,7 @@ public void testGetActionItemRank() {
|
| registerCategory(mSuggestionSource,
|
| new CategoryInfoBuilder(KnownCategories.DOWNLOADS).withViewAllAction().build(), 3);
|
|
|
| - SectionList sectionList = new SectionList(mNtpManager, mOfflinePageBridge);
|
| + SectionList sectionList = new SectionList(mUiDelegate, mOfflinePageBridge);
|
| SuggestionRanker ranker = sectionList.getSuggestionRankerForTesting();
|
|
|
| assertThat(ranker.getActionItemRank(KnownCategories.ARTICLES), equalTo(0));
|
| @@ -158,7 +158,7 @@ public void testGetActionItemRank() {
|
| public void testGetActionItemRankWhenInvisible() {
|
| registerCategory(mSuggestionSource, KnownCategories.ARTICLES, 3);
|
|
|
| - SectionList sectionList = new SectionList(mNtpManager, mOfflinePageBridge);
|
| + SectionList sectionList = new SectionList(mUiDelegate, mOfflinePageBridge);
|
|
|
| sectionList.getSuggestionRankerForTesting().getActionItemRank(KnownCategories.ARTICLES);
|
| }
|
|
|