QPID-2662: Use actual SocketAddress instead of the String representation

Applied patch from Andrew Kennedy <andrew.international@gmail.com>


git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk/qpid@955617 13f79535-47bb-0310-9956-ffa450edef68

2 files changed, 31 insertions, 41 deletions

diff --git a/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallConfigurationTest.java b/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallConfigurationTest.java
index e688114461..ab8957e7ef 100644
--- a/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallConfigurationTest.java
+++ b/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallConfigurationTest.java
@@ -24,33 +24,16 @@ import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.io.RandomAccessFile;
+import java.net.InetSocketAddress;
 
-import junit.framework.TestCase;
-
-import org.apache.qpid.server.protocol.AMQProtocolEngine;
-import org.apache.qpid.server.protocol.AMQProtocolSession;
 import org.apache.qpid.server.registry.ApplicationRegistry;
 import org.apache.qpid.server.registry.ConfigurationFileApplicationRegistry;
+import org.apache.qpid.server.util.InternalBrokerBaseCase;
 import org.apache.qpid.server.virtualhost.VirtualHost;
 import org.apache.qpid.server.virtualhost.VirtualHostRegistry;
-import org.apache.qpid.transport.TestNetworkDriver;
 
-public class FirewallConfigurationTest extends TestCase
+public class FirewallConfigurationTest extends InternalBrokerBaseCase
 {
-    @Override
-    public void setUp()
-    {
-        //Highlight that this test will cause a new AR to be created
-        //ApplicationRegistry.getInstance();
-    }
-
-    @Override
-    public void tearDown() throws Exception
-    {
-        //Correctly Close the AR we created
-        //ApplicationRegistry.remove();
-    }
-
     public void testFirewallConfiguration() throws Exception
     {
         // Write out config
@@ -65,8 +48,8 @@ public class FirewallConfigurationTest extends TestCase
             ApplicationRegistry.initialise(reg, 1);
 
             // Test config
-            assertFalse(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
-            assertTrue(reg.getSecurityManager().accessVirtualhost("test", "127.1.2.3"));
+            assertFalse(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
+            assertTrue(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.1.2.3", 65535)));
         }
         finally
         {
@@ -94,6 +77,7 @@ public class FirewallConfigurationTest extends TestCase
         out = new FileWriter(fileA);
         out.write("<broker>\n");
         out.write("\t<plugin-directory>${QPID_HOME}/lib/plugins</plugin-directory>\n");
+        out.write("\t<cache-directory>${QPID_WORK}/cache</cache-directory>\n");
         out.write("\t<management><enabled>false</enabled></management>\n");
         out.write("\t<security>\n");
         out.write("\t\t<principal-databases>\n");
@@ -137,7 +121,7 @@ public class FirewallConfigurationTest extends TestCase
             ApplicationRegistry.initialise(reg, 1);
 
             // Test config
-            assertFalse(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertFalse(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
         }
         finally
         {
@@ -160,14 +144,14 @@ public class FirewallConfigurationTest extends TestCase
             ApplicationRegistry.initialise(reg, 1);
 
             // Test config
-            assertFalse(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertFalse(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
 
             // Switch to deny the connection
             writeConfigFile(mainFile, true);
 
             reg.getConfiguration().reparseConfigFileSecuritySections();
 
-            assertTrue(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertTrue(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
         }
         finally
         {
@@ -238,7 +222,7 @@ public class FirewallConfigurationTest extends TestCase
             ApplicationRegistry.initialise(reg, 1);
 
             // Test config
-            assertFalse(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertFalse(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
 
             RandomAccessFile fileBRandom = new RandomAccessFile(fileB, "rw");
             fileBRandom.setLength(0);
@@ -253,7 +237,7 @@ public class FirewallConfigurationTest extends TestCase
 
             reg.getConfiguration().reparseConfigFileSecuritySections();
 
-            assertTrue(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertTrue(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
 
             fileBRandom = new RandomAccessFile(fileB, "rw");
             fileBRandom.setLength(0);
@@ -268,7 +252,7 @@ public class FirewallConfigurationTest extends TestCase
 
             reg.getConfiguration().reparseConfigFileSecuritySections();
 
-            assertFalse(reg.getSecurityManager().accessVirtualhost("test", "127.0.0.1"));
+            assertFalse(reg.getSecurityManager().accessVirtualhost("test", new InetSocketAddress("127.0.0.1", 65535)));
         }
         finally
         {
diff --git a/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallPluginTest.java b/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallPluginTest.java
index 89dba035e4..2b04962c89 100644
--- a/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallPluginTest.java
+++ b/java/broker-plugins/firewall/src/test/java/org/apache/qpid/server/security/access/FirewallPluginTest.java
@@ -23,11 +23,10 @@ import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.net.InetSocketAddress;
-
+import java.net.SocketAddress;
 
 import org.apache.commons.configuration.ConfigurationException;
 import org.apache.commons.configuration.XMLConfiguration;
-import org.apache.qpid.server.configuration.ServerConfiguration;
 import org.apache.qpid.server.security.Result;
 import org.apache.qpid.server.security.access.plugins.Firewall;
 import org.apache.qpid.server.security.access.plugins.FirewallConfiguration;
@@ -73,8 +72,15 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
     }
 
     // IP address
-    private String _address= "127.0.0.1";
+    private SocketAddress _address;
 
+    @Override
+    public void setUp() throws Exception
+    {
+        super.setUp();
+        
+        _address = new InetSocketAddress("127.0.0.1", 65535);
+    }
 
     private Firewall initialisePlugin(String defaultAction, RuleInfo[] rules) throws IOException, ConfigurationException
     {
@@ -139,7 +145,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
     
@@ -154,7 +160,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
 
@@ -167,7 +173,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         Firewall plugin = initialisePlugin("deny", new RuleInfo[]{rule});
 
         // Set IP so that we're connected from the right address
-        _address = "127.0.0.1";
+        _address = new InetSocketAddress("127.0.0.1", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
 
@@ -180,7 +186,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         Firewall plugin = initialisePlugin("deny", new RuleInfo[]{rule});
 
         // Set IP so that we're connected from the right address
-        _address = "127.0.0.1";
+        _address = new InetSocketAddress("127.0.0.1", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
     
@@ -203,7 +209,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
     
@@ -226,7 +232,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
 
@@ -240,7 +246,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
     
@@ -254,7 +260,7 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "192.168.23.23";
+        _address = new InetSocketAddress("192.168.23.23", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
     
@@ -266,11 +272,11 @@ public class FirewallPluginTest extends InternalBrokerBaseCase
         Firewall plugin = initialisePlugin("deny", new RuleInfo[]{firstRule});
         
         // Set IP so that we're connected from the right address
-        _address = "10.0.0.1";
+        _address = new InetSocketAddress("10.0.0.1", 65535);
         assertEquals(Result.DENIED, plugin.access(ObjectType.VIRTUALHOST, _address));
         
         // Set IP so that we're connected from the right address
-        _address = "127.0.0.1";
+        _address = new InetSocketAddress("127.0.0.1", 65535);
         assertEquals(Result.ALLOWED, plugin.access(ObjectType.VIRTUALHOST, _address));
     }
 }