[jboss-remoting-commits] JBoss Remoting SVN: r4236 - remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security.

[jboss-remoting-commits at lists.jboss.org]

Author: ron.sigal at jboss.com
Date: 2008-05-23 01:42:40 -0400 (Fri, 23 May 2008)
New Revision: 4236

Modified:
   remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/NetworkRegistryProxyTestCase.java
   remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/TestNetworkRegistry.java
Log:
JBREM-977: Added filter to network registry.

Modified: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/NetworkRegistryProxyTestCase.java
===================================================================
--- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/NetworkRegistryProxyTestCase.java	2008-05-23 05:19:53 UTC (rev 4235)
+++ remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/NetworkRegistryProxyTestCase.java	2008-05-23 05:42:40 UTC (rev 4236)
@@ -100,7 +100,8 @@
       
       // Start registry.
       MBeanServer server = MBeanServerFactory.createMBeanServer();
-      TestNetworkRegistry networkRegistry = new TestNetworkRegistry();
+      InvokerLocator locator = new InvokerLocator(locatorURI);
+      TestNetworkRegistry networkRegistry = new TestNetworkRegistry(locator);
       ObjectName name = new ObjectName("test:type=TestNetworkRegistry");
       server.registerMBean(networkRegistry, name);
       

Modified: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/TestNetworkRegistry.java
===================================================================
--- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/TestNetworkRegistry.java	2008-05-23 05:19:53 UTC (rev 4235)
+++ remoting2/branches/2.x/src/tests/org/jboss/test/remoting/security/TestNetworkRegistry.java	2008-05-23 05:42:40 UTC (rev 4236)
@@ -23,6 +23,7 @@
 package org.jboss.test.remoting.security;
 
 import org.apache.log4j.Logger;
+import org.jboss.remoting.InvokerLocator;
 import org.jboss.remoting.detection.ServerInvokerMetadata;
 import org.jboss.remoting.ident.Identity;
 import org.jboss.remoting.network.NetworkRegistry;
@@ -39,11 +40,24 @@
    static Logger log = Logger.getLogger(TestNetworkRegistry.class);
    
    int counter;
+   InvokerLocator locator;
    
+   public TestNetworkRegistry(InvokerLocator locator)
+   {
+      this.locator = locator;
+   }
+   
    public void addServer(Identity identity, ServerInvokerMetadata[] invokers)
    {
-      counter++;
       log.info("addServer() called");
+      for (int i = 0; i < invokers.length; i++)
+      {
+         if (locator.isSameEndpoint(invokers[i].getInvokerLocator()))
+         {
+            counter++;
+            log.info("addServer(): counter incremented");
+         }
+      }
    }
    
    public int getCounter()