// Copyright 2015 The Bazel Authors. All rights reserved.
//
// Licensed 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 com.google.devtools.build.lib.bazel.repository.downloader;
import static com.google.common.truth.Truth.assertThat;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
import com.google.common.collect.ImmutableMap;
import java.io.IOException;
import java.net.Proxy;
import java.net.URL;
import java.util.Map;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/**
* Tests for {@link ProxyHelper}.
*/
@RunWith(JUnit4.class)
public class ProxyHelperTest {
@Test
public void testCreateIfNeededHttpLowerCase() throws Exception {
ProxyHelper helper = new ProxyHelper(ImmutableMap.of("http_proxy", "http://my.example.com"));
Proxy proxy = helper.createProxyIfNeeded(new URL("http://www.something.com"));
assertThat(proxy.toString()).endsWith("my.example.com:80");
}
@Test
public void testCreateIfNeededHttpUpperCase() throws Exception {
ProxyHelper helper = new ProxyHelper(ImmutableMap.of("HTTP_PROXY", "http://my.example.com"));
Proxy proxy = helper.createProxyIfNeeded(new URL("http://www.something.com"));
assertThat(proxy.toString()).endsWith("my.example.com:80");
}
@Test
public void testCreateIfNeededHttpsLowerCase() throws Exception {
ProxyHelper helper = new ProxyHelper(ImmutableMap.of("https_proxy", "https://my.example.com"));
Proxy proxy = helper.createProxyIfNeeded(new URL("https://www.something.com"));
assertThat(proxy.toString()).endsWith("my.example.com:443");
}
@Test
public void testCreateIfNeededHttpsUpperCase() throws Exception {
ProxyHelper helper = new ProxyHelper(ImmutableMap.of("HTTPS_PROXY", "https://my.example.com"));
Proxy proxy = helper.createProxyIfNeeded(new URL("https://www.something.com"));
assertThat(proxy.toString()).endsWith("my.example.com:443");
}
@Test
public void testNoProxy() throws Exception {
// Empty address.
Proxy proxy = ProxyHelper.createProxy(null);
assertEquals(Proxy.NO_PROXY, proxy);
proxy = ProxyHelper.createProxy("");
assertEquals(Proxy.NO_PROXY, proxy);
Map<String, String> env = ImmutableMap.of();
ProxyHelper helper = new ProxyHelper(env);
proxy = helper.createProxyIfNeeded(new URL("https://www.something.com"));
assertEquals(Proxy.NO_PROXY, proxy);
}
@Test
public void testProxyDefaultPort() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://my.example.com");
assertEquals(Proxy.Type.HTTP, proxy.type());
assertThat(proxy.toString()).endsWith(":80");
assertEquals("my.example.com", System.getProperty("http.proxyHost"));
assertEquals("80", System.getProperty("http.proxyPort"));
proxy = ProxyHelper.createProxy("https://my.example.com");
assertThat(proxy.toString()).endsWith(":443");
assertEquals("my.example.com", System.getProperty("https.proxyHost"));
assertEquals("443", System.getProperty("https.proxyPort"));
}
@Test
public void testProxyExplicitPort() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://my.example.com:12345");
assertThat(proxy.toString()).endsWith(":12345");
assertEquals("my.example.com", System.getProperty("http.proxyHost"));
assertEquals("12345", System.getProperty("http.proxyPort"));
proxy = ProxyHelper.createProxy("https://my.example.com:12345");
assertThat(proxy.toString()).endsWith(":12345");
assertEquals("my.example.com", System.getProperty("https.proxyHost"));
assertEquals("12345", System.getProperty("https.proxyPort"));
}
@Test
public void testProxyNoProtocol() throws Exception {
try {
ProxyHelper.createProxy("my.example.com");
fail("Expected protocol error");
} catch (IOException e) {
assertThat(e.getMessage()).contains("Proxy address my.example.com is not a valid URL");
}
}
@Test
public void testProxyNoProtocolWithPort() throws Exception {
try {
ProxyHelper.createProxy("my.example.com:12345");
fail("Expected protocol error");
} catch (IOException e) {
assertThat(e.getMessage()).contains("Proxy address my.example.com:12345 is not a valid URL");
}
}
@Test
public void testProxyPortParsingError() throws Exception {
try {
ProxyHelper.createProxy("http://my.example.com:foo");
fail("Should have thrown an error for invalid port");
} catch (IOException e) {
assertThat(e.getMessage())
.contains("Proxy address http://my.example.com:foo is not a valid URL");
}
}
@Test
public void testProxyAuth() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://foo:barbaz@my.example.com");
assertEquals(Proxy.Type.HTTP, proxy.type());
assertThat(proxy.toString()).endsWith(":80");
assertEquals("my.example.com", System.getProperty("http.proxyHost"));
assertEquals("80", System.getProperty("http.proxyPort"));
assertEquals("foo", System.getProperty("http.proxyUser"));
assertEquals("barbaz", System.getProperty("http.proxyPassword"));
proxy = ProxyHelper.createProxy("https://biz:bat@my.example.com");
assertThat(proxy.toString()).endsWith(":443");
assertEquals("my.example.com", System.getProperty("https.proxyHost"));
assertEquals("443", System.getProperty("https.proxyPort"));
assertEquals("biz", System.getProperty("https.proxyUser"));
assertEquals("bat", System.getProperty("https.proxyPassword"));
}
@Test
public void testEncodedProxyAuth() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://foo:b%40rb%40z@my.example.com");
assertEquals(Proxy.Type.HTTP, proxy.type());
assertThat(proxy.toString()).endsWith(":80");
assertEquals("my.example.com", System.getProperty("http.proxyHost"));
assertEquals("80", System.getProperty("http.proxyPort"));
assertEquals("foo", System.getProperty("http.proxyUser"));
assertEquals("b@rb@z", System.getProperty("http.proxyPassword"));
}
@Test
public void testInvalidAuth() throws Exception {
try {
ProxyHelper.createProxy("http://foo@my.example.com");
fail("Should have thrown an error for invalid auth");
} catch (IOException e) {
assertThat(e.getMessage()).contains("No password given for proxy");
}
}
@Test
public void testNoProxyAuth() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://localhost:3128/");
assertEquals(Proxy.Type.HTTP, proxy.type());
assertThat(proxy.toString()).endsWith(":3128");
assertEquals("localhost", System.getProperty("http.proxyHost"));
assertEquals("3128", System.getProperty("http.proxyPort"));
}
@Test
public void testTrailingSlash() throws Exception {
Proxy proxy = ProxyHelper.createProxy("http://foo:bar@example.com:8000/");
assertEquals(Proxy.Type.HTTP, proxy.type());
assertThat(proxy.toString()).endsWith(":8000");
assertEquals("example.com", System.getProperty("http.proxyHost"));
assertEquals("8000", System.getProperty("http.proxyPort"));
assertEquals("foo", System.getProperty("http.proxyUser"));
assertEquals("bar", System.getProperty("http.proxyPassword"));
}
}