1 files changed, 94 insertions, 0 deletions
diff --git a/src/test/java/com/btr/proxy/selector/misc/ProxyListFallbackSelectorTest.java b/src/test/java/com/btr/proxy/selector/misc/ProxyListFallbackSelectorTest.java
new file mode 100644
index 0000000..12c172c
--- /dev/null
+++ b/src/test/java/com/btr/proxy/selector/misc/ProxyListFallbackSelectorTest.java
@@ -0,0 +1,94 @@
+package com.btr.proxy.selector.misc;
+
+import static org.junit.Assert.*;
+
+import java.io.IOException;
+import java.net.Proxy;
+import java.net.ProxySelector;
+import java.net.SocketAddress;
+import java.net.URI;
+import java.util.Arrays;
+import java.util.List;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import com.btr.proxy.TestUtil;
+
+/*****************************************************************************
+ *  Unit Tests for the ProxyListFallbackSelector
+ *  
+ * @author Bernd Rosstauscher (proxyvole@rosstauscher.de) Copyright 2011
+ ****************************************************************************/
+
+public class ProxyListFallbackSelectorTest {
+
+	private ProxyListFallbackSelector selector;
+
+	/*************************************************************************
+	 * Setup before tests.
+	 ************************************************************************/
+	@Before
+	public void setup() {
+		this.selector = new ProxyListFallbackSelector(
+		new ProxySelector() {
+			@Override
+			public List<Proxy> select(URI uri) {
+				return Arrays.asList(TestUtil.HTTP_TEST_PROXY, TestUtil.HTTPS_TEST_PROXY, Proxy.NO_PROXY);
+			}
+			@Override
+			public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
+				// Not used on the delegate
+			}
+		});
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testList() {
+		List<Proxy> result = this.selector.select(TestUtil.HTTP_TEST_URI);
+		assertEquals(TestUtil.HTTP_TEST_PROXY, result.get(0));
+		assertEquals(TestUtil.HTTPS_TEST_PROXY, result.get(1));
+		assertEquals(Proxy.NO_PROXY, result.get(2));
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testFailedProxy() {
+		this.selector.connectFailed(
+				TestUtil.HTTP_TEST_URI, 
+				TestUtil.HTTP_TEST_PROXY.address(), 
+				new IOException("TEST"));
+		
+		List<Proxy> result = this.selector.select(TestUtil.HTTP_TEST_URI);
+		
+		assertEquals(TestUtil.HTTPS_TEST_PROXY, result.get(0));
+		assertEquals(Proxy.NO_PROXY, result.get(1));
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 * @throws InterruptedException if the test wait period was interrupted
+	 ************************************************************************/
+	@Test
+	public void testFailedProxyRetry() throws InterruptedException {
+		this.selector.setRetryAfterMs(100);
+		this.selector.connectFailed(
+				TestUtil.HTTP_TEST_URI, 
+				TestUtil.HTTP_TEST_PROXY.address(), 
+				new IOException("TEST"));
+		
+		List<Proxy> result = this.selector.select(TestUtil.HTTP_TEST_URI);
+		assertEquals(2, result.size());
+		
+		Thread.sleep(200);
+		result = this.selector.select(TestUtil.HTTP_TEST_URI);
+		assertEquals(3, result.size());
+	}
+
+}
+