[jboss-cvs] JBossRemoting/src/tests/org/jboss/test/remoting/callback/asynch ...

[Ron Sigal]

  User: rsigal  
  Date: 06/11/28 13:44:09

  Modified:    src/tests/org/jboss/test/remoting/callback/asynch   Tag:
                        remoting_2_x AsynchCallbackTestClient.java
                        AsynchCallbackTestServer.java
  Log:
  JBREM-640:  Added test for size of oneway threadool.
  
  Revision  Changes    Path
  No                   revision
  
  
  No                   revision
  
  
  1.1.2.2   +3 -1      JBossRemoting/src/tests/org/jboss/test/remoting/callback/asynch/AsynchCallbackTestClient.java
  
  (In the diff below, changes in quantity of whitespace are not shown.)
  
  Index: AsynchCallbackTestClient.java
  ===================================================================
  RCS file: /cvsroot/jboss/JBossRemoting/src/tests/org/jboss/test/remoting/callback/asynch/AsynchCallbackTestClient.java,v
  retrieving revision 1.1.2.1
  retrieving revision 1.1.2.2
  diff -u -b -r1.1.2.1 -r1.1.2.2
  --- AsynchCallbackTestClient.java	28 Nov 2006 17:24:57 -0000	1.1.2.1
  +++ AsynchCallbackTestClient.java	28 Nov 2006 18:44:09 -0000	1.1.2.2
  @@ -34,7 +34,7 @@
   
   /** 
    * @author <a href="ron.sigal at jboss.com">Ron Sigal</a>
  - * @version $Revision: 1.1.2.1 $
  + * @version $Revision: 1.1.2.2 $
    * <p>
    * Copyright Nov 25, 2006
    * </p>
  @@ -119,6 +119,8 @@
         Boolean done = (Boolean) client.invoke(AsynchCallbackTestServer.GET_STATUS);
         assertTrue(done.booleanValue());
         assertTrue(callbackHandler.receivedCallback);
  +      String threadCount = (String) client.invoke(AsynchCallbackTestServer.GET_THREAD_COUNT);
  +      assertEquals(AsynchCallbackTestServer.THREAD_COUNT, threadCount);
         client.invoke(AsynchCallbackTestServer.RESET);
         client.removeListener(callbackHandler);
         client.disconnect();
  
  
  
  1.1.2.2   +31 -3     JBossRemoting/src/tests/org/jboss/test/remoting/callback/asynch/AsynchCallbackTestServer.java
  
  (In the diff below, changes in quantity of whitespace are not shown.)
  
  Index: AsynchCallbackTestServer.java
  ===================================================================
  RCS file: /cvsroot/jboss/JBossRemoting/src/tests/org/jboss/test/remoting/callback/asynch/AsynchCallbackTestServer.java,v
  retrieving revision 1.1.2.1
  retrieving revision 1.1.2.2
  diff -u -b -r1.1.2.1 -r1.1.2.2
  --- AsynchCallbackTestServer.java	28 Nov 2006 17:24:57 -0000	1.1.2.1
  +++ AsynchCallbackTestServer.java	28 Nov 2006 18:44:09 -0000	1.1.2.2
  @@ -21,7 +21,11 @@
   */
   package org.jboss.test.remoting.callback.asynch;
   
  +import java.lang.reflect.Field;
   import java.net.InetAddress;
  +import java.util.HashMap;
  +import java.util.Iterator;
  +import java.util.Map;
   
   import javax.management.MBeanServer;
   
  @@ -29,6 +33,7 @@
   
   import org.jboss.jrunit.extensions.ServerTestCase;
   import org.jboss.logging.Logger;
  +import org.jboss.remoting.Client;
   import org.jboss.remoting.InvocationRequest;
   import org.jboss.remoting.InvokerLocator;
   import org.jboss.remoting.ServerInvocationHandler;
  @@ -37,11 +42,14 @@
   import org.jboss.remoting.callback.Callback;
   import org.jboss.remoting.callback.HandleCallbackException;
   import org.jboss.remoting.callback.InvokerCallbackHandler;
  +import org.jboss.remoting.callback.ServerInvokerCallbackHandler;
   import org.jboss.remoting.transport.Connector;
  +import org.jboss.util.threadpool.BasicThreadPool;
  +import org.jboss.util.threadpool.ThreadPool;
   
   /** 
    * @author <a href="ron.sigal at jboss.com">Ron Sigal</a>
  - * @version $Revision: 1.1.2.1 $
  + * @version $Revision: 1.1.2.2 $
    * <p>
    * Copyright Nov 25, 2006
    * </p>
  @@ -50,16 +58,19 @@
   {
      public static String transport = "socket";
      public static int port = 5413;
  +   
      public static String SYNCHRONOUS_TEST = "synchronousTest";
      public static String ASYNCHRONOUS_SERVER_SIDE_TEST = "asynchronousServerSideTest";
      public static String ASYNCHRONOUS_CLIENT_SIDE_TEST = "asynchronousClientSideTest";
      public static String GET_STATUS = "getStatus";
      public static String RESET = "reset";
  +   public static String GET_THREAD_COUNT = "getThreadCount";
  +   public static String THREAD_COUNT = "17";
      
      private static Logger log = Logger.getLogger(AsynchCallbackTestServer.class);
      
      // remoting server connector
  -   private Connector connector;
  +   static private Connector connector;
      String serverLocatorURI;
   
      
  @@ -72,7 +83,9 @@
         String locatorURI =  transport + "://" + InetAddress.getLocalHost().getHostAddress() + ":" + port; 
         InvokerLocator serverLocator = new InvokerLocator(locatorURI);
         System.out.println("Starting remoting server with locator uri of: " + locatorURI);
  -      connector = new Connector(serverLocator);
  +      Map config = new HashMap();
  +      config.put(ServerInvoker.MAX_NUM_ONEWAY_THREADS_KEY, THREAD_COUNT);
  +      connector = new Connector(serverLocator, config);
         connector.create();
         SampleInvocationHandler invocationHandler = new SampleInvocationHandler();
         connector.addInvocationHandler("sample", invocationHandler);
  @@ -156,6 +169,21 @@
                     done = false;
                  } 
               }
  +            else if (test.equals(GET_THREAD_COUNT))
  +            {
  +               ServerInvoker invoker = connector.getServerInvoker();
  +               Field field = ServerInvoker.class.getDeclaredField("callbackHandlers");
  +               field.setAccessible(true);
  +               Map callbackHandlers = (Map) field.get(invoker);
  +               Iterator it = callbackHandlers.values().iterator();
  +               ServerInvokerCallbackHandler callbackHandler = (ServerInvokerCallbackHandler) it.next();
  +               Client callbackClient = callbackHandler.getCallbackClient();
  +               field = Client.class.getDeclaredField("onewayThreadPool");
  +               field.setAccessible(true);
  +               BasicThreadPool threadPool = (BasicThreadPool) field.get(callbackClient);
  +               int size = threadPool.getMaximumPoolSize();
  +               return Integer.toString(size);
  +            }
               else
               {
                  log.error("unrecognized test: " + test);