1 files changed, 87 insertions, 0 deletions
diff --git a/src/test/java/com/btr/proxy/selector/misc/ProtocolDispatchTest.java b/src/test/java/com/btr/proxy/selector/misc/ProtocolDispatchTest.java
new file mode 100644
index 0000000..da27327
--- /dev/null
+++ b/src/test/java/com/btr/proxy/selector/misc/ProtocolDispatchTest.java
@@ -0,0 +1,87 @@
+package com.btr.proxy.selector.misc;
+
+import static org.junit.Assert.*;
+
+import java.net.Proxy;
+import java.util.List;
+
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+import com.btr.proxy.TestUtil;
+import com.btr.proxy.selector.fixed.FixedProxySelector;
+
+/*****************************************************************************
+ *  Unit Tests for the ProtocolDispatchSelector
+ *  
+ * @author Bernd Rosstauscher (proxyvole@rosstauscher.de) Copyright 2009
+ ****************************************************************************/
+
+public class ProtocolDispatchTest {
+
+	private static ProtocolDispatchSelector ps;
+
+	@BeforeClass
+	public static void setup() {
+		ps = new ProtocolDispatchSelector();
+		ps.setSelector("http", new FixedProxySelector(TestUtil.HTTP_TEST_PROXY));
+		ps.setSelector("https", new FixedProxySelector(TestUtil.HTTPS_TEST_PROXY));
+		ps.setSelector("ftp", new FixedProxySelector(TestUtil.FTP_TEST_PROXY));
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testDispatchHttp() {
+		List<Proxy> result = ps.select(TestUtil.HTTP_TEST_URI);
+		assertEquals(TestUtil.HTTP_TEST_PROXY, result.get(0));
+	}
+
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testDispatchHttps() {
+		List<Proxy> result = ps.select(TestUtil.HTTPS_TEST_URI);
+		assertEquals(TestUtil.HTTPS_TEST_PROXY, result.get(0));
+	}
+
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testDispatchFtp() {
+		List<Proxy> result = ps.select(TestUtil.FTP_TEST_URI);
+		assertEquals(TestUtil.FTP_TEST_PROXY, result.get(0));
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testRemove() {
+		ProtocolDispatchSelector px = new ProtocolDispatchSelector();
+		FixedProxySelector selector = new FixedProxySelector(TestUtil.HTTP_TEST_PROXY);
+		px.setSelector("http", selector);
+		assertEquals(selector, px.getSelector("http"));
+		px.removeSelector("http");
+		assertNull(px.getSelector("http"));
+	}
+	
+	/*************************************************************************
+	 * Test method
+	 ************************************************************************/
+	@Test
+	public void testFallback() {
+		ProtocolDispatchSelector px = new ProtocolDispatchSelector();
+		FixedProxySelector selector = new FixedProxySelector(TestUtil.HTTP_TEST_PROXY);
+		px.setFallbackSelector(selector);
+		
+		List<Proxy> proxies = px.select(TestUtil.HTTP_TEST_URI);
+		
+		assertEquals(TestUtil.HTTP_TEST_PROXY, proxies.get(0));
+	}
+
+}
+