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

 // Copyright 2015 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.support.test.filters.MediumTest;
 import android.support.test.filters.SmallTest;
 import android.util.Pair;

 import org.chromium.android_webview.AwContents;
 import org.chromium.android_webview.ErrorCodeConversionHelper;
 import org.chromium.android_webview.policy.AwPolicyProvider;
 import org.chromium.base.ThreadUtils;
 import org.chromium.base.test.util.Feature;
 import org.chromium.base.test.util.RetryOnFailure;
 import org.chromium.base.test.util.parameter.ParameterizedTest;
 import org.chromium.content.browser.test.util.TestCallbackHelperContainer;
 import org.chromium.net.test.util.TestWebServer;
 import org.chromium.policy.AbstractAppRestrictionsProvider;
 import org.chromium.policy.CombinedPolicyProvider;
 import org.chromium.policy.test.PolicyData;
 import org.chromium.policy.test.annotations.Policies;

 import java.util.ArrayList;
 import java.util.Arrays;

 /** Tests for the policy based URL filtering. */
 public class PolicyUrlFilteringTest extends AwTestBase {
     private TestAwContentsClient mContentsClient;
     private AwContents mAwContents;
     private TestWebServer mWebServer;
     private String mFooTestUrl;
     private String mBarTestUrl;
     private static final String sFooTestFilePath = "/foo.html";
     private static final String sFooWhitelistFilter = "localhost" + sFooTestFilePath;

     private static final String sBlacklistPolicyName = "com.android.browser:URLBlacklist";
     private static final String sWhitelistPolicyName = "com.android.browser:URLWhitelist";

     @Override
     public void setUp() throws Exception {
         super.setUp();
         mContentsClient = new TestAwContentsClient();
         mAwContents = createAwTestContainerViewOnMainSync(mContentsClient).getAwContents();
         mWebServer = TestWebServer.start();
         mFooTestUrl = mWebServer.setResponse(sFooTestFilePath, "<html><body>foo</body></html>",
                 new ArrayList<Pair<String, String>>());
         mBarTestUrl = mWebServer.setResponse("/bar.html", "<html><body>bar</body></html>",
                 new ArrayList<Pair<String, String>>());

         getInstrumentation().waitForIdleSync();
     }

     @Override
     public void tearDown() throws Exception {
         mWebServer.shutdown();
         super.tearDown();
     }

     // Tests transforming the bundle to native policies, reloading the policies and blocking
     // the navigation.
     @MediumTest
     @Feature({"AndroidWebView", "Policy"})
     // Run in single process only. crbug.com/615484
     @ParameterizedTest.Set
     @RetryOnFailure
     public void testBlacklistedUrl() throws Throwable {
         final AwPolicyProvider testProvider =
                 new AwPolicyProvider(getActivity().getApplicationContext());
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 CombinedPolicyProvider.get().registerProvider(testProvider);
             }
         });

         navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before */, 0 /* error count after*/);

         setFilteringPolicy(testProvider, new String[] {"localhost"}, new String[] {});

         navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before */, 1 /* error count after */);
         assertEquals(ErrorCodeConversionHelper.ERROR_CONNECT,
                 mContentsClient.getOnReceivedErrorHelper().getErrorCode());
     }

     // Tests getting a successful navigation with a whitelist.
     @MediumTest
     @Feature({"AndroidWebView", "Policy"})
     @Policies.Add({
             @Policies.Item(key = sBlacklistPolicyName, stringArray = {"*"}),
             @Policies.Item(key = sWhitelistPolicyName, stringArray = {sFooWhitelistFilter})})
     // Run in single process only. crbug.com/660517
     @ParameterizedTest.Set
     public void testWhitelistedUrl() throws Throwable {
         navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before */, 0 /* error count after */);

         // Make sure it goes through the blacklist
         navigateAndCheckOutcome(mBarTestUrl, 0 /* error count before */, 1 /* error count after */);
         assertEquals(ErrorCodeConversionHelper.ERROR_CONNECT,
                 mContentsClient.getOnReceivedErrorHelper().getErrorCode());
     }

     // Tests that bad policy values are properly handled
     @SmallTest
     @Feature({"AndroidWebView", "Policy"})
     @Policies.Add({
             @Policies.Item(key = sBlacklistPolicyName, string = "shouldBeAJsonArrayNotAString")})
     public void testBadPolicyValue() throws Exception {
         navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before */, 0 /* error count after */);
         // At the moment this test is written, a failure is a crash, a success is no crash.
     }

     /**
      * Synchronously loads the provided URL and checks that the number or reported errors for the
      * current context is the expected one.
      */
     private void navigateAndCheckOutcome(String url, int startingErrorCount, int expectedErrorCount)
             throws Exception {
         if (expectedErrorCount < startingErrorCount) {
             throw new IllegalArgumentException(
                     "The navigation error count can't decrease over time");
         }
         TestCallbackHelperContainer.OnReceivedErrorHelper onReceivedErrorHelper =
                 mContentsClient.getOnReceivedErrorHelper();
         TestCallbackHelperContainer.OnPageFinishedHelper onPageFinishedHelper =
                 mContentsClient.getOnPageFinishedHelper();

         assertEquals(startingErrorCount, onReceivedErrorHelper.getCallCount());

         loadUrlSync(mAwContents, onPageFinishedHelper, url);
         assertEquals(url, onPageFinishedHelper.getUrl());

         if (expectedErrorCount > startingErrorCount) {
             onReceivedErrorHelper.waitForCallback(
                     startingErrorCount, expectedErrorCount - startingErrorCount);
         }
         assertEquals(expectedErrorCount, onReceivedErrorHelper.getCallCount());
     }

     private void setFilteringPolicy(final AwPolicyProvider testProvider,
             final String[] blacklistUrls, final String[] whitelistUrls) {
         final PolicyData[] policies = {
             new PolicyData.StrArray(sBlacklistPolicyName, blacklistUrls),
             new PolicyData.StrArray(sWhitelistPolicyName, whitelistUrls)
         };

         AbstractAppRestrictionsProvider.setTestRestrictions(
                 PolicyData.asBundle(Arrays.asList(policies)));

         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 testProvider.refresh();
             }
         });

         // To avoid race conditions
         getInstrumentation().waitForIdleSync();
     }
 }
	// Copyright 2015 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.support.test.filters.MediumTest;
	import android.support.test.filters.SmallTest;
	import android.util.Pair;

	import org.chromium.android_webview.AwContents;
	import org.chromium.android_webview.ErrorCodeConversionHelper;
	import org.chromium.android_webview.policy.AwPolicyProvider;
	import org.chromium.base.ThreadUtils;
	import org.chromium.base.test.util.Feature;
	import org.chromium.base.test.util.RetryOnFailure;
	import org.chromium.base.test.util.parameter.ParameterizedTest;
	import org.chromium.content.browser.test.util.TestCallbackHelperContainer;
	import org.chromium.net.test.util.TestWebServer;
	import org.chromium.policy.AbstractAppRestrictionsProvider;
	import org.chromium.policy.CombinedPolicyProvider;
	import org.chromium.policy.test.PolicyData;
	import org.chromium.policy.test.annotations.Policies;

	import java.util.ArrayList;
	import java.util.Arrays;

	/** Tests for the policy based URL filtering. */
	public class PolicyUrlFilteringTest extends AwTestBase {
	private TestAwContentsClient mContentsClient;
	private AwContents mAwContents;
	private TestWebServer mWebServer;
	private String mFooTestUrl;
	private String mBarTestUrl;
	private static final String sFooTestFilePath = "/foo.html";
	private static final String sFooWhitelistFilter = "localhost" + sFooTestFilePath;

	private static final String sBlacklistPolicyName = "com.android.browser:URLBlacklist";
	private static final String sWhitelistPolicyName = "com.android.browser:URLWhitelist";

	@Override
	public void setUp() throws Exception {
	super.setUp();
	mContentsClient = new TestAwContentsClient();
	mAwContents = createAwTestContainerViewOnMainSync(mContentsClient).getAwContents();
	mWebServer = TestWebServer.start();
	mFooTestUrl = mWebServer.setResponse(sFooTestFilePath, "<html><body>foo</body></html>",
	new ArrayList<Pair<String, String>>());
	mBarTestUrl = mWebServer.setResponse("/bar.html", "<html><body>bar</body></html>",
	new ArrayList<Pair<String, String>>());

	getInstrumentation().waitForIdleSync();
	}

	@Override
	public void tearDown() throws Exception {
	mWebServer.shutdown();
	super.tearDown();
	}

	// Tests transforming the bundle to native policies, reloading the policies and blocking
	// the navigation.
	@MediumTest
	@Feature({"AndroidWebView", "Policy"})
	// Run in single process only. crbug.com/615484
	@ParameterizedTest.Set
	@RetryOnFailure
	public void testBlacklistedUrl() throws Throwable {
	final AwPolicyProvider testProvider =
	new AwPolicyProvider(getActivity().getApplicationContext());
	ThreadUtils.runOnUiThreadBlocking(new Runnable() {
	@Override
	public void run() {
	CombinedPolicyProvider.get().registerProvider(testProvider);
	}
	});

	navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before /, 0 / error count after*/);

	setFilteringPolicy(testProvider, new String[] {"localhost"}, new String[] {});

	navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before /, 1 / error count after */);
	assertEquals(ErrorCodeConversionHelper.ERROR_CONNECT,
	mContentsClient.getOnReceivedErrorHelper().getErrorCode());
	}

	// Tests getting a successful navigation with a whitelist.
	@MediumTest
	@Feature({"AndroidWebView", "Policy"})
	@Policies.Add({
	@Policies.Item(key = sBlacklistPolicyName, stringArray = {"*"}),
	@Policies.Item(key = sWhitelistPolicyName, stringArray = {sFooWhitelistFilter})})
	// Run in single process only. crbug.com/660517
	@ParameterizedTest.Set
	public void testWhitelistedUrl() throws Throwable {
	navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before /, 0 / error count after */);

	// Make sure it goes through the blacklist
	navigateAndCheckOutcome(mBarTestUrl, 0 /* error count before /, 1 / error count after */);
	assertEquals(ErrorCodeConversionHelper.ERROR_CONNECT,
	mContentsClient.getOnReceivedErrorHelper().getErrorCode());
	}

	// Tests that bad policy values are properly handled
	@SmallTest
	@Feature({"AndroidWebView", "Policy"})
	@Policies.Add({
	@Policies.Item(key = sBlacklistPolicyName, string = "shouldBeAJsonArrayNotAString")})
	public void testBadPolicyValue() throws Exception {
	navigateAndCheckOutcome(mFooTestUrl, 0 /* error count before /, 0 / error count after */);
	// At the moment this test is written, a failure is a crash, a success is no crash.
	}

	/**
	* Synchronously loads the provided URL and checks that the number or reported errors for the
	* current context is the expected one.
	*/
	private void navigateAndCheckOutcome(String url, int startingErrorCount, int expectedErrorCount)
	throws Exception {
	if (expectedErrorCount < startingErrorCount) {
	throw new IllegalArgumentException(
	"The navigation error count can't decrease over time");
	}
	TestCallbackHelperContainer.OnReceivedErrorHelper onReceivedErrorHelper =
	mContentsClient.getOnReceivedErrorHelper();
	TestCallbackHelperContainer.OnPageFinishedHelper onPageFinishedHelper =
	mContentsClient.getOnPageFinishedHelper();

	assertEquals(startingErrorCount, onReceivedErrorHelper.getCallCount());

	loadUrlSync(mAwContents, onPageFinishedHelper, url);
	assertEquals(url, onPageFinishedHelper.getUrl());

	if (expectedErrorCount > startingErrorCount) {
	onReceivedErrorHelper.waitForCallback(
	startingErrorCount, expectedErrorCount - startingErrorCount);
	}
	assertEquals(expectedErrorCount, onReceivedErrorHelper.getCallCount());
	}

	private void setFilteringPolicy(final AwPolicyProvider testProvider,
	final String[] blacklistUrls, final String[] whitelistUrls) {
	final PolicyData[] policies = {
	new PolicyData.StrArray(sBlacklistPolicyName, blacklistUrls),
	new PolicyData.StrArray(sWhitelistPolicyName, whitelistUrls)
	};

	AbstractAppRestrictionsProvider.setTestRestrictions(
	PolicyData.asBundle(Arrays.asList(policies)));

	ThreadUtils.runOnUiThreadBlocking(new Runnable() {
	@Override
	public void run() {
	testProvider.refresh();
	}
	});

	// To avoid race conditions
	getInstrumentation().waitForIdleSync();
	}
	}