+ * Copyright Jul 30, 2010 + */ +public class IdleTimerTaskNotifyTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(IdleTimerTaskNotifyTestC= ase.class); + = + private static boolean firstTime =3D true; + private static String DELAY =3D "delay"; + private static String MAX_POOL_SIZE =3D "2"; + private static String IDLE_TIMEOUT =3D "10"; + = + 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(XLevel.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); = + } + } + + = + public void tearDown() + { + } + = + = + public void testIdleTimerTaskNotify() throws Throwable + { + log.info("entering " + getName()); + = + // Start server and verify key parameters. + setupServer(); + SocketServerInvoker serverInvoker =3D (SocketServerInvoker) connecto= r.getServerInvoker(); + assertEquals(MAX_POOL_SIZE, Integer.toString(serverInvoker.getMaxPoo= lSize())); + assertEquals(IDLE_TIMEOUT, Integer.toString(serverInvoker.getIdleTim= eout())); + = + // Set up invocations. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + + TestThread tt1 =3D new TestThread("1", new Integer(25000), clientLoc= ator, clientConfig); + TestThread tt2 =3D new TestThread("2", new Integer(5000), clientLoca= tor, clientConfig); + TestThread tt3 =3D new TestThread("3", new Integer(0), clientLocator= , clientConfig); + tt1.start(); + Thread.sleep(1000); + tt2.start(); + Thread.sleep(1000); + tt3.start(); + Thread.sleep(10000); + = + // At this point, SocketServerInvoker has created a socket for the i= nvocation from tt3, + // has called clientpool.evict(), and is waiting on WorkerThread#0 t= o shut down. = + // * WorkerThread#0 is still running the invocation from tt1. + // * WorkerThread#1 has completed the invocation from tt2 and is wai= ting for another one. + // * The invocation from tt2 is still waiting for a thread. + assertFalse(tt1.ok); + assertTrue(tt2.ok); + assertFalse(tt3.ok); + Thread.sleep(15000); + = + // At this point, WorkerThread#0 is still executing the invocation f= rom tt1, but IdleTimerTask + // has detected that WorkerThread#1 has been idle. It shuts down Wo= rkerThread#1 and calls + // clientpool.notify(). SocketServerInvoker wakes up and creates Wo= rkerThread#2, which processes + // the invocation from tt3. + assertTrue(tt3.ok); + Thread.sleep(5000); + = + // At this point, the invocation from tt1 has completed. + assertTrue(tt1.ok); + = + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "socket"; + } + = + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + protected void setupServer() throws Exception + { + host =3D InetAddress.getLocalHost().getHostAddress(); + port =3D PortUtil.findFreePort(host); + locatorURI =3D getTransport() + "://" + host + ":" + port + "/?"; + 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"); + config.put("maxPoolSize", MAX_POOL_SIZE); + config.put("idleTimeout", IDLE_TIMEOUT); + addExtraServerConfig(config); + connector =3D new Connector(serverLocator, config); + connector.create(); + invocationHandler =3D new TestInvocationHandler(); + connector.addInvocationHandler("test", invocationHandler); + 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 + { + Map metadata =3D invocation.getRequestPayload(); + int delay =3D 5000; + if (metadata !=3D null) + { + Object o =3D metadata.get(DELAY); + if (o !=3D null && o instanceof Integer) + { + delay =3D ((Integer)o).intValue(); + } + } + log.info(this + " sleeping for " + delay + " ms"); + Thread.sleep(delay); + log.info(this + " returned from sleep"); + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + = + static class TestThread extends Thread + { + public boolean ok; + String id; + Integer delay; + InvokerLocator locator; + Map config; + = + public TestThread(String id, Integer delay, InvokerLocator locator, = Map config) + { + this.id =3D id; + this.delay =3D delay; + this.locator =3D locator; + this.config =3D config; + setName("TestThread[" + id + "]"); + } + = + public void run() + { + Client client =3D null; + = + try + { + client =3D new Client(locator, config); + client.connect(); + log.info(this + " client connected"); + Map metadata =3D new HashMap(); + metadata.put(DELAY, delay); + if (id.equals(client.invoke(id, metadata))) + { + ok =3D true; + log.info(this + " invocation successful"); + } + } + catch (Throwable e) + { + log.error("invocation on " + client + " failed", e); + } + finally + { + client.disconnect(); + } + } + = + public String toString() + { + return getName(); + } + } +} \ No newline at end of file Property changes on: remoting2/branches/2.2/src/tests/org/jboss/test/remoti= ng/transport/socket/IdleTimerTaskNotifyTestCase.java ___________________________________________________________________ Name: svn:executable + * --===============1930875944654504733==--