[jboss-remoting-commits] JBoss Remoting SVN: r5466 - remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease.

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

Author: ron.sigal at jboss.com
Date: 2009-09-13 11:51:06 -0400 (Sun, 13 Sep 2009)
New Revision: 5466

Modified:
   remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/UniquePingerTaskTestCase.java
Log:
JBREM-794, JBREM-1154:  Causing LeasePinger to throw exception necessitated changes in this old unit test.

Modified: remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/UniquePingerTaskTestCase.java
===================================================================
--- remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/UniquePingerTaskTestCase.java	2009-09-12 23:19:12 UTC (rev 5465)
+++ remoting2/branches/2.2/src/tests/org/jboss/test/remoting/lease/UniquePingerTaskTestCase.java	2009-09-13 15:51:06 UTC (rev 5466)
@@ -22,16 +22,29 @@
 package org.jboss.test.remoting.lease;
 
 import java.lang.reflect.Field;
+import java.lang.reflect.Method;
+import java.net.InetAddress;
+import java.util.HashMap;
 import java.util.TimerTask;
 
+import javax.management.MBeanServer;
+
 import junit.framework.TestCase;
 
 import org.apache.log4j.ConsoleAppender;
 import org.apache.log4j.Level;
 import org.apache.log4j.Logger;
 import org.apache.log4j.PatternLayout;
+import org.jboss.remoting.Client;
+import org.jboss.remoting.ConnectionListener;
+import org.jboss.remoting.InvocationRequest;
 import org.jboss.remoting.InvokerLocator;
 import org.jboss.remoting.LeasePinger;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+import org.jboss.remoting.transport.PortUtil;
 import org.jboss.remoting.transport.socket.SocketClientInvoker;
 
 
@@ -48,6 +61,13 @@
 {
    private static Logger log = Logger.getLogger(UniquePingerTaskTestCase.class);
    private static boolean firstTime = true;
+   
+   protected String host;
+   protected int port;
+   protected String locatorURI;
+   protected InvokerLocator serverLocator;
+   protected Connector connector;
+   protected TestInvocationHandler invocationHandler;
 
    
    /**
@@ -80,10 +100,14 @@
    {
       log.info("entering " + getName());
       
-      InvokerLocator locator = new InvokerLocator("socket://localhost:8888");
-      SocketClientInvoker clientInvoker = new SocketClientInvoker(locator);
+      setupServer();
+      SocketClientInvoker clientInvoker = new SocketClientInvoker(serverLocator);
+      clientInvoker.connect();
       LeasePinger leasePinger = new LeasePinger(clientInvoker, "abc", 2000);
       log.info("expect WARN \"failed to ping to server\"");
+      Method method = LeasePinger.class.getDeclaredMethod("setLeasePingerId", new Class[]{String.class});
+      method.setAccessible(true);
+      method.invoke(leasePinger, new Object[]{"dummyID"});
       leasePinger.addClient("def", null, 1000);
       
       // Verify LeasePingerTimerTask has not been created.
@@ -103,4 +127,55 @@
       
       log.info(getName() + " PASSES");
    }
+   
+   protected void setupServer() throws Exception
+   {
+      host = InetAddress.getLocalHost().getHostAddress();
+      port = PortUtil.findFreePort(host);
+      locatorURI = "socket://" + host + ":" + port;
+      String metadata = System.getProperty("remoting.metadata");
+      if (metadata != null)
+      {
+         locatorURI += "/?" + metadata;
+      }
+      serverLocator = new InvokerLocator(locatorURI);
+      log.info("Starting remoting server with locator uri of: " + locatorURI);
+      HashMap config = new HashMap();
+      config.put(InvokerLocator.FORCE_REMOTE, "true");
+      config.put("leasePeriod", "1000");
+      connector = new Connector(serverLocator, config);
+      connector.create();
+      invocationHandler = new TestInvocationHandler();
+      connector.addInvocationHandler("test", invocationHandler);
+      connector.addConnectionListener(new TestConnectionListener());
+      connector.start();
+   }
+   
+   
+   protected void shutdownServer() throws Exception
+   {
+      if (connector != null)
+         connector.stop();
+   }
+   
+   
+   static class TestInvocationHandler implements ServerInvocationHandler
+   {
+      public void addListener(InvokerCallbackHandler callbackHandler) {}
+      public Object invoke(final InvocationRequest invocation) throws Throwable
+      {
+         return invocation.getParameter();
+      }
+      public void removeListener(InvokerCallbackHandler callbackHandler) {}
+      public void setMBeanServer(MBeanServer server) {}
+      public void setInvoker(ServerInvoker invoker) {}
+   }
+   
+   
+   static class TestConnectionListener implements ConnectionListener
+   {
+      public void handleConnectionException(Throwable throwable, Client client)
+      {
+      }  
+   }
 }
\ No newline at end of file