// Copyright 2012 The Chromium Authors
// 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.os.Message;
import android.util.Base64;
import androidx.test.InstrumentationRegistry;
import androidx.test.filters.SmallTest;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.UseParametersRunnerFactory;
import org.chromium.android_webview.AwContents;
import org.chromium.base.ThreadUtils;
import org.chromium.base.test.util.Feature;
import org.chromium.content_public.browser.test.util.TestCallbackHelperContainer;
import org.chromium.content_public.common.ContentUrlConstants;
import org.chromium.net.test.util.TestWebServer;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
/** Tests if resubmission of post data is handled properly. */
@RunWith(Parameterized.class)
@UseParametersRunnerFactory(AwJUnit4ClassRunnerWithParameters.Factory.class)
public class AwContentsClientOnFormResubmissionTest extends AwParameterizedTest {
@Rule public AwActivityTestRule mActivityTestRule;
private static class TestAwContentsClient
extends org.chromium.android_webview.test.TestAwContentsClient {
// Number of times onFormResubmit is called.
private int mResubmissions;
// Whether to resubmit Post data on reload.
private boolean mResubmit;
// Whether to do resend/dontResend automatically;
private boolean mAutoProcess;
public int getResubmissions() {
return mResubmissions;
}
public void setResubmit(boolean resubmit) {
mResubmit = resubmit;
}
public void setAutoProcess(boolean autoProcess) {
mAutoProcess = autoProcess;
}
@Override
public void onFormResubmission(Message dontResend, Message resend) {
super.onFormResubmission(dontResend, resend);
mResubmissions++;
if (!mAutoProcess) {
return;
}
if (mResubmit) {
resend.sendToTarget();
} else {
dontResend.sendToTarget();
}
}
}
// Server responses for load and reload of posts.
private static final String LOAD_RESPONSE =
"<html><head><title>Load</title></head><body>HELLO</body></html>";
private static final String RELOAD_RESPONSE =
"<html><head><title>Reload</title></head><body>HELLO</body></html>";
// Server timeout in seconds. Used to detect dontResend case.
private static final long TIMEOUT = 3L;
// The web server.
private TestWebServer mServer;
// The mock client.
private TestAwContentsClient mContentsClient;
private AwContents mAwContents;
public AwContentsClientOnFormResubmissionTest(AwSettingsMutation param) {
this.mActivityTestRule = new AwActivityTestRule(param.getMutation());
}
@Before
public void setUp() throws Exception {
mServer = TestWebServer.start();
mContentsClient = new TestAwContentsClient();
final AwTestContainerView testContainerView =
mActivityTestRule.createAwTestContainerViewOnMainSync(mContentsClient);
mAwContents = testContainerView.getAwContents();
}
@After
public void tearDown() {
mServer.shutdown();
}
@Test
@SmallTest
@Feature({"AndroidWebView", "Navigation"})
public void testResend() throws Throwable {
mContentsClient.setAutoProcess(true);
mContentsClient.setResubmit(true);
doReload();
Assert.assertEquals(1, mContentsClient.getResubmissions());
Assert.assertEquals("Reload", mActivityTestRule.getTitleOnUiThread(mAwContents));
}
@Test
@SmallTest
@Feature({"AndroidWebView", "Navigation"})
public void testDontResend() throws Throwable {
mContentsClient.setAutoProcess(true);
mContentsClient.setResubmit(false);
doReload();
Assert.assertEquals(1, mContentsClient.getResubmissions());
Assert.assertEquals("Load", mActivityTestRule.getTitleOnUiThread(mAwContents));
}
@Test
@SmallTest
@Feature({"AndroidWebView", "Navigation"})
public void testPendingResubmit() throws Throwable {
mContentsClient.setResubmit(true);
mContentsClient.setAutoProcess(false);
String url = mServer.setResponse("/form", LOAD_RESPONSE, null);
String postData = "content=blabla";
byte[] data = Base64.encode(postData.getBytes("UTF-8"), Base64.DEFAULT);
mActivityTestRule.postUrlSync(
mAwContents, mContentsClient.getOnPageFinishedHelper(), url, data);
Assert.assertEquals(0, mContentsClient.getResubmissions());
Assert.assertEquals("Load", mActivityTestRule.getTitleOnUiThread(mAwContents));
// Verify reload works as expected.
mServer.setResponse("/form", RELOAD_RESPONSE, null);
TestCallbackHelperContainer.OnPageFinishedHelper onPageFinishedHelper =
mContentsClient.getOnPageFinishedHelper();
Assert.assertEquals(1, onPageFinishedHelper.getCallCount());
TestAwContentsClient.OnFormResubmissionHelper onFormResubmissionHelper =
mContentsClient.getOnFormResubmissionHelper();
// Run reload on UI thread.
ThreadUtils.runOnUiThreadBlocking(() -> mAwContents.getNavigationController().reload(true));
// Load another url to cancel form resubmission.
mActivityTestRule.loadUrlSync(
mAwContents, onPageFinishedHelper, ContentUrlConstants.ABOUT_BLANK_DISPLAY_URL);
Assert.assertEquals(2, onPageFinishedHelper.getCallCount());
Assert.assertEquals(1, mContentsClient.getResubmissions());
// Doing a resend, but expecting to fail.
onFormResubmissionHelper.resend();
try {
// Wait for page finished callback.
onPageFinishedHelper.waitForNext();
} catch (TimeoutException e) {
// Exception expected from pending resubmission case.
}
// Resend expects to fail, so onPageFinished will not be triggered again.
Assert.assertEquals(2, onPageFinishedHelper.getCallCount());
}
protected void doReload() throws Throwable {
String url = mServer.setResponse("/form", LOAD_RESPONSE, null);
String postData = "content=blabla";
byte[] data = Base64.encode(postData.getBytes("UTF-8"), Base64.DEFAULT);
mActivityTestRule.postUrlSync(
mAwContents, mContentsClient.getOnPageFinishedHelper(), url, data);
Assert.assertEquals(0, mContentsClient.getResubmissions());
Assert.assertEquals("Load", mActivityTestRule.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.
InstrumentationRegistry.getInstrumentation()
.runOnMainSync(() -> mAwContents.getNavigationController().reload(true));
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) {
// Exception expected from testDontResend case.
}
}
}
Codebase Browser
chromium