| 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 9a43670be3724ec50e037d1b25b742d0e09e57ba..c1653ebc3edf034d1beb2e550af1f3b1ffee198e 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
|
| @@ -13,6 +13,7 @@ import org.chromium.base.metrics.RecordHistogram;
|
| import org.chromium.base.test.util.Feature;
|
| import org.chromium.chrome.browser.ntp.NewTabPageView.NewTabPageManager;
|
| import org.chromium.chrome.browser.ntp.snippets.CategoryStatus;
|
| +import org.chromium.chrome.browser.ntp.snippets.KnownCategories;
|
| import org.chromium.chrome.browser.ntp.snippets.SnippetArticleListItem;
|
| import org.chromium.chrome.browser.ntp.snippets.SnippetsBridge;
|
| import org.chromium.chrome.browser.ntp.snippets.SnippetsSource.SnippetsObserver;
|
| @@ -73,9 +74,9 @@ public class NewTabPageAdapterTest {
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_SPACING, ntpa.getItemViewType(3));
|
|
|
| List<SnippetArticleListItem> snippets = createDummySnippets();
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
|
|
| - List<NewTabPageListItem> loadedItems = new ArrayList<>(ntpa.getItemsForTesting());
|
| + List<NewTabPageListItem> loadedItems = new ArrayList<>(ntpa.getItems());
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_ABOVE_THE_FOLD, ntpa.getItemViewType(0));
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_HEADER, ntpa.getItemViewType(1));
|
| assertEquals(snippets, loadedItems.subList(2, loadedItems.size() - 1));
|
| @@ -83,10 +84,10 @@ public class NewTabPageAdapterTest {
|
| NewTabPageListItem.VIEW_TYPE_SPACING, ntpa.getItemViewType(loadedItems.size() - 1));
|
|
|
| // The adapter should ignore any new incoming data.
|
| - mSnippetsObserver.onSnippetsReceived(
|
| - Arrays.asList(new SnippetArticleListItem[] {new SnippetArticleListItem(
|
| - "foo", "title1", "pub1", "txt1", "foo", "bar", 0, 0, 0)}));
|
| - assertEquals(loadedItems, ntpa.getItemsForTesting());
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES,
|
| + Arrays.asList(new SnippetArticleListItem[] { new SnippetArticleListItem(
|
| + "foo", "title1", "pub1", "txt1", "foo", "bar", 0, 0, 0) }));
|
| + assertEquals(loadedItems, ntpa.getItems());
|
| }
|
|
|
| /**
|
| @@ -100,7 +101,8 @@ public class NewTabPageAdapterTest {
|
| new NewTabPageAdapter(mNewTabPageManager, null, mSnippetsBridge, null);
|
|
|
| // If we don't get anything, we should be in the same situation as the initial one.
|
| - mSnippetsObserver.onSnippetsReceived(new ArrayList<SnippetArticleListItem>());
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES,
|
| + new ArrayList<SnippetArticleListItem>());
|
| assertEquals(4, ntpa.getItemCount());
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_ABOVE_THE_FOLD, ntpa.getItemViewType(0));
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_HEADER, ntpa.getItemViewType(1));
|
| @@ -109,8 +111,8 @@ public class NewTabPageAdapterTest {
|
|
|
| // We should load new snippets when we get notified about them.
|
| List<SnippetArticleListItem> snippets = createDummySnippets();
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| - List<NewTabPageListItem> loadedItems = new ArrayList<>(ntpa.getItemsForTesting());
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| + List<NewTabPageListItem> loadedItems = new ArrayList<>(ntpa.getItems());
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_ABOVE_THE_FOLD, ntpa.getItemViewType(0));
|
| assertEquals(NewTabPageListItem.VIEW_TYPE_HEADER, ntpa.getItemViewType(1));
|
| assertEquals(snippets, loadedItems.subList(2, loadedItems.size() - 1));
|
| @@ -118,10 +120,11 @@ public class NewTabPageAdapterTest {
|
| NewTabPageListItem.VIEW_TYPE_SPACING, ntpa.getItemViewType(loadedItems.size() - 1));
|
|
|
| // The adapter should ignore any new incoming data.
|
| - mSnippetsObserver.onSnippetsReceived(
|
| + mSnippetsObserver.onSuggestionsReceived(
|
| + KnownCategories.ARTICLES,
|
| Arrays.asList(new SnippetArticleListItem[] {new SnippetArticleListItem(
|
| "foo", "title1", "pub1", "txt1", "foo", "bar", 0, 0, 0)}));
|
| - assertEquals(loadedItems, ntpa.getItemsForTesting());
|
| + assertEquals(loadedItems, ntpa.getItems());
|
| }
|
|
|
| /**
|
| @@ -134,22 +137,25 @@ public class NewTabPageAdapterTest {
|
| new NewTabPageAdapter(mNewTabPageManager, null, mSnippetsBridge, null);
|
|
|
| List<SnippetArticleListItem> snippets = createDummySnippets();
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
|
|
| // If we get told that snippets are enabled, we just leave the current
|
| // ones there and not clear.
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.AVAILABLE);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.AVAILABLE);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
|
|
| // When snippets are disabled, we clear them and we should go back to
|
| // the situation with the status card.
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.SIGNED_OUT);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.SIGNED_OUT);
|
| assertEquals(4, ntpa.getItemCount());
|
|
|
| // The adapter should now be waiting for new snippets.
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.AVAILABLE);
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.AVAILABLE);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
| }
|
|
|
| @@ -165,28 +171,31 @@ public class NewTabPageAdapterTest {
|
| List<SnippetArticleListItem> snippets = createDummySnippets();
|
|
|
| // By default, status is INITIALIZING, so we can load snippets
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
|
|
| // If we have snippets, we should not load the new list.
|
| snippets.add(new SnippetArticleListItem("https://site.com/url1", "title1", "pub1", "txt1",
|
| "https://site.com/url1", "https://amp.site.com/url1", 0, 0, 0));
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size() - 1, ntpa.getItemCount());
|
|
|
| // When snippets are disabled, we should not be able to load them
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.SIGNED_OUT);
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.SIGNED_OUT);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(4, ntpa.getItemCount());
|
|
|
| // INITIALIZING lets us load snippets still.
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.INITIALIZING);
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.INITIALIZING);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
|
|
| // The adapter should now be waiting for new snippets.
|
| - mSnippetsObserver.onCategoryStatusChanged(CategoryStatus.AVAILABLE);
|
| - mSnippetsObserver.onSnippetsReceived(snippets);
|
| + mSnippetsObserver.onCategoryStatusChanged(KnownCategories.ARTICLES,
|
| + CategoryStatus.AVAILABLE);
|
| + mSnippetsObserver.onSuggestionsReceived(KnownCategories.ARTICLES, snippets);
|
| assertEquals(3 + snippets.size(), ntpa.getItemCount());
|
| }
|
|
|
|
|