Chromium Code Reviews| Index: android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientOnFormResubmissionTest.java |
| diff --git a/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientOnFormResubmissionTest.java b/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientOnFormResubmissionTest.java |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..f53eb57314d6f286b855c030764dee0dcc39488b |
| --- /dev/null |
| +++ b/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientOnFormResubmissionTest.java |
| @@ -0,0 +1,127 @@ |
| +// Copyright (c) 2012 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.android_webview.test; |
| + |
| +import android.content.Context; |
| +import android.os.Message; |
| +import android.test.suitebuilder.annotation.SmallTest; |
| + |
| +import org.apache.http.util.EncodingUtils; |
| +import org.chromium.android_webview.AwContents; |
| +import org.chromium.android_webview.test.util.TestWebServer; |
| +import org.chromium.base.test.util.Feature; |
| +import org.chromium.content.browser.ContentViewCore; |
| +import org.chromium.content.browser.test.util.TestCallbackHelperContainer; |
| + |
| +import java.util.concurrent.atomic.AtomicReference; |
| +import java.util.concurrent.TimeUnit; |
| +import java.util.concurrent.TimeoutException; |
| + |
| +/** |
| + * Tests if resubmission of post data is handled properly. |
| + */ |
| +public class AwContentsClientOnFormResubmissionTest extends AndroidWebViewTestBase { |
| + |
| + private static class TestAwContentsClient |
| + extends org.chromium.android_webview.test.TestAwContentsClient { |
| + |
| + // Number of times onFormResubmit is called. |
| + private int mResubmissions = 0; |
| + // Whether to resubmit Post data on reload. |
| + private boolean mResubmit = false; |
| + |
| + public int getResubmissions() { |
| + return mResubmissions; |
| + } |
| + public void setResubmit(boolean resubmit) { |
| + mResubmit = resubmit; |
| + } |
| + @Override |
| + public void onFormResubmission(Message dontResend, Message resend) { |
| + mResubmissions++; |
| + if (mResubmit) { |
| + resend.sendToTarget(); |
| + } else { |
| + dontResend.sendToTarget(); |
| + } |
| + } |
| + } |
| + |
| + // Server responses for load and reload of posts. |
| + private final String LOAD_RESPONSE = |
| + "<html><head><title>Load</title></head><body>HELLO</body></html>"; |
|
Ted C
2012/10/24 23:25:51
+4 indent
and next bit.
sgurun-gerrit only
2012/10/24 23:41:44
Done.
|
| + private final String RELOAD_RESPONSE = |
| + "<html><head><title>Reload</title></head><body>HELLO</body></html>"; |
| + |
| + // Server timeout in seconds. Used to detect dontResend case. |
| + private final int TIMEOUT = 3; |
| + |
| + // The web server. |
| + private TestWebServer mServer; |
| + // The mock client. |
| + private TestAwContentsClient mContentsClient; |
| + private AwContents mAwContents; |
| + |
| + @Override |
| + public void setUp() throws Exception { |
| + super.setUp(); |
| + mServer = new TestWebServer(false); |
| + mContentsClient = new TestAwContentsClient(); |
| + final AwTestContainerView testContainerView = |
| + createAwTestContainerViewOnMainSync(mContentsClient); |
|
Ted C
2012/10/24 23:25:51
+4 indent
sgurun-gerrit only
2012/10/24 23:41:44
Done.
|
| + mAwContents = testContainerView.getAwContents(); |
| + } |
| + |
| + @Override |
| + public void tearDown() throws Exception { |
| + mServer.shutdown(); |
| + super.tearDown(); |
| + } |
| + |
| + @SmallTest |
| + @Feature({"Android-WebView", "Navigation"}) |
| + public void testResend() throws Throwable { |
| + mContentsClient.setResubmit(true); |
| + doReload(); |
| + assertEquals(1, mContentsClient.getResubmissions()); |
| + assertEquals("Reload", getTitleOnUiThread(mAwContents)); |
| + } |
| + |
| + @SmallTest |
| + @Feature({"Android-WebView", "Navigation"}) |
| + public void testDontResend() throws Throwable { |
| + mContentsClient.setResubmit(false); |
| + doReload(); |
| + assertEquals(1, mContentsClient.getResubmissions()); |
| + assertEquals("Load", getTitleOnUiThread(mAwContents)); |
| + } |
| + |
| + protected void doReload() throws Throwable { |
| + String url = mServer.setResponse("/form", LOAD_RESPONSE, null); |
| + String postData = "content=blabla"; |
| + byte[] data = EncodingUtils.getBytes(postData, "BASE64"); |
| + postUrlSync(mAwContents, mContentsClient.getOnPageFinishedHelper(), url, data); |
| + assertEquals(0, mContentsClient.getResubmissions()); |
| + assertEquals("Load", getTitleOnUiThread(mAwContents)); |
| + // Verify reload works as expected. |
| + mServer.setResponse("/form", RELOAD_RESPONSE, null); |
| + TestCallbackHelperContainer.OnPageFinishedHelper onPageFinishedHelper = |
| + mContentsClient.getOnPageFinishedHelper(); |
| + int callCount = onPageFinishedHelper.getCallCount(); |
| + // Run reload on UI thread. |
| + getInstrumentation().runOnMainSync(new Runnable() { |
| + @Override |
| + public void run() { |
| + mAwContents.getContentViewCore().reload(); |
| + } |
| + }); |
| + try { |
| + // Wait for page finished callback, or a timeout. A timeout is necessary |
| + // to detect a dontResend response. |
| + onPageFinishedHelper.waitForCallback(callCount, 1, TIMEOUT, TimeUnit.SECONDS); |
| + } catch (TimeoutException e) { |
| + } |
| + } |
| +} |