Index: chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java |
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java |
index 6b3fa15eb6d7e0026734706529e1c04517e3cb14..b3414961d2c01e0a2ed0128b93f3ac8df6fc9d24 100644 |
--- a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java |
+++ b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java |
@@ -53,8 +53,8 @@ |
import org.chromium.chrome.R; |
import org.chromium.chrome.browser.ChromeFeatureList; |
import org.chromium.chrome.browser.EnableFeatures; |
+import org.chromium.chrome.browser.ntp.ContextMenuManager; |
import org.chromium.chrome.browser.ntp.NewTabPage.DestructionObserver; |
-import org.chromium.chrome.browser.ntp.NewTabPageView.NewTabPageManager; |
import org.chromium.chrome.browser.ntp.cards.ContentSuggestionsTestUtils.CategoryInfoBuilder; |
import org.chromium.chrome.browser.ntp.cards.SignInPromo.SigninObserver; |
import org.chromium.chrome.browser.ntp.snippets.CategoryInt; |
@@ -68,6 +68,7 @@ |
import org.chromium.chrome.browser.signin.SigninManager.SignInAllowedObserver; |
import org.chromium.chrome.browser.signin.SigninManager.SignInStateObserver; |
import org.chromium.chrome.browser.suggestions.SuggestionsMetricsReporter; |
+import org.chromium.chrome.browser.suggestions.SuggestionsUiDelegate; |
import org.chromium.testing.local.LocalRobolectricTestRunner; |
import java.util.ArrayList; |
@@ -92,7 +93,7 @@ |
@Mock |
private OfflinePageBridge mOfflinePageBridge; |
@Mock |
- private NewTabPageManager mNewTabPageManager; |
+ private SuggestionsUiDelegate mUiDelegate; |
/** |
* Stores information about a section that should be present in the adapter. |
@@ -210,10 +211,8 @@ public void setUp() { |
mSource.setInfoForCategory(category, |
new CategoryInfoBuilder(category).showIfEmpty().build()); |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(mSource); |
- when(mNewTabPageManager.getSuggestionsMetricsReporter()) |
- .thenReturn(mock(SuggestionsMetricsReporter.class)); |
- when(mNewTabPageManager.isCurrentPage()).thenReturn(true); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(mSource); |
+ when(mUiDelegate.getMetricsReporter()).thenReturn(mock(SuggestionsMetricsReporter.class)); |
reloadNtp(); |
} |
@@ -481,7 +480,7 @@ public void testSectionVisibleIfEmpty() { |
new CategoryInfoBuilder(category).showIfEmpty().build()); |
// 1.1 - Initial state |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
reloadNtp(); |
assertItemsFor(sectionWithStatusCard().withProgress()); |
@@ -507,7 +506,7 @@ public void testSectionVisibleIfEmpty() { |
suggestionsSource.setInfoForCategory(category, new CategoryInfoBuilder(category).build()); |
// 2.1 - Initial state |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
reloadNtp(); |
assertItemsFor(); |
@@ -537,7 +536,7 @@ public void testMoreButton() { |
.build()); |
// 1.1 - Initial state. |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
reloadNtp(); |
assertItemsFor(sectionWithStatusCard().withActionButton().withProgress()); |
@@ -564,7 +563,7 @@ public void testMoreButton() { |
new CategoryInfoBuilder(category).showIfEmpty().build()); |
// 2.1 - Initial state. |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
reloadNtp(); |
assertItemsFor(sectionWithStatusCard().withProgress()); |
@@ -638,7 +637,7 @@ public void testDynamicCategories() { |
@Feature({"Ntp"}) |
public void testCategoryOrder() { |
FakeSuggestionsSource suggestionsSource = new FakeSuggestionsSource(); |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
registerCategory(suggestionsSource, KnownCategories.ARTICLES, 0); |
registerCategory(suggestionsSource, KnownCategories.BOOKMARKS, 0); |
registerCategory(suggestionsSource, KnownCategories.PHYSICAL_WEB_PAGES, 0); |
@@ -658,7 +657,7 @@ public void testCategoryOrder() { |
// With a different order. |
suggestionsSource = new FakeSuggestionsSource(); |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
registerCategory(suggestionsSource, KnownCategories.ARTICLES, 0); |
registerCategory(suggestionsSource, KnownCategories.PHYSICAL_WEB_PAGES, 0); |
registerCategory(suggestionsSource, KnownCategories.DOWNLOADS, 0); |
@@ -678,7 +677,7 @@ public void testCategoryOrder() { |
// With unknown categories. |
suggestionsSource = new FakeSuggestionsSource(); |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
registerCategory(suggestionsSource, KnownCategories.ARTICLES, 0); |
registerCategory(suggestionsSource, KnownCategories.PHYSICAL_WEB_PAGES, 0); |
registerCategory(suggestionsSource, KnownCategories.DOWNLOADS, 0); |
@@ -703,7 +702,7 @@ public void testCategoryOrder() { |
public void testChangeNotifications() { |
FakeSuggestionsSource suggestionsSource = spy(new FakeSuggestionsSource()); |
registerCategory(suggestionsSource, KnownCategories.ARTICLES, 3); |
- when(mNewTabPageManager.getSuggestionsSource()).thenReturn(suggestionsSource); |
+ when(mUiDelegate.getSuggestionsSource()).thenReturn(suggestionsSource); |
@SuppressWarnings("unchecked") |
Callback<String> itemDismissedCallback = mock(Callback.class); |
@@ -799,7 +798,7 @@ public void testSigninPromo() { |
ArgumentCaptor<DestructionObserver> observers = |
ArgumentCaptor.forClass(DestructionObserver.class); |
- doNothing().when(mNewTabPageManager).addDestructionObserver(observers.capture()); |
+ doNothing().when(mUiDelegate).addDestructionObserver(observers.capture()); |
reloadNtp(); |
assertTrue(isSignInPromoVisible()); |
@@ -868,7 +867,7 @@ public void testAllDismissedVisibility() { |
ArgumentCaptor<DestructionObserver> observers = |
ArgumentCaptor.forClass(DestructionObserver.class); |
- verify(mNewTabPageManager, atLeastOnce()).addDestructionObserver(observers.capture()); |
+ verify(mUiDelegate, atLeastOnce()).addDestructionObserver(observers.capture()); |
SigninObserver signinObserver = null; |
for (DestructionObserver observer : observers.getAllValues()) { |
@@ -1046,8 +1045,8 @@ private SectionDescriptor sectionWithStatusCard() { |
} |
private void reloadNtp() { |
- mAdapter = new NewTabPageAdapter(mNewTabPageManager, mock(View.class), null, |
- mOfflinePageBridge); |
+ mAdapter = new NewTabPageAdapter(mUiDelegate, mock(View.class), null, mOfflinePageBridge, |
+ mock(ContextMenuManager.class)); |
} |
private void assertArticlesEqual(List<SnippetArticle> articles, int start, int end) { |