android_webview/javatests/src/org/chromium/android_webview/test/ArchiveTest.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.support.test.filters.SmallTest;

 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.Callback;
 import org.chromium.base.task.PostTask;
 import org.chromium.base.test.util.Feature;
 import org.chromium.base.test.util.UrlUtils;
 import org.chromium.content_public.browser.UiThreadTaskTraits;
 import org.chromium.content_public.browser.test.util.TestThreadUtils;

 import java.io.File;
 import java.util.concurrent.Semaphore;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicReference;

 /**
  * Test suite for the WebView.saveWebArchive feature.
  */
 @RunWith(AwJUnit4ClassRunner.class)
 public class ArchiveTest {
     @Rule
     public AwActivityTestRule mActivityTestRule = new AwActivityTestRule();

     private static final long TEST_TIMEOUT = scaleTimeout(20000L);

     private static final String TEST_PAGE = UrlUtils.encodeHtmlDataUri(
             "<html><head></head><body>test</body></html>");

     private TestAwContentsClient mContentsClient = new TestAwContentsClient();
     private AwTestContainerView mTestContainerView;

     @Before
     public void setUp() throws Exception {
         mTestContainerView = mActivityTestRule.createAwTestContainerViewOnMainSync(mContentsClient);
     }

     private void deleteFile(String path) {
         File file = new File(path);
         if (file.exists()) Assert.assertTrue(file.delete());
         Assert.assertFalse(file.exists());
     }

     private void doArchiveTest(final AwContents contents, final String path,
             final boolean autoName, String expectedPath) throws InterruptedException {
         if (expectedPath != null) {
             deleteFile(expectedPath);
         }

         // Set up a handler to handle the completion callback
         final Semaphore s = new Semaphore(0);
         final AtomicReference<String> msgPath = new AtomicReference<String>();
         final Callback<String> callback = path1 -> {
             msgPath.set(path1);
             s.release();
         };

         // Generate MHTML and wait for completion
         PostTask.runOrPostTask(UiThreadTaskTraits.DEFAULT,
                 () -> contents.saveWebArchive(path, autoName, callback));
         Assert.assertTrue(s.tryAcquire(TEST_TIMEOUT, TimeUnit.MILLISECONDS));

         Assert.assertEquals(expectedPath, msgPath.get());
         if (expectedPath != null) {
             File file = new File(expectedPath);
             Assert.assertTrue(file.exists());
             Assert.assertTrue(file.length() > 0);
         } else {
             // A path was provided, but the expected path was null. This means the save should have
             // failed, and so there shouldn't be a file path path.
             if (path != null) {
                 Assert.assertFalse(new File(path).exists());
             }
         }
     }

     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testExplicitGoodPath() throws Throwable {
         final String path = new File(mActivityTestRule.getActivity().getFilesDir(), "test.mht")
                                     .getAbsolutePath();
         deleteFile(path);

         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         doArchiveTest(mTestContainerView.getAwContents(), path, false, path);
     }

     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testAutoGoodPath() throws Throwable {
         final String path = mActivityTestRule.getActivity().getFilesDir().getAbsolutePath() + "/";

         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         // Create the first archive
         {
             String expectedPath = path + "index.mht";
             doArchiveTest(mTestContainerView.getAwContents(), path, true, expectedPath);
         }

         // Create a second archive, making sure that the second archive's name is auto incremented.
         {
             String expectedPath = path + "index-1.mht";
             doArchiveTest(mTestContainerView.getAwContents(), path, true, expectedPath);
         }
     }

     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testExplicitBadPath() throws Throwable {
         final String path = new File("/foo/bar/baz.mht").getAbsolutePath();
         deleteFile(path);

         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         doArchiveTest(mTestContainerView.getAwContents(), path, false, null);
     }

     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testAutoBadPath() throws Throwable {
         final String path = new File("/foo/bar/").getAbsolutePath();
         deleteFile(path);

         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         doArchiveTest(mTestContainerView.getAwContents(), path, true, null);
     }

     /**
      * Ensure passing a null callback to saveWebArchive doesn't cause a crash.
      */
     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testNullCallbackNullPath() throws Throwable {
         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         saveWebArchiveAndWaitForUiPost(null, false /* autoname */, null /* callback */);
     }

     /**
      * Ensure passing a null callback to saveWebArchive doesn't cause a crash.
      */
     @Test
     @SmallTest
     @Feature({"AndroidWebView"})
     public void testNullCallbackGoodPath() throws Throwable {
         final String path = new File(mActivityTestRule.getActivity().getFilesDir(), "test.mht")
                 .getAbsolutePath();
         deleteFile(path);

         mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
                 mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

         saveWebArchiveAndWaitForUiPost(path, false /* autoname */, null /* callback */);
     }

     private void saveWebArchiveAndWaitForUiPost(
             final String path, boolean autoname, final Callback<String> callback) {
         PostTask.runOrPostTask(UiThreadTaskTraits.DEFAULT,
                 () -> mTestContainerView.getAwContents().saveWebArchive(path, false, callback));
         TestThreadUtils.runOnUiThreadBlocking(
                 () -> {
                         // Just wait for this task to having been posted on the UI thread.
                         // This ensures that if the implementation of saveWebArchive posts a
                         // task to the UI
                         // thread we will allow that task to run before finishing our test.
                       });
     }
 }
	// 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.support.test.filters.SmallTest;

	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.Callback;
	import org.chromium.base.task.PostTask;
	import org.chromium.base.test.util.Feature;
	import org.chromium.base.test.util.UrlUtils;
	import org.chromium.content_public.browser.UiThreadTaskTraits;
	import org.chromium.content_public.browser.test.util.TestThreadUtils;

	import java.io.File;
	import java.util.concurrent.Semaphore;
	import java.util.concurrent.TimeUnit;
	import java.util.concurrent.atomic.AtomicReference;

	/**
	* Test suite for the WebView.saveWebArchive feature.
	*/
	@RunWith(AwJUnit4ClassRunner.class)
	public class ArchiveTest {
	@Rule
	public AwActivityTestRule mActivityTestRule = new AwActivityTestRule();

	private static final long TEST_TIMEOUT = scaleTimeout(20000L);

	private static final String TEST_PAGE = UrlUtils.encodeHtmlDataUri(
	"<html><head></head><body>test</body></html>");

	private TestAwContentsClient mContentsClient = new TestAwContentsClient();
	private AwTestContainerView mTestContainerView;

	@Before
	public void setUp() throws Exception {
	mTestContainerView = mActivityTestRule.createAwTestContainerViewOnMainSync(mContentsClient);
	}

	private void deleteFile(String path) {
	File file = new File(path);
	if (file.exists()) Assert.assertTrue(file.delete());
	Assert.assertFalse(file.exists());
	}

	private void doArchiveTest(final AwContents contents, final String path,
	final boolean autoName, String expectedPath) throws InterruptedException {
	if (expectedPath != null) {
	deleteFile(expectedPath);
	}

	// Set up a handler to handle the completion callback
	final Semaphore s = new Semaphore(0);
	final AtomicReference<String> msgPath = new AtomicReference<String>();
	final Callback<String> callback = path1 -> {
	msgPath.set(path1);
	s.release();
	};

	// Generate MHTML and wait for completion
	PostTask.runOrPostTask(UiThreadTaskTraits.DEFAULT,
	() -> contents.saveWebArchive(path, autoName, callback));
	Assert.assertTrue(s.tryAcquire(TEST_TIMEOUT, TimeUnit.MILLISECONDS));

	Assert.assertEquals(expectedPath, msgPath.get());
	if (expectedPath != null) {
	File file = new File(expectedPath);
	Assert.assertTrue(file.exists());
	Assert.assertTrue(file.length() > 0);
	} else {
	// A path was provided, but the expected path was null. This means the save should have
	// failed, and so there shouldn't be a file path path.
	if (path != null) {
	Assert.assertFalse(new File(path).exists());
	}
	}
	}

	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testExplicitGoodPath() throws Throwable {
	final String path = new File(mActivityTestRule.getActivity().getFilesDir(), "test.mht")
	.getAbsolutePath();
	deleteFile(path);

	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	doArchiveTest(mTestContainerView.getAwContents(), path, false, path);
	}

	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testAutoGoodPath() throws Throwable {
	final String path = mActivityTestRule.getActivity().getFilesDir().getAbsolutePath() + "/";

	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	// Create the first archive
	{
	String expectedPath = path + "index.mht";
	doArchiveTest(mTestContainerView.getAwContents(), path, true, expectedPath);
	}

	// Create a second archive, making sure that the second archive's name is auto incremented.
	{
	String expectedPath = path + "index-1.mht";
	doArchiveTest(mTestContainerView.getAwContents(), path, true, expectedPath);
	}
	}

	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testExplicitBadPath() throws Throwable {
	final String path = new File("/foo/bar/baz.mht").getAbsolutePath();
	deleteFile(path);

	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	doArchiveTest(mTestContainerView.getAwContents(), path, false, null);
	}

	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testAutoBadPath() throws Throwable {
	final String path = new File("/foo/bar/").getAbsolutePath();
	deleteFile(path);

	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	doArchiveTest(mTestContainerView.getAwContents(), path, true, null);
	}

	/**
	* Ensure passing a null callback to saveWebArchive doesn't cause a crash.
	*/
	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testNullCallbackNullPath() throws Throwable {
	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	saveWebArchiveAndWaitForUiPost(null, false /* autoname /, null / callback */);
	}

	/**
	* Ensure passing a null callback to saveWebArchive doesn't cause a crash.
	*/
	@Test
	@SmallTest
	@Feature({"AndroidWebView"})
	public void testNullCallbackGoodPath() throws Throwable {
	final String path = new File(mActivityTestRule.getActivity().getFilesDir(), "test.mht")
	.getAbsolutePath();
	deleteFile(path);

	mActivityTestRule.loadUrlSync(mTestContainerView.getAwContents(),
	mContentsClient.getOnPageFinishedHelper(), TEST_PAGE);

	saveWebArchiveAndWaitForUiPost(path, false /* autoname /, null / callback */);
	}

	private void saveWebArchiveAndWaitForUiPost(
	final String path, boolean autoname, final Callback<String> callback) {
	PostTask.runOrPostTask(UiThreadTaskTraits.DEFAULT,
	() -> mTestContainerView.getAwContents().saveWebArchive(path, false, callback));
	TestThreadUtils.runOnUiThreadBlocking(
	() -> {
	// Just wait for this task to having been posted on the UI thread.
	// This ensures that if the implementation of saveWebArchive posts a
	// task to the UI
	// thread we will allow that task to run before finishing our test.
	});
	}
	}