src/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientVisitedHistoryTest.java

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 package org.chromium.android_webview.test;
   6
   7 import android.test.suitebuilder.annotation.SmallTest;
   8 import android.webkit.ValueCallback;
   9
  10 import org.chromium.android_webview.AwContents;
  11 import org.chromium.android_webview.test.TestAwContentsClient.DoUpdateVisitedHistoryHelper;
  12 import org.chromium.base.test.util.Feature;
  13 import org.chromium.content.browser.test.util.CallbackHelper;
  14 import org.chromium.net.test.util.TestWebServer;
  15
  16 /**
  17  * Tests for AwContentsClient.getVisitedHistory and AwContents.doUpdateVisitedHistory callbacks.
  18  */
  19 public class AwContentsClientVisitedHistoryTest extends AwTestBase {
  20     private static class GetVisitedHistoryHelper extends CallbackHelper {
  21         private ValueCallback<String[]> mCallback;
  22         private boolean mSaveCallback = false;
  23
  24         public ValueCallback<String[]> getCallback() {
  25             assert getCallCount() > 0;
  26             return mCallback;
  27         }
  28
  29         public void setSaveCallback(boolean value) {
  30             mSaveCallback = value;
  31         }
  32
  33         public void notifyCalled(ValueCallback<String[]> callback) {
  34             if (mSaveCallback) {
  35                 mCallback = callback;
  36             }
  37             notifyCalled();
  38         }
  39     }
  40
  41     private static class VisitedHistoryTestAwContentsClient extends TestAwContentsClient {
  42
  43         private GetVisitedHistoryHelper mGetVisitedHistoryHelper;
  44
  45         public VisitedHistoryTestAwContentsClient() {
  46             mGetVisitedHistoryHelper = new GetVisitedHistoryHelper();
  47         }
  48
  49         public GetVisitedHistoryHelper getGetVisitedHistoryHelper() {
  50             return mGetVisitedHistoryHelper;
  51         }
  52
  53         @Override
  54         public void getVisitedHistory(ValueCallback<String[]> callback) {
  55             getGetVisitedHistoryHelper().notifyCalled(callback);
  56         }
  57
  58     }
  59
  60     private VisitedHistoryTestAwContentsClient mContentsClient =
  61         new VisitedHistoryTestAwContentsClient();
  62
  63     @Feature({"AndroidWebView"})
  64     @SmallTest
  65     public void testUpdateVisitedHistoryCallback() throws Throwable {
  66         AwTestContainerView testView = createAwTestContainerViewOnMainSync(mContentsClient);
  67         AwContents awContents = testView.getAwContents();
  68
  69         final String path = "/testUpdateVisitedHistoryCallback.html";
  70         final String html = "testUpdateVisitedHistoryCallback";
  71
  72         TestWebServer webServer = null;
  73         try {
  74             webServer = new TestWebServer(false);
  75             final String pageUrl = webServer.setResponse(path, html, null);
  76             final DoUpdateVisitedHistoryHelper doUpdateVisitedHistoryHelper =
  77                 mContentsClient.getDoUpdateVisitedHistoryHelper();
  78             int callCount = doUpdateVisitedHistoryHelper.getCallCount();
  79             loadUrlAsync(awContents, pageUrl);
  80             doUpdateVisitedHistoryHelper.waitForCallback(callCount);
  81             assertEquals(pageUrl, doUpdateVisitedHistoryHelper.getUrl());
  82             assertEquals(false, doUpdateVisitedHistoryHelper.getIsReload());
  83
  84             // Reload
  85             callCount = doUpdateVisitedHistoryHelper.getCallCount();
  86             loadUrlAsync(awContents, pageUrl);
  87             doUpdateVisitedHistoryHelper.waitForCallback(callCount);
  88             assertEquals(pageUrl, doUpdateVisitedHistoryHelper.getUrl());
  89             assertEquals(true, doUpdateVisitedHistoryHelper.getIsReload());
  90         } finally {
  91             if (webServer != null) webServer.shutdown();
  92         }
  93     }
  94
  95     @Feature({"AndroidWebView"})
  96     @SmallTest
  97     public void testGetVisitedHistoryExerciseCodePath() throws Throwable {
  98         // Due to security/privacy restrictions around the :visited css property, it is not
  99         // possible test this end to end without using the flaky and brittle capturing picture of
 100         // the web page. So we are doing the next best thing, exercising all the code paths.
 101         final GetVisitedHistoryHelper visitedHistoryHelper =
 102             mContentsClient.getGetVisitedHistoryHelper();
 103         final int callCount = visitedHistoryHelper.getCallCount();
 104         visitedHistoryHelper.setSaveCallback(true);
 105
 106         AwTestContainerView testView = createAwTestContainerViewOnMainSync(mContentsClient);
 107         AwContents awContents = testView.getAwContents();
 108
 109         final String path = "/testGetVisitedHistoryExerciseCodePath.html";
 110         final String visitedLinks[] = {"http://foo.com", "http://bar.com", null};
 111         final String html = "<a src=\"http://foo.com\">foo</a><a src=\"http://bar.com\">bar</a>";
 112
 113         TestWebServer webServer = null;
 114         try {
 115             webServer = new TestWebServer(false);
 116             final String pageUrl = webServer.setResponse(path, html, null);
 117             loadUrlSync(awContents, mContentsClient.getOnPageFinishedHelper(), pageUrl);
 118             visitedHistoryHelper.waitForCallback(callCount);
 119             assertNotNull(visitedHistoryHelper.getCallback());
 120
 121             visitedHistoryHelper.getCallback().onReceiveValue(visitedLinks);
 122             visitedHistoryHelper.getCallback().onReceiveValue(null);
 123
 124             loadUrlSync(awContents, mContentsClient.getOnPageFinishedHelper(), pageUrl);
 125         } finally {
 126             if (webServer != null) webServer.shutdown();
 127         }
 128     }
 129
 130     @Feature({"AndroidWebView"})
 131     @SmallTest
 132     public void testGetVisitedHistoryCallbackAfterDestroy() throws Throwable {
 133         GetVisitedHistoryHelper visitedHistoryHelper =
 134             mContentsClient.getGetVisitedHistoryHelper();
 135         visitedHistoryHelper.setSaveCallback(true);
 136         final int callCount = visitedHistoryHelper.getCallCount();
 137         AwTestContainerView testView = createAwTestContainerViewOnMainSync(mContentsClient);
 138         AwContents awContents = testView.getAwContents();
 139         loadUrlAsync(awContents, "about:blank");
 140         visitedHistoryHelper.waitForCallback(callCount);
 141         assertNotNull(visitedHistoryHelper.getCallback());
 142
 143         destroyAwContentsOnMainSync(awContents);
 144         visitedHistoryHelper.getCallback().onReceiveValue(new String[] {"abc.def"});
 145         visitedHistoryHelper.getCallback().onReceiveValue(null);
 146     }
 147 }