android_webview/javatests/src/org/chromium/android_webview/test/AwContentsClientOnFormResubmissionTest.java - chromium/src - Git at Google

 // 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.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.
         mActivityTestRule.runOnUiThread(() -> 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.
         }
     }
 }
	// 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.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.
	mActivityTestRule.runOnUiThread(() -> 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.
	}
	}
	}