| Index: chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java
|
| index 450008e09e7430010450826c8935ed3bfbc5039d..980b9c93f6c84abc2aa16719c89000745e3efc6f 100644
|
| --- a/chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java
|
| @@ -9,6 +9,8 @@ import android.test.suitebuilder.annotation.MediumTest;
|
| import org.chromium.base.ThreadUtils;
|
| import org.chromium.base.test.util.FlakyTest;
|
| import org.chromium.base.test.util.Restriction;
|
| +import org.chromium.base.test.util.UrlUtils;
|
| +import org.chromium.chrome.browser.tab.EmptyTabObserver;
|
| import org.chromium.chrome.browser.tab.Tab;
|
| import org.chromium.chrome.browser.tabmodel.TabModel.TabLaunchType;
|
| import org.chromium.chrome.browser.tabmodel.TabModel.TabSelectionType;
|
| @@ -24,6 +26,7 @@ import java.util.concurrent.TimeoutException;
|
| */
|
| public class UndoTabModelTest extends ChromeTabbedActivityTestBase {
|
| private static final Tab[] EMPTY = new Tab[] { };
|
| + private static final String TEST_URL = UrlUtils.encodeHtmlDataUri("<html>poit.</html>");
|
|
|
| @Override
|
| public void startMainActivity() throws InterruptedException {
|
| @@ -75,6 +78,17 @@ public class UndoTabModelTest extends ChromeTabbedActivityTestBase {
|
| });
|
| }
|
|
|
| + private void createTabOnUiThread(final ChromeTabCreator tabCreator, final String url) {
|
| + ThreadUtils.runOnUiThreadBlocking(new Runnable() {
|
| + @Override
|
| + public void run() {
|
| + tabCreator.createNewTab(new LoadUrlParams(url),
|
| + TabLaunchType.FROM_CHROME_UI, null);
|
| + }
|
| + });
|
| + }
|
| +
|
| +
|
| private void selectTabOnUiThread(final TabModel model, final Tab tab) {
|
| ThreadUtils.runOnUiThreadBlocking(new Runnable() {
|
| @Override
|
| @@ -321,6 +335,15 @@ public class UndoTabModelTest extends ChromeTabbedActivityTestBase {
|
| }
|
| }
|
|
|
| + private void openRecentClosedTabOnUiThread(final TabModelSelector selector) {
|
| + ThreadUtils.runOnUiThreadBlocking(new Runnable() {
|
| + @Override
|
| + public void run() {
|
| + ((TabModelSelectorImpl) selector).getCurrentModel().openRecentClosedTab();
|
| + }
|
| + });
|
| + }
|
| +
|
| /**
|
| * Test undo with a single tab with the following actions/expected states:
|
| * Action Model List Close List Comprehensive List
|
| @@ -1410,4 +1433,94 @@ public class UndoTabModelTest extends ChromeTabbedActivityTestBase {
|
| assertTrue(tab0.isClosing());
|
| assertFalse(tab0.isInitialized());
|
| }
|
| +
|
| + /**
|
| + * Test open recently closed tab which uses rewinding list in java.
|
| + * Open recently closed tab first uses rewind list in java, then try to use
|
| + * native tab_restore_service if nothing in rewind list.
|
| + * @throws InterruptedException
|
| + */
|
| + @MediumTest
|
| + public void testOpenRecentClosedTab() throws InterruptedException {
|
| + TabModelSelector selector = getActivity().getTabModelSelector();
|
| + TabModel model = selector.getModel(false);
|
| + ChromeTabCreator tabCreator = getActivity().getTabCreator(false);
|
| +
|
| + createTabOnUiThread(tabCreator, TEST_URL);
|
| +
|
| + Tab tab0 = model.getTabAt(0);
|
| + Tab tab1 = model.getTabAt(1);
|
| + Tab[] allTabs = new Tab[]{tab0, tab1};
|
| +
|
| + closeTabOnUiThread(model, tab1, true);
|
| + checkState(model, new Tab[]{tab0}, tab0, new Tab[]{tab1}, allTabs, tab0);
|
| +
|
| + // ensure tab recovery, ordering, and reuse of {@link Tab} objects in java.
|
| + openRecentClosedTabOnUiThread(selector);
|
| + checkState(model, allTabs, tab0, EMPTY, allTabs, tab0);
|
| + }
|
| +
|
| + // helper class to wait for fully loaded page.
|
| + private class TestTabObserver extends EmptyTabObserver {
|
| + private CallbackHelper mCallbackHelper;
|
| + private TabModel mModel;
|
| + public TestTabObserver(CallbackHelper callbackHelper, TabModel model) {
|
| + super();
|
| + mModel = model;
|
| + mCallbackHelper = callbackHelper;
|
| + }
|
| + @Override
|
| + public void onPageLoadFinished(Tab tab) {
|
| + // close the page after tab is loaded, close it and commit closure.
|
| + try {
|
| + assertEquals(mModel.getCount(), 2);
|
| + UndoTabModelTest.this.closeTabOnUiThread(mModel, tab, false);
|
| + assertEquals(1, mModel.getCount());
|
| + mCallbackHelper.notifyCalled();
|
| + } catch (InterruptedException e) {
|
| + // ignore interrupt.
|
| + }
|
| + }
|
| + }
|
| +
|
| + /**
|
| + * Test open recently closed tab which uses native code from tab restore service.
|
| + * Open recently closed tab first use rewind list in java, then try to use
|
| + * native tab_restore_service if nothing in rewind list.
|
| + * @throws InterruptedException
|
| + */
|
| + @MediumTest
|
| + public void testOpenRecentClosedTabNative() throws InterruptedException {
|
| + final TabModelSelector selector = getActivity().getTabModelSelector();
|
| + final TabModel model = selector.getModel(false);
|
| + final ChromeTabCreator tabCreator = getActivity().getTabCreator(false);
|
| + final CallbackHelper tabCallbackHelper = new CallbackHelper();
|
| + final TestTabObserver observer = new TestTabObserver(tabCallbackHelper, model);
|
| +
|
| + // create new tab and attach observer to listen to loaded event.
|
| + // reason to do this is we need to wait the page to be fully loaded, then it has navigation
|
| + // history, then native code can successfully recover the page.
|
| + ThreadUtils.runOnUiThreadBlocking(new Runnable() {
|
| + @Override
|
| + public void run() {
|
| + tabCreator.createNewTab(new LoadUrlParams(TEST_URL),
|
| + TabLaunchType.FROM_CHROME_UI, null).addObserver(observer);
|
| + }
|
| + });
|
| +
|
| + // wait for the page to be fully loaded.
|
| + try {
|
| + tabCallbackHelper.waitForCallback(0);
|
| + } catch (TimeoutException e) {
|
| + fail();
|
| + }
|
| +
|
| + // recover the page.
|
| + openRecentClosedTabOnUiThread(selector);
|
| +
|
| + assertEquals(2, model.getCount());
|
| + Tab newTab = model.getTabAt(1);
|
| + assertEquals(TEST_URL, newTab.getUrl());
|
| + assertTrue(TabModelUtils.getCurrentTab(model) == newTab);
|
| + }
|
| }
|
|
|