+ * Copyright Nov 17, 2010 + *
+ */ +public class ServerIdentityTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(ServerIdentityTestCase.c= lass); + = + private static boolean firstTime =3D true; + = + protected String host; + protected int port; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected TestInvocationHandler invocationHandler; + + = + public void setUp() throws Exception + { + if (firstTime) + { + firstTime =3D false; + Logger.getLogger("org.jboss.remoting").setLevel(XLevel.TRACE); + Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO); + String pattern =3D "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n"; + PatternLayout layout =3D new PatternLayout(pattern); + ConsoleAppender consoleAppender =3D new ConsoleAppender(layout); + Logger.getRootLogger().addAppender(consoleAppender); + } + = + host =3D InetAddress.getLocalHost().getHostAddress(); + port =3D PortUtil.findFreePort(host); + } + + = + public void tearDown() + { + } + = + = + public void testServerIdentityWithoutLeasing() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(false); + = + // Create client. Adding "dummy=3Ddummy" assures that InvokerRegistr= y will not create a + // LocalClientInvoker for the ConnectionValidator. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI + "&d= ummy=3Ddummy"); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Install connection listener. + TestConnectionListener listener =3D new TestConnectionListener(); + HashMap metadata =3D new HashMap(); + metadata.put(Remoting.USE_SERVER_CONNECTION_IDENTITY, "true"); + metadata.put(ConnectionValidator.VALIDATOR_PING_PERIOD, "10000"); + metadata.put(ConnectionValidator.VALIDATOR_PING_TIMEOUT, "10000"); + client.addConnectionListener(listener, metadata); + log.info(this + " added connection listener: " + listener); + // Allow time to get serverId of first server. + Thread.sleep(15000); + = + // Verify listener is notified if server bounces. + shutdownServer(); + log.info("SHUT DOWN SERVER"); + setupServer(false); + log.info("SET UP NEW SERVER"); + Thread.sleep(10000); + log.info(this + " listener.connectionFailed: " + listener.connection= Failed); + assertTrue(listener.connectionFailed); + = + client.disconnect(); + shutdownServer(); + = + log.info(getName() + " PASSES"); + } + = + public void testServerIdentityWithLeasing() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(true); + = + // Create client. Adding "dummy=3Ddummy" assures that InvokerRegistr= y will not create a + // LocalClientInvoker for the ConnectionValidator. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI + "&d= ummy=3Ddummy"); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Install connection listener. + TestConnectionListener listener =3D new TestConnectionListener(); + HashMap metadata =3D new HashMap(); + metadata.put(Remoting.USE_SERVER_CONNECTION_IDENTITY, "true"); + metadata.put(ConnectionValidator.VALIDATOR_PING_PERIOD, "10000"); + metadata.put(ConnectionValidator.VALIDATOR_PING_TIMEOUT, "10000"); + client.addConnectionListener(listener, metadata); + log.info(this + " added connection listener: " + listener); + // Allow time to get serverId of first server. + Thread.sleep(15000); + = + // Verify listener is notified if server bounces. + shutdownServer(); + log.info("SHUT DOWN SERVER"); + setupServer(true); + log.info("SET UP NEW SERVER"); + Thread.sleep(10000); + log.info(this + " listener.connectionFailed: " + listener.connection= Failed); + assertTrue(listener.connectionFailed); + = + client.disconnect(); + shutdownServer(); + = + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "socket"; + } + = + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + protected void setupServer(boolean useLeasing) throws Exception + { + locatorURI =3D getTransport() + "://" + host + ":" + port + "?" + Re= moting.USE_SERVER_CONNECTION_IDENTITY + "=3Dtrue"; + if (useLeasing) + { + locatorURI +=3D "&" + InvokerLocator.CLIENT_LEASE + "=3Dtrue"; + locatorURI +=3D "&" + InvokerLocator.CLIENT_LEASE_PERIOD + "=3D20= 000"; + } + String metadata =3D System.getProperty("remoting.metadata"); + if (metadata !=3D null) + { + locatorURI +=3D "&" + metadata; + } + serverLocator =3D new InvokerLocator(locatorURI); + log.info("Starting remoting server with locator uri of: " + locatorU= RI); + HashMap config =3D new HashMap(); + config.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraServerConfig(config); + connector =3D new Connector(serverLocator, config); + connector.create(); + invocationHandler =3D new TestInvocationHandler(); + connector.addInvocationHandler("test", invocationHandler); + if (useLeasing) + { + connector.addConnectionListener(new TestConnectionListener()); + } + connector.start(); + } + = + = + protected void shutdownServer() throws Exception + { + if (connector !=3D null) + { + // Remoting versions 1.x and 2.2.x don't necessarily shut down al= l of their + // ServerThreads, so an exiting connection could connect to a Ser= verThread + // associated with the old ServerInvoker. + SocketServerInvoker invoker =3D (SocketServerInvoker) connector.g= etServerInvoker(); + Field clientpoolField =3D SocketServerInvoker.class.getDeclaredFi= eld("clientpool"); + clientpoolField.setAccessible(true); + Field socketWrapperField =3D ServerThread.class.getDeclaredField(= "socketWrapper"); + socketWrapperField.setAccessible(true); + LRUPool clientpool =3D (LRUPool) clientpoolField.get(invoker); + Set threads =3D clientpool.getContents(); + Iterator it =3D threads.iterator(); + while (it.hasNext()) + { + ServerThread t =3D (ServerThread) it.next(); + ServerSocketWrapper socketWrapper =3D (ServerSocketWrapper) so= cketWrapperField.get(t); + socketWrapper.close(); + } + connector.stop(); + } + } + = + = + static class TestInvocationHandler implements ServerInvocationHandler + { + public void addListener(InvokerCallbackHandler callbackHandler) {} + public Object invoke(final InvocationRequest invocation) throws Thro= wable + { + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + + static class TestConnectionListener implements ConnectionListener + { + public boolean connectionFailed; + public Throwable throwable; + = + public void handleConnectionException(Throwable throwable, Client cl= ient) + { + connectionFailed =3D true; + this.throwable =3D throwable; + log.info(this + " received connection notification: connectionFai= led: " + connectionFailed); + } + = + } +} \ No newline at end of file --===============5119081058367131826==--