xref: /external/apache-harmony/luni/src/test/api/common/org/apache/harmony/luni/tests/java/net/ProxySelectorTest.java

/* Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.harmony.luni.tests.java.net;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.NetPermission;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.SocketException;
import java.net.URI;
import java.net.URISyntaxException;
import java.security.Permission;
import java.util.List;
import java.util.Properties;

import junit.framework.TestCase;

public class ProxySelectorTest extends TestCase {

	private static final String HTTP_PROXY_HOST = "127.0.0.1";

	private static final int HTTP_PROXY_PORT = 80;

	private static final String HTTPS_PROXY_HOST = "127.0.0.2";

	private static final int HTTPS_PROXY_PORT = 443;

	private static final String FTP_PROXY_HOST = "127.0.0.3";

	private static final int FTP_PROXY_PORT = 80;

	private static final String SOCKS_PROXY_HOST = "127.0.0.4";

	private static final int SOCKS_PROXY_PORT = 1080;

	private static URI httpUri;

	private static URI ftpUri;

	private static URI httpsUri;

	private static URI tcpUri;

	private List proxyList;

	private ProxySelector selector = ProxySelector.getDefault();

	static {
		try {
			httpUri = new URI("http://test.com");
			ftpUri = new URI("ftp://test.com");
			httpsUri = new URI("https://test.com");
			tcpUri = new URI("socket://host.com");
		} catch (URISyntaxException e) {

		}
	}

	/*
	 * Original system properties must be restored after running each test case.
	 */
	private Properties originalSystemProperties;

	/**
	 * @tests java.net.ProxySelector#getDefault()
	 */
	public void test_getDefault() {
		ProxySelector selector1 = ProxySelector.getDefault();
		assertNotNull(selector1);

		ProxySelector selector2 = ProxySelector.getDefault();
		assertSame(selector1, selector2);
	}

	/**
	 * @tests java.net.ProxySelector#setDefault(ProxySelector)}
	 */
	public void test_setDefaultLjava_net_ProxySelector() {
		ProxySelector originalSelector = ProxySelector.getDefault();
		try {
			ProxySelector newSelector = new MockProxySelector();
			ProxySelector.setDefault(newSelector);
			assertSame(newSelector, ProxySelector.getDefault());
			// use null to unset
			ProxySelector.setDefault(null);
			assertSame(null, ProxySelector.getDefault());
		} finally {
			ProxySelector.setDefault(originalSelector);
		}
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectExact()
			throws URISyntaxException {
		// no proxy, return a proxyList only contains NO_PROXY
		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.NO_PROXY);

		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTP_PROXY_HOST,HTTP_PROXY_PORT);

		proxyList = selector.select(httpsUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTPS_PROXY_HOST,HTTPS_PROXY_PORT);

		proxyList = selector.select(ftpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,FTP_PROXY_HOST,FTP_PROXY_PORT);

		proxyList = selector.select(tcpUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);

	}

    /**
     * @tests java.net.ProxySelector#select(URI)
     */
    public void test_selectLjava_net_URI_SelectExact_NullHost()
            throws URISyntaxException {
        // regression test for Harmony-1063
        httpUri = new URI("http://a@");
        ftpUri = new URI("ftp://a@");
        httpsUri = new URI("https://a@");
        tcpUri = new URI("socket://a@");
        // no proxy, return a proxyList only contains NO_PROXY
        proxyList = selector.select(httpUri);
        assertProxyEquals(proxyList, Proxy.NO_PROXY);

        // set http proxy
        System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
        System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
        // set https proxy
        System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
        System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
        // set ftp proxy
        System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
        System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
        // set socks proxy
        System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
        System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

        proxyList = selector.select(httpUri);
        assertProxyEquals(proxyList, Proxy.Type.HTTP, HTTP_PROXY_HOST,
                HTTP_PROXY_PORT);

        proxyList = selector.select(httpsUri);
        assertProxyEquals(proxyList, Proxy.Type.HTTP, HTTPS_PROXY_HOST,
                HTTPS_PROXY_PORT);

        proxyList = selector.select(ftpUri);
        assertProxyEquals(proxyList, Proxy.Type.HTTP, FTP_PROXY_HOST,
                FTP_PROXY_PORT);

        proxyList = selector.select(tcpUri);
        assertProxyEquals(proxyList, Proxy.Type.SOCKS, SOCKS_PROXY_HOST,
                SOCKS_PROXY_PORT);

    }

    //Regression for HARMONY-4281
    public void test_selectLjava_net_URI_SelectExact_NullHost_withNoProxyHostsProperty(){
        System.setProperty("http.nonProxyHosts", "localhost|127.0.0.1");
        System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
        System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
        // set https proxy
        System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
        System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
        // set ftp proxy
        System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
        System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
        // set socks proxy
        System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
        System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

        proxyList = selector.select(httpUri);
        assertProxyEquals(proxyList, Proxy.Type.HTTP, HTTP_PROXY_HOST,
                HTTP_PROXY_PORT);
    }

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectExact_DefaultPort()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);

		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);

		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTP_PROXY_HOST,HTTP_PROXY_PORT);

		proxyList = selector.select(httpsUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTPS_PROXY_HOST,HTTPS_PROXY_PORT);

		proxyList = selector.select(ftpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,FTP_PROXY_HOST,FTP_PROXY_PORT);

		proxyList = selector.select(tcpUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);

	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectExact_InvalidPort()
			throws URISyntaxException {
		final String INVALID_PORT = "abc";

		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", INVALID_PORT);
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", INVALID_PORT);
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", INVALID_PORT);
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksproxyPort", INVALID_PORT);

		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTP_PROXY_HOST,HTTP_PROXY_PORT);

		proxyList = selector.select(httpsUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,HTTPS_PROXY_HOST,HTTPS_PROXY_PORT);

		proxyList = selector.select(ftpUri);
		assertProxyEquals(proxyList,Proxy.Type.HTTP,FTP_PROXY_HOST,FTP_PROXY_PORT);

		proxyList = selector.select(tcpUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	// RI may fail this test case.
	// Uncomment this test case when regex.jar is ready.
	/*
	public void test_selectLjava_net_URI_Select_NonProxyHosts()
			throws URISyntaxException {
		// RI's bug. Some RIs may fail this test case.
		URI[] httpUris = { new URI("http://test.com"),
				new URI("http://10.10.1.2"), new URI("http://a"),
				new URI("http://def.abc.com") };
		URI[] ftpUris = { new URI("ftp://test.com"),
				new URI("ftp://10.10.1.2"), new URI("ftp://a"),
				new URI("ftp://def.abc.com") };

		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.nonProxyHosts", "a|b|tes*|10.10.*|*.abc.com");
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.nonProxyHosts", "a|b|tes*|10.10.*|*.abc.com");

		for (int i = 0; i < httpUris.length; i++) {
			proxyList = selector.select(httpUris[i]);
			assertProxyEquals(proxyList,Proxy.NO_PROXY);
		}

		for (int i = 0; i < ftpUris.length; i++) {
			proxyList = selector.select(ftpUris[i]);
			assertProxyEquals(proxyList,Proxy.NO_PROXY);
		}
	}*/

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectLikeHTTP()
			throws URISyntaxException {
		System.setProperty("http.proxyHost", "");
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectNoHTTP()
			throws URISyntaxException {
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));

		proxyList = selector.select(httpUri);
		assertProxyEquals(proxyList,Proxy.NO_PROXY);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectLikeHTTPS()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set https proxy host empty
		System.setProperty("http.proxyHost", "");
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

		proxyList = selector.select(httpsUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectNoHTTPS()
			throws URISyntaxException {
		// set https proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));

		proxyList = selector.select(httpsUri);
		assertProxyEquals(proxyList,Proxy.NO_PROXY);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectLikeFTP()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set ftp host empty
		System.setProperty("ftp.proxyHost", "");
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

		proxyList = selector.select(ftpUri);
		assertProxyEquals(proxyList,Proxy.Type.SOCKS,SOCKS_PROXY_HOST,SOCKS_PROXY_PORT);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectNoFTP()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));

		proxyList = selector.select(ftpUri);
		assertProxyEquals(proxyList,Proxy.NO_PROXY);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_SelectNoSOCKS()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set socks proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));

		proxyList = selector.select(tcpUri);
		assertProxyEquals(proxyList,Proxy.NO_PROXY);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_connectionFailedLjava_net_URILjava_net_SocketAddressLjava_io_IOException()
			throws URISyntaxException {
		// set http proxy
		System.setProperty("http.proxyHost", HTTP_PROXY_HOST);
		System.setProperty("http.proxyPort", String.valueOf(HTTP_PROXY_PORT));
		// set https proxy
		System.setProperty("https.proxyHost", HTTPS_PROXY_HOST);
		System.setProperty("https.proxyPort", String.valueOf(HTTPS_PROXY_PORT));
		// set ftp proxy
		System.setProperty("ftp.proxyHost", FTP_PROXY_HOST);
		System.setProperty("ftp.proxyPort", String.valueOf(FTP_PROXY_PORT));
		// set socks proxy
		System.setProperty("socksProxyHost", SOCKS_PROXY_HOST);
		System.setProperty("socksProxyPort", String.valueOf(SOCKS_PROXY_PORT));

		List proxyList1 = selector.select(httpUri);
		assertNotNull(proxyList1);
		assertEquals(1, proxyList1.size());
		Proxy proxy1 = (Proxy) proxyList1.get(0);
		selector
				.connectFailed(httpUri, proxy1.address(), new SocketException());

		List proxyList2 = selector.select(httpUri);
		assertNotNull(proxyList2);
		assertEquals(1, proxyList2.size());
		Proxy proxy2 = (Proxy) proxyList2.get(0);
		// Default implementation doesn't change the proxy list
		assertEquals(proxy1, proxy2);
	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_connectionFailedLjava_net_URILjava_net_SocketAddressLjava_io_IOException_IllegalArguement()
			throws URISyntaxException {
		SocketAddress sa = InetSocketAddress.createUnresolved("127.0.0.1", 0);
		try {
			selector.connectFailed(null, sa, new SocketException());
			fail("should throw IllegalArgumentException if any argument is null.");
		} catch (IllegalArgumentException e) {
			// expected
		}
		try {
			selector.connectFailed(httpUri, null, new SocketException());
			fail("should throw IllegalArgumentException if any argument is null.");
		} catch (IllegalArgumentException e) {
			// expected
		}
		try {
			selector.connectFailed(httpUri, sa, null);
			fail("should throw IllegalArgumentException if any argument is null.");
		} catch (IllegalArgumentException e) {
			// expected
		}

	}

	/**
	 * @tests java.net.ProxySelector#select(URI)
	 */
	public void test_selectLjava_net_URI_IllegalArgument()
			throws URISyntaxException {
		URI[] illegalUris = { new URI("abc"), new URI("http"), null };
		for (int i = 0; i < illegalUris.length; i++) {
			try {
				selector.select(illegalUris[i]);
				fail("should throw IllegalArgumentException");
			} catch (IllegalArgumentException e) {
				// expected
			}
		}
	}

	/*
	 * asserts whether selectedProxyList contains one and only one element,
	 * and the element equals proxy.
	 */
	private void assertProxyEquals(List selectedProxyList, Proxy proxy) {
		assertNotNull(selectedProxyList);
		assertEquals(1, selectedProxyList.size());
		assertEquals((Proxy) selectedProxyList.get(0), proxy);
	}

	/*
	 * asserts whether selectedProxyList contains one and only one element,
	 * and the element equals proxy which is represented by arguments "type",
	 * "host","port".
	 */
	private void assertProxyEquals(List selectedProxyList, Proxy.Type type,
			String host, int port) {
		SocketAddress sa = InetSocketAddress.createUnresolved(host, port);
		Proxy proxy = new Proxy(type, sa);
		assertProxyEquals(selectedProxyList, proxy);
	}

	/*
	 * Mock selector for setDefault test
	 */
	static class MockProxySelector extends ProxySelector {

		public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {

		}

		public List <Proxy> select(URI uri) {
			return null;
		}
	}

	/*
	 * @see junit.framework.TestCase#setUp()
	 */
	protected void setUp() throws Exception {
		super.setUp();
		// save original system properties
		originalSystemProperties = (Properties) System.getProperties().clone();
	}

	/*
	 * @see junit.framework.TestCase#tearDown()
	 */
	protected void tearDown() throws Exception {
		// restore original system properties
		System.setProperties(originalSystemProperties);
		super.tearDown();
	}
}