+ * Copyright Nov 17, 2010 + *
+ */ +public class ServerIdentityTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(ServerIdentityTestCase.c= lass); + = + protected String host; + protected int port; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected TestInvocationHandler invocationHandler; + protected ByteArrayOutputStream baosOut; + protected PrintStream originalOutPrintStream; + protected ByteArrayOutputStream baosErr; + protected PrintStream originalErrPrintStream; + = + public void setUp() throws Exception + { + originalOutPrintStream =3D System.out; + baosOut =3D new ByteArrayOutputStream(); + PrintStream ps =3D new PrintStream(baosOut); + setOut(ps); + = + originalErrPrintStream =3D System.err; + baosErr =3D new ByteArrayOutputStream(); + ps =3D new PrintStream(baosErr); + setErr(ps); + = + Logger.getLogger("org.jboss.remoting").setLevel(Level.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); + = + setOut(originalOutPrintStream); + String sOut =3D new String(baosOut.toByteArray()); + System.out.println(sOut); + setErr(originalErrPrintStream); + String sErr =3D new String(baosErr.toByteArray()); + System.out.println(sErr); + assertTrue(sOut.indexOf("detected new serverId:") > -1 || sErr.index= Of("detected new serverId:") > -1); + = + 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); + = + setOut(originalOutPrintStream); + String sOut =3D new String(baosOut.toByteArray()); + System.out.println(sOut); + setErr(originalErrPrintStream); + String sErr =3D new String(baosErr.toByteArray()); + System.out.println(sErr); + assertTrue(sOut.indexOf("detected new serverId:") > -1 || sErr.index= Of("detected new serverId:") > -1); + = + client.disconnect(); + shutdownServer(); + = + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "socket"; + } + = + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + private void setOut(final PrintStream ps) + { + if (SecurityUtility.skipAccessControl()) + { + System.setOut(ps); + return; + } + = + try + { + AccessController.doPrivileged( new PrivilegedExceptionAction() + { + public Object run() throws Exception + { + System.setOut(ps); + return null; + } + }); + } + catch (PrivilegedActionException e) + { + throw (RuntimeException) e.getCause(); + } + } + = + = + private void setErr(final PrintStream ps) + { + if (SecurityUtility.skipAccessControl()) + { + System.setErr(ps); + return; + } + = + try + { + AccessController.doPrivileged( new PrivilegedExceptionAction() + { + public Object run() throws Exception + { + System.setErr(ps); + return null; + } + }); + } + catch (PrivilegedActionException e) + { + throw (RuntimeException) e.getCause(); + } + } + = + = + 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) + 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 --===============7196967852955918916==--