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

 // Copyright 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 static org.chromium.base.test.util.ScalableTimeout.scaleTimeout;

 import android.os.Message;
 import android.support.test.InstrumentationRegistry;
 import android.support.test.filters.SmallTest;
 import android.util.Base64;

 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.chromium.android_webview.AwContents;
 import org.chromium.base.test.util.Feature;
 import org.chromium.content_public.browser.test.util.TestCallbackHelperContainer;
 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(AwJUnit4ClassRunner.class)
 public class AwContentsClientOnFormResubmissionTest {
     @Rule
     public AwActivityTestRule mActivityTestRule = new AwActivityTestRule();

     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;

         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 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 = scaleTimeout(3);

     // The web server.
     private TestWebServer mServer;
     // The mock client.
     private TestAwContentsClient mContentsClient;
     private AwContents mAwContents;

     @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() throws Exception {
         mServer.shutdown();
     }

     @Test
     @SmallTest
     @Feature({"AndroidWebView", "Navigation"})
     public void testResend() throws Throwable {
         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.setResubmit(false);
         doReload();
         Assert.assertEquals(1, mContentsClient.getResubmissions());
         Assert.assertEquals("Load", mActivityTestRule.getTitleOnUiThread(mAwContents));
     }

     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. 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 static org.chromium.base.test.util.ScalableTimeout.scaleTimeout;

	import android.os.Message;
	import android.support.test.InstrumentationRegistry;
	import android.support.test.filters.SmallTest;
	import android.util.Base64;

	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.chromium.android_webview.AwContents;
	import org.chromium.base.test.util.Feature;
	import org.chromium.content_public.browser.test.util.TestCallbackHelperContainer;
	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(AwJUnit4ClassRunner.class)
	public class AwContentsClientOnFormResubmissionTest {
	@Rule
	public AwActivityTestRule mActivityTestRule = new AwActivityTestRule();

	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;

	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 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 = scaleTimeout(3);

	// The web server.
	private TestWebServer mServer;
	// The mock client.
	private TestAwContentsClient mContentsClient;
	private AwContents mAwContents;

	@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() throws Exception {
	mServer.shutdown();
	}

	@Test
	@SmallTest
	@Feature({"AndroidWebView", "Navigation"})
	public void testResend() throws Throwable {
	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.setResubmit(false);
	doReload();
	Assert.assertEquals(1, mContentsClient.getResubmissions());
	Assert.assertEquals("Load", mActivityTestRule.getTitleOnUiThread(mAwContents));
	}

	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.
	}
	}
	}