Dynamic port allocation

Author: Alexander Andryashin

pom.xml

@@ -257,6 +257,14 @@
             <artifactId>selenium-api</artifactId>
             <version>2.0b3</version>
         </dependency>
+
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.9</version>
+            <scope>test</scope>
+        </dependency>
+
     </dependencies>
 
     <reporting>

src/main/java/org/browsermob/proxy/ProxyServer.java

@@ -8,6 +8,7 @@
 import org.browsermob.core.har.HarPage;
 import org.browsermob.proxy.http.BrowserMobHttpClient;
 import org.browsermob.proxy.jetty.http.HttpContext;
+import org.browsermob.proxy.jetty.http.HttpListener;
 import org.browsermob.proxy.jetty.http.SocketListener;
 import org.browsermob.proxy.jetty.jetty.Server;
 import org.browsermob.proxy.jetty.util.InetAddrPort;
@@ -40,7 +41,8 @@ public void start() throws Exception {
         }
 
         server = new Server();
-        server.addListener(new SocketListener(new InetAddrPort(getPort()))); // todo: arg?
+        HttpListener listener = new SocketListener(new InetAddrPort(getPort()));
+        server.addListener(listener);
         HttpContext context = new HttpContext();
         context.setContextPath("/");
         server.addContext(context);
@@ -56,6 +58,8 @@ public void start() throws Exception {
         context.addHandler(handler);
 
         server.start();
+
+        setPort(listener.getPort());
     }
 
     public org.openqa.selenium.Proxy seleniumProxy() {

src/test/java/org/browsermob/proxy/ProxyServerTest.java

@@ -0,0 +1,31 @@
+package org.browsermob.proxy;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExternalResource;
+
+import static org.hamcrest.CoreMatchers.equalTo;
+import static org.hamcrest.CoreMatchers.not;
+import static org.junit.Assert.assertThat;
+
+public class ProxyServerTest {
+
+    private ProxyServer server = new ProxyServer(0);
+
+    @Before
+    public void startServer() throws Exception {
+        server.start();
+    }
+
+    @After
+    public void stopServer() throws Exception {
+        server.stop();
+    }
+
+    @Test
+    public void portAllocation() throws Exception {
+        assertThat(server.getPort(), not(equalTo(0)));
+    }
+}