[jboss-cvs] JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/server ...
Tom Elrod
tom.elrod at jboss.com
Thu Nov 9 16:35:23 EST 2006
User: telrod
Date: 06/11/09 16:35:23
Added: src/tests/org/jboss/test/remoting/callback/multiple/server
CallbackTestCase.java CallbackTestClient.java
CallbackTestServer.java
Log:
JBREM-622 & JBREM-629 - sync with remoting_2_x branch to pull in bug fixes.
Revision Changes Path
1.2 +83 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/server/CallbackTestCase.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestCase.java
===================================================================
RCS file: CallbackTestCase.java
diff -N CallbackTestCase.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestCase.java 9 Nov 2006 21:35:23 -0000 1.2
@@ -0,0 +1,83 @@
+/*
+* JBoss, a division of Red Hat
+* Copyright 2006, Red Hat Middleware, LLC, and individual contributors as indicated
+* by the @authors tag. See the copyright.txt in the distribution for a
+* full listing of individual contributors.
+*
+* This is free software; you can redistribute it and/or modify it
+* under the terms of the GNU Lesser General Public License as
+* published by the Free Software Foundation; either version 2.1 of
+* the License, or (at your option) any later version.
+*
+* This software is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+* Lesser General Public License for more details.
+*
+* You should have received a copy of the GNU Lesser General Public
+* License along with this software; if not, write to the Free
+* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+*/
+package org.jboss.test.remoting.callback.multiple.server;
+
+import org.apache.log4j.Level;
+import org.jboss.jrunit.harness.TestDriver;
+
+/**
+ * @author <a href="mailto:tom at jboss.org">Tom Elrod</a>
+ */
+public class CallbackTestCase extends TestDriver
+{
+ public void declareTestClasses()
+ {
+ addTestClasses(CallbackTestClient.class.getName(),
+ 1,
+ CallbackTestServer.class.getName());
+ }
+
+ protected Level getTestHarnessLogLevel()
+ {
+ return Level.DEBUG;
+ }
+
+ protected Level getTestLogLevel()
+ {
+ return Level.DEBUG;
+ }
+
+ /**
+ * How long to wait for test results to be returned from the client(s). If goes longer than the
+ * specified limit, will throw an exception and kill the running test cases. Default value is
+ * RESULTS_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getResultsTimeout()
+ {
+ return 600000;
+ }
+
+ /**
+ * How long for the server test case to wait for tear down message. If exceeds timeout,
+ * will throw exception. The default value is TEARDOWN_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getTearDownTimeout()
+ {
+ return 600000;
+ }
+
+ /**
+ * How long to allow each of the test cases to run their tests. If exceeds this timeout
+ * will throw exception and kill tests. The default value is RUN_TEST_TIMEOUT.
+ *
+ * @return
+ */
+ protected long getRunTestTimeout()
+ {
+ return 600000;
+ }
+
+}
\ No newline at end of file
1.2 +238 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/server/CallbackTestClient.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestClient.java
===================================================================
RCS file: CallbackTestClient.java
diff -N CallbackTestClient.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestClient.java 9 Nov 2006 21:35:23 -0000 1.2
@@ -0,0 +1,238 @@
+package org.jboss.test.remoting.callback.multiple.server;
+
+import junit.framework.TestCase;
+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;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.Callback;
+import org.jboss.remoting.callback.HandleCallbackException;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+
+import javax.management.MBeanServer;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+/**
+ * @author <a href="mailto:telrod at e2technologies.net">Tom Elrod</a>
+ */
+public class CallbackTestClient extends TestCase
+{
+ // Default locator values
+ private static String transport = "socket";
+ private static String host = "localhost";
+ private static int port = 5500;
+
+ private String locatorURI = transport + "://" + host + ":" + port;
+
+ private Connector connector;
+ private Connector connector2;
+
+ private static final Logger log = Logger.getLogger(CallbackTestClient.class);
+
+ public static final String RESPONSE_VALUE = "This is the return to SampleInvocationHandler invocation";
+
+
+ public void tearDown() throws Exception
+ {
+ if (connector != null)
+ {
+ connector.stop();
+ connector.destroy();
+ }
+ if (connector2 != null)
+ {
+ connector2.stop();
+ connector2.destroy();
+ }
+ }
+
+ public void testPushCallback() throws Throwable
+ {
+
+ CallbackTestClient.CallbackHandler callbackHandler = new CallbackTestClient.CallbackHandler();
+
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ Client remotingClient = new Client(locator);
+ remotingClient.connect();
+
+ try
+ {
+
+ // Using loctor with port value one higher than the target server
+ String callbackLocatorURI = transport + "://" + host + ":" + (port + 5);
+ InvokerLocator callbackLocator = new InvokerLocator(callbackLocatorURI);
+
+ String callbackLocatorURI2 = transport + "://" + host + ":" + (port + 10);
+ InvokerLocator callbackLocator2 = new InvokerLocator(callbackLocatorURI2);
+
+ log.info("testPushCallback - Setting up server.");
+ // call to create remoting server to
+ // receive client callbacks.
+ connector = new Connector();
+ connector.setInvokerLocator(callbackLocator.getLocatorURI());
+ connector.start();
+
+ CallbackTestClient.SampleInvocationHandler invocationHandler = new CallbackTestClient.SampleInvocationHandler();
+ connector.addInvocationHandler("sample", invocationHandler);
+
+ connector2 = new Connector();
+ connector2.setInvokerLocator(callbackLocator2.getLocatorURI());
+ connector2.start();
+
+ connector.addInvocationHandler("sample", invocationHandler);
+
+ // Callback handle object will be passed back as part of the callback object
+ String callbackHandleObject = "myCallbackHandleObject";
+ log.info("testPushCallback - adding listener.");
+ // by passing only the callback handler, will indicate pull callbacks
+ remotingClient.addListener(callbackHandler, callbackLocator, callbackHandleObject);
+ remotingClient.addListener(callbackHandler, callbackLocator2, callbackHandleObject);
+ log.info("testPushCallback - make invocation");
+ // now make invocation on server, which should cause a callback to happen
+ Object response = remotingClient.invoke("Do something", null);
+
+ // need to wait for brief moment so server can callback
+ Thread.sleep(5000);
+
+ int count = callbackHandler.getCallbackCount();
+ System.out.println("callback count = " + count);
+ assertEquals(2, count);
+ }
+ finally
+ {
+ if (remotingClient != null)
+ {
+ // remove callback handler from server
+ remotingClient.removeListener(callbackHandler);
+ remotingClient.disconnect();
+ }
+ }
+
+
+ }
+
+ /**
+ * Simple invocation handler implementation.
+ */
+ public static class SampleInvocationHandler implements ServerInvocationHandler
+ {
+
+ private List listeners = new ArrayList();
+
+
+ /**
+ * called to handle a specific invocation
+ *
+ * @param invocation
+ * @return
+ * @throws Throwable
+ */
+ public Object invoke(InvocationRequest invocation) throws Throwable
+ {
+ // Just going to return static string as this is just simple example code.
+
+ // Will also fire callback to listeners if they were to exist using
+ // simple invocation request.
+ Callback callback = new Callback("This is the payload of callback invocation.");
+ Iterator itr = listeners.iterator();
+ while (itr.hasNext())
+ {
+ InvokerCallbackHandler callbackHandler = (InvokerCallbackHandler) itr.next();
+ try
+ {
+ callbackHandler.handleCallback(callback);
+ }
+ catch (HandleCallbackException e)
+ {
+ e.printStackTrace();
+ }
+ }
+
+ return RESPONSE_VALUE;
+
+ }
+
+ /**
+ * Adds a callback handler that will listen for callbacks from
+ * the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void addListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.add(callbackHandler);
+ }
+
+ /**
+ * Removes the callback handler that was listening for callbacks
+ * from the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void removeListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.remove(callbackHandler);
+ }
+
+ /**
+ * set the mbean server that the handler can reference
+ *
+ * @param server
+ */
+ public void setMBeanServer(MBeanServer server)
+ {
+ // NO OP as do not need reference to MBeanServer for this handler
+ }
+
+ /**
+ * set the invoker that owns this handler
+ *
+ * @param invoker
+ */
+ public void setInvoker(ServerInvoker invoker)
+ {
+ // NO OP as do not need reference back to the server invoker
+ }
+
+ }
+
+
+ public class CallbackHandler implements InvokerCallbackHandler
+ {
+ private Callback callback;
+ private int callbackNumber = 0;
+
+ /**
+ * Will take the callback message and send back to client.
+ * If client locator is null, will store them till client polls to get them.
+ *
+ * @param callback
+ * @throws org.jboss.remoting.callback.HandleCallbackException
+ *
+ */
+ public void handleCallback(Callback callback) throws HandleCallbackException
+ {
+ callbackNumber++;
+ this.callback = callback;
+ System.out.println("Received callback value of: " + callback.getCallbackObject());
+ System.out.println("Received callback handle object of: " + callback.getCallbackHandleObject());
+ System.out.println("Received callback server invoker of: " + callback.getServerLocator());
+ }
+
+ public Callback getCallback()
+ {
+ return callback;
+ }
+
+ public int getCallbackCount()
+ {
+ return callbackNumber;
+ }
+ }
+
+}
1.2 +163 -0 JBossRemoting/src/tests/org/jboss/test/remoting/callback/multiple/server/CallbackTestServer.java
(In the diff below, changes in quantity of whitespace are not shown.)
Index: CallbackTestServer.java
===================================================================
RCS file: CallbackTestServer.java
diff -N CallbackTestServer.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ CallbackTestServer.java 9 Nov 2006 21:35:23 -0000 1.2
@@ -0,0 +1,163 @@
+package org.jboss.test.remoting.callback.multiple.server;
+
+import org.apache.log4j.Level;
+import org.jboss.jrunit.extensions.ServerTestCase;
+import org.jboss.remoting.InvocationRequest;
+import org.jboss.remoting.InvokerLocator;
+import org.jboss.remoting.ServerInvocationHandler;
+import org.jboss.remoting.ServerInvoker;
+import org.jboss.remoting.callback.Callback;
+import org.jboss.remoting.callback.InvokerCallbackHandler;
+import org.jboss.remoting.transport.Connector;
+
+import javax.management.MBeanServer;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+/**
+ * Simple remoting server. Uses inner class SampleInvocationHandler
+ * as the invocation target handler class.
+ *
+ * @author <a href="mailto:telrod at e2technologies.net">Tom Elrod</a>
+ */
+public class CallbackTestServer extends ServerTestCase
+{
+ // Default locator values
+ private static String transport = "socket";
+ private static String host = "localhost";
+ private static int port = 5500;
+
+ private String locatorURI = transport + "://" + host + ":" + port;
+ private Connector connector;
+
+ // String to be returned from invocation handler upon client invocation calls.
+ public static final String RESPONSE_VALUE = "This is the return to SampleInvocationHandler invocation";
+ public static final String CALLBACK_VALUE = "This is the payload of callback invocation.";
+
+
+ public void setupServer() throws Exception
+ {
+ InvokerLocator locator = new InvokerLocator(locatorURI);
+ System.out.println("Starting remoting server with locator uri of: " + locatorURI);
+ connector = new Connector();
+ connector.setInvokerLocator(locator.getLocatorURI());
+ connector.start();
+
+ CallbackTestServer.SampleInvocationHandler invocationHandler = new CallbackTestServer.SampleInvocationHandler();
+ // first parameter is sub-system name. can be any String value.
+ connector.addInvocationHandler("sample", invocationHandler);
+ }
+
+ protected void setUp() throws Exception
+ {
+ org.apache.log4j.BasicConfigurator.configure();
+ org.apache.log4j.Category.getRoot().setLevel(Level.DEBUG);
+ org.apache.log4j.Category.getInstance("org.jboss.remoting").setLevel(Level.DEBUG);
+ org.apache.log4j.Category.getInstance("org.jgroups").setLevel(Level.FATAL);
+
+ setupServer();
+ }
+
+ protected void tearDown() throws Exception
+ {
+ if(connector != null)
+ {
+ connector.stop();
+ connector.destroy();
+ }
+ }
+
+ public static void main(String[] args)
+ {
+ CallbackTestServer server = new CallbackTestServer();
+ try
+ {
+ server.setUp();
+
+ Thread.sleep(600000);
+ }
+ catch (Exception e)
+ {
+ e.printStackTrace();
+ }
+ }
+
+ /**
+ * Simple invocation handler implementation.
+ */
+ public static class SampleInvocationHandler implements ServerInvocationHandler
+ {
+
+ private List listeners = new ArrayList();
+
+
+ /**
+ * called to handle a specific invocation
+ *
+ * @param invocation
+ * @return
+ * @throws Throwable
+ */
+ public Object invoke(InvocationRequest invocation) throws Throwable
+ {
+ // Just going to return static string as this is just simple example code.
+
+ // Will also fire callback to listeners if they were to exist using
+ // simple invocation request.
+ Callback callback = new Callback(CALLBACK_VALUE);
+ Iterator itr = listeners.iterator();
+ while(itr.hasNext())
+ {
+ InvokerCallbackHandler callbackHandler = (InvokerCallbackHandler) itr.next();
+ callbackHandler.handleCallback(callback);
+ }
+
+ return RESPONSE_VALUE;
+
+ }
+
+ /**
+ * Adds a callback handler that will listen for callbacks from
+ * the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void addListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.add(callbackHandler);
+ }
+
+ /**
+ * Removes the callback handler that was listening for callbacks
+ * from the server invoker handler.
+ *
+ * @param callbackHandler
+ */
+ public void removeListener(InvokerCallbackHandler callbackHandler)
+ {
+ listeners.remove(callbackHandler);
+ }
+
+ /**
+ * set the mbean server that the handler can reference
+ *
+ * @param server
+ */
+ public void setMBeanServer(MBeanServer server)
+ {
+ // NO OP as do not need reference to MBeanServer for this handler
+ }
+
+ /**
+ * set the invoker that owns this handler
+ *
+ * @param invoker
+ */
+ public void setInvoker(ServerInvoker invoker)
+ {
+ // NO OP as do not need reference back to the server invoker
+ }
+
+ }
+}
More information about the jboss-cvs-commits
mailing list