Add NewTabPageRecyclerViewTest.

chrome/android/javatests/src/org/chromium/chrome/browser/ntp/cards/NewTabPageRecyclerViewTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/ntp/cards/NewTabPageRecyclerViewTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/ntp/cards/NewTabPageRecyclerViewTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/ntp/cards/NewTabPageRecyclerViewTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..621394479a485ed87bede1991a092719f63f4132
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/ntp/cards/NewTabPageRecyclerViewTest.java
@@ -0,0 +1,335 @@
+// 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.support.v7.widget.RecyclerView;
+import android.support.v7.widget.RecyclerView.ViewHolder;
+import android.test.suitebuilder.annotation.MediumTest;
+import android.view.View;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.CallbackHelper;
+import org.chromium.base.test.util.CommandLineFlags;
+import org.chromium.base.test.util.Feature;
+import org.chromium.base.test.util.RetryOnFailure;
+import org.chromium.chrome.R;
+import org.chromium.chrome.browser.UrlConstants;
+import org.chromium.chrome.browser.ntp.ContextMenuHandler;
+import org.chromium.chrome.browser.ntp.FakeMostVisitedSites;
+import org.chromium.chrome.browser.ntp.NTPTileSource;
+import org.chromium.chrome.browser.ntp.NewTabPage;
+import org.chromium.chrome.browser.ntp.NewTabPageView;
+import org.chromium.chrome.browser.ntp.snippets.CategoryStatus;
+import org.chromium.chrome.browser.ntp.snippets.ContentSuggestionsCardLayout;
+import org.chromium.chrome.browser.ntp.snippets.FakeSuggestionsSource;
+import org.chromium.chrome.browser.ntp.snippets.KnownCategories;
+import org.chromium.chrome.browser.ntp.snippets.SnippetArticle;
+import org.chromium.chrome.browser.tab.Tab;
+import org.chromium.chrome.test.ChromeTabbedActivityTestBase;
+import org.chromium.chrome.test.util.ChromeTabUtils;
+import org.chromium.chrome.test.util.NewTabPageTestUtils;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.content.browser.test.util.TestTouchUtils;
+import org.chromium.net.test.EmbeddedTestServer;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.TimeoutException;
+
+/**
+ * Instrumentation tests for {@link NewTabPageRecyclerView}.
+ */
+@RetryOnFailure
+public class NewTabPageRecyclerViewTest extends ChromeTabbedActivityTestBase {
+    private static final String TEST_PAGE = "/chrome/test/data/android/navigate/simple.html";
+    private static final String[] FAKE_MOST_VISITED_TITLES = new String[] {"Simple"};
+    private static final String[] FAKE_MOST_VISITED_WHITELIST_ICON_PATHS = new String[] {""};
+    private static final int[] FAKE_MOST_VISITED_SOURCES = new int[] {NTPTileSource.TOP_SITES};
+
+    private Tab mTab;
+    private NewTabPage mNtp;
+    private String[] mFakeMostVisitedUrls;
+    private FakeMostVisitedSites mFakeMostVisitedSites;
+    private EmbeddedTestServer mTestServer;
+    private FakeSuggestionsSource mSource;
+
+    @Override
+    protected void setUp() throws Exception {
+        mTestServer = EmbeddedTestServer.createAndStartServer(getInstrumentation().getContext());
+        mFakeMostVisitedUrls = new String[] {mTestServer.getURL(TEST_PAGE)};
+
+        mSource = new FakeSuggestionsSource();
+        mSource.setInfoForCategory(KnownCategories.ARTICLES,
+                new SuggestionsCategoryInfo(KnownCategories.ARTICLES, "Articles test title",
+                        ContentSuggestionsCardLayout.FULL_CARD, /*hasMoreAction=*/true,
+                        /*hasReloadAction=*/true, /*hasViewAllAction=*/false, /*showIfEmpty=*/true,
+                        "noSuggestionsMessage"));
+        mSource.setStatusForCategory(KnownCategories.ARTICLES, CategoryStatus.INITIALIZING);
+        NewTabPage.setSuggestionsSourceForTests(mSource);
+
+        super.setUp();
+    }
+
+    @Override
+    protected void tearDown() throws Exception {
+        NewTabPage.setMostVisitedSitesForTests(null);
+        NewTabPage.setSuggestionsSourceForTests(null);
+        mTestServer.stopAndDestroyServer();
+
+        super.tearDown();
+    }
+
+    @Override
+    public void startMainActivity() throws InterruptedException {
+        startMainActivityOnBlankPage();
+        mTab = getActivity().getActivityTab();
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                // Create FakeMostVisitedSites after starting the activity, since it depends on
+                // native code.
+                mFakeMostVisitedSites = new FakeMostVisitedSites(mTab.getProfile(),
+                        FAKE_MOST_VISITED_TITLES, mFakeMostVisitedUrls,
+                        FAKE_MOST_VISITED_WHITELIST_ICON_PATHS, FAKE_MOST_VISITED_SOURCES);
+            }
+        });
+        NewTabPage.setMostVisitedSitesForTests(mFakeMostVisitedSites);
+
+        loadUrl(UrlConstants.NTP_URL);
+        NewTabPageTestUtils.waitForNtpLoaded(mTab);
+
+        assertTrue(mTab.getNativePage() instanceof NewTabPage);
+        mNtp = (NewTabPage) mTab.getNativePage();
+    }
+
+    @MediumTest
+    @Feature({"NewTabPage"})
+    @CommandLineFlags.Add("enable-features=NTPSnippets")
+    public void testClickSuggestion() throws InterruptedException, TimeoutException {
+        setSuggestionsAndWaitForUpdate(10);
+        List<SnippetArticle> suggestions =
+                mSource.getSuggestionsForCategory(KnownCategories.ARTICLES);
+
+        // Scroll the last suggestion into view and click it.
+        SnippetArticle suggestion = suggestions.get(suggestions.size() - 1);
+        int suggestionPosition = getAdapter().getSuggestionPosition(suggestion);
+        scrollToPosition(suggestionPosition);
+        final View suggestionView = waitForView(suggestionPosition);
+        ChromeTabUtils.waitForTabPageLoaded(mTab, new Runnable() {
+            @Override
+            public void run() {
+                singleClickView(suggestionView);
+            }
+        });
+        assertEquals(suggestion.mUrl, mTab.getUrl());
+    }
+
+    @MediumTest
+    @Feature({"NewTabPage"})
+    @CommandLineFlags.Add("enable-features=NTPSnippets")
+    public void testAllDismissed() throws InterruptedException, TimeoutException {
+        setSuggestionsAndWaitForUpdate(3);
+        assertEquals(3, mSource.getSuggestionsForCategory(KnownCategories.ARTICLES).size());
+        assertFalse(getAdapter().hasAllBeenDismissed());
+        assertEquals(1, mSource.getCategories().length);
+        assertEquals(KnownCategories.ARTICLES, mSource.getCategories()[0]);
+
+        // Dismiss the sign in promo.
+        int signinPromoPosition = getAdapter().getSignInPromoPosition();
+        scrollToPosition(signinPromoPosition);
+        View signinPromoView = waitForView(signinPromoPosition);
+        getAdapter().dismissItem(signinPromoPosition);
+        waitForViewToDetach(signinPromoView);
+
+        // Dismiss all the cards, including status cards, which dismisses the associated category.
+        int cardPosition = getAdapter().getFirstCardPosition();
+        while (cardPosition != RecyclerView.NO_POSITION) {
+            scrollToPosition(cardPosition);
+            View cardView = waitForView(cardPosition);
+            getAdapter().dismissItem(cardPosition);
+            waitForViewToDetach(cardView);
+            cardPosition = getAdapter().getFirstCardPosition();
+        }
+        assertTrue(getAdapter().hasAllBeenDismissed());
+        assertEquals(0, mSource.getCategories().length);
+
+        // Click the refresh button on the all dismissed item.
+        int allDismissedPosition = getAdapter().getLastContentItemPosition();
+        scrollToPosition(allDismissedPosition);
+        View allDismissedView = waitForView(allDismissedPosition);
+        singleClickView(allDismissedView.findViewById(R.id.action_button));
+        waitForViewToDetach(allDismissedView);
+        assertEquals(1, mSource.getCategories().length);
+        assertEquals(KnownCategories.ARTICLES, mSource.getCategories()[0]);
+    }
+
+    @MediumTest
+    @Feature({"NewTabPage"})
+    @CommandLineFlags.Add("enable-features=NTPSnippets")
+    public void testDismissArticleWithContextMenu() throws InterruptedException, TimeoutException {
+        setSuggestionsAndWaitForUpdate(10);
+        List<SnippetArticle> suggestions =
+                mSource.getSuggestionsForCategory(KnownCategories.ARTICLES);
+        assertEquals(10, suggestions.size());
+
+        // Scroll a suggestion into view.
+        int suggestionPosition =
+                getAdapter().getSuggestionPosition(suggestions.get(suggestions.size() - 1));
+        scrollToPosition(suggestionPosition);
+        View suggestionView = waitForView(suggestionPosition);
+
+        // Dismiss the suggestion using the context menu.
+        invokeContextMenu(suggestionView, ContextMenuHandler.ID_REMOVE);
+        waitForViewToDetach(suggestionView);
+
+        suggestions = mSource.getSuggestionsForCategory(KnownCategories.ARTICLES);
+        assertEquals(9, suggestions.size());
+    }
+
+    private NewTabPageView getNtpView() {
+        return mNtp.getNewTabPageView();
+    }
+
+    private NewTabPageRecyclerView getRecyclerView() {
+        return (NewTabPageRecyclerView) getNtpView().getWrapperView();
+    }
+
+    private NewTabPageAdapter getAdapter() {
+        return getRecyclerView().getNewTabPageAdapter();
+    }
+
+    private void scrollToPosition(final int position) {
+        final NewTabPageRecyclerView recyclerView = getRecyclerView();
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                recyclerView.getLinearLayoutManager().scrollToPositionWithOffset(
+                        position, getActivity().getResources().getDimensionPixelSize(
+                                          R.dimen.tab_strip_height));
+            }
+        });
+    }
+
+    private void setSuggestionsAndWaitForUpdate(final int suggestionsCount)
+            throws InterruptedException, TimeoutException {
+        final FakeSuggestionsSource source = mSource;
+        final NewTabPageAdapter adapter = getAdapter();
+        final CallbackHelper callback = new CallbackHelper();
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                adapter.registerAdapterDataObserver(new RecyclerView.AdapterDataObserver() {
+                    @Override
+                    public void onItemRangeInserted(int positionStart, int itemCount) {
+                        adapter.unregisterAdapterDataObserver(this);
+                        callback.notifyCalled();
+                    }
+                });
+                source.setStatusForCategory(KnownCategories.ARTICLES, CategoryStatus.AVAILABLE);
+                source.setSuggestionsForCategory(
+                        KnownCategories.ARTICLES, buildSuggestions(suggestionsCount));
+            }
+        });
+        callback.waitForCallback(0);
+        waitForStableRecyclerView();
+    }
+
+    private List<SnippetArticle> buildSuggestions(int suggestionsCount) {
+        List<SnippetArticle> suggestions = new ArrayList<>();
+        for (int i = 0; i < suggestionsCount; i++) {
+            String url = mTestServer.getURL(TEST_PAGE) + "#" + i;
+            suggestions.add(new SnippetArticle(KnownCategories.ARTICLES, "id" + i, "title" + i,
+                    "publisher" + i, "previewText" + i, url, url, 1466614774 + i, 10f, 0,
+                    ContentSuggestionsCardLayout.FULL_CARD));
+        }
+        return suggestions;
+    }
+
+    private void invokeContextMenu(View view, int contextMenuItemId) {
+        TestTouchUtils.longClickView(getInstrumentation(), view);
+        assertTrue(
+                getInstrumentation().invokeContextMenuAction(getActivity(), contextMenuItemId, 0));
+    }
+
+    private View waitForView(final int position) throws InterruptedException {
+        final NewTabPageRecyclerView recyclerView = getRecyclerView();
+
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                ViewHolder viewHolder = recyclerView.findViewHolderForAdapterPosition(position);
+
+                if (viewHolder == null) {
+                    updateFailureReason("Cannot find view holder for position " + position + ".");
+                    return false;
+                }
+
+                if (viewHolder.itemView.getParent() == null) {
+                    updateFailureReason("The view is not attached for position " + position + ".");
+                    return false;
+                }
+
+                return true;
+            }
+        });
+
+        waitForStableRecyclerView();
+
+        return recyclerView.findViewHolderForAdapterPosition(position).itemView;
+    }
+
+    private void waitForViewToDetach(final View view)
+            throws InterruptedException, TimeoutException {
+        final RecyclerView recyclerView = getRecyclerView();
+        final CallbackHelper callback = new CallbackHelper();
+
+        recyclerView.addOnChildAttachStateChangeListener(
+                new RecyclerView.OnChildAttachStateChangeListener() {
+                    @Override
+                    public void onChildViewAttachedToWindow(View view) {}
+
+                    @Override
+                    public void onChildViewDetachedFromWindow(View detachedView) {
+                        if (detachedView == view) {
+                            recyclerView.removeOnChildAttachStateChangeListener(this);
+                            callback.notifyCalled();
+                        }
+                    }
+                });
+        callback.waitForCallback("The view did not detach.", 0);
+
+        waitForStableRecyclerView();
+    }
+
+    private void waitForStableRecyclerView() throws InterruptedException {
+        final RecyclerView recyclerView = getRecyclerView();
+
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                if (recyclerView.isComputingLayout()) {
+                    updateFailureReason("The recycler view is computing layout.");
+                    return false;
+                }
+
+                if (recyclerView.isLayoutFrozen()) {
+                    updateFailureReason("The recycler view layout is frozen.");
+                    return false;
+                }
+
+                if (recyclerView.isAnimating()) {
+                    updateFailureReason("The recycler view is animating.");
+                    return false;
+                }
+
+                return true;
+            }
+        });
+    }
+}