adamfisk#131 Fixed FTEMain

asashour · Jan 31, 2014 · 70c3f81 · 70c3f81
1 parent e5d3ada
commit 70c3f81
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/test/java/org/littleshoot/proxy/ChainedMain.java b/src/test/java/org/littleshoot/proxy/ChainedMain.java
@@ -58,7 +58,7 @@ public void lookupChainedProxies(HttpRequest httpRequest,
                             @Override
                             public InetSocketAddress getChainedProxyAddress() {
                                 return new InetSocketAddress(LOCALHOST,
-                                        LITTLEPROXY_UPSTREAM_PORT);
+                                        getUpstreamPort());
                             }
 
                             @Override
@@ -86,5 +86,9 @@ public SSLEngine newSslEngine() {
                 .start();
 
     }
+
+    protected int getUpstreamPort() {
+        return LITTLEPROXY_UPSTREAM_PORT;
+    }
 
 }
diff --git a/src/test/java/org/littleshoot/proxy/FTEMain.java b/src/test/java/org/littleshoot/proxy/FTEMain.java
@@ -45,6 +45,11 @@ public void run() throws Exception {
                 "--proxy_port", LITTLEPROXY_UPSTREAM_PORT);
     }
 
+    @Override
+    protected int getUpstreamPort() {
+        return FTEPROXY_CLIENT_PORT;
+    }
+
     private void fteProxy(Object... args) throws Exception {
         Executor cmdExec = new DefaultExecutor();
         cmdExec.setStreamHandler(new PumpStreamHandler(System.out, System.err,