+ * Copyright Feb 16, 2010 + */ +public class ErrorTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(ErrorTestCase.class); + = + private static boolean firstTime =3D true; + private static int whenToFail; + = + 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(Level.INFO); + 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); = + } + InvokerRegistry.registerInvokerFactories(getTransport(), TestClientF= actory.class, TestServerFactory.class); + TestServerThread.threadCounter =3D 0; + } + + = + public void tearDown() + { + } + = + = + public void testErrorFirstTime() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + whenToFail =3D 0; + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put("timeout", "2000"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test invocations. + assertEquals("abc", client.invoke("abc")); + log.info("first invocation succeeded"); + assertEquals("lmn", client.invoke("lmn")); + log.info("second invocation succeeded"); + assertEquals("xyz", client.invoke("xyz")); + log.info("third invocation succeeded"); + assertEquals(1, TestServerThread.threadCounter); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testErrorSecondTime() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + whenToFail =3D 1; + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put("timeout", "2000"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test invocations. + assertEquals("abc", client.invoke("abc")); + log.info("first invocation succeeded"); + assertEquals("lmn", client.invoke("lmn")); + log.info("second invocation succeeded"); + assertEquals("xyz", client.invoke("xyz")); + log.info("third invocation succeeded"); + assertEquals(1, TestServerThread.threadCounter); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + + public void testErrorThirdTime() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + whenToFail =3D 2; + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put("timeout", "2000"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test invocations. + assertEquals("abc", client.invoke("abc")); + log.info("first invocation succeeded"); + assertEquals("lmn", client.invoke("lmn")); + log.info("second invocation succeeded"); + assertEquals("xyz", client.invoke("xyz")); + log.info("third invocation succeeded"); + assertEquals(1, TestServerThread.threadCounter); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "test"; + } + = + = + 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"); + 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 + { + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + = + static class TestServerInvoker extends SocketServerInvoker + { + int counter; + = + public TestServerInvoker(InvokerLocator locator, Map configuration) + { + super(locator, configuration); + } + public TestServerInvoker(InvokerLocator locator) + { + super(locator); + } + protected void processInvocation(Socket socket) throws Exception + { + clientpool =3D new LRUPool(2, maxPoolSize); + clientpool.create(); + threadpool =3D new LinkedList(); + ServerThread worker =3D new TestServerThread(socket, this, client= pool, threadpool, 0, 0, ServerSocketWrapper.class.getName()); + worker.start(); + } + } + = + = + public static class TestServerThread extends ServerThread + { + static public int threadCounter; + private int counter; + = + public TestServerThread(Socket socket, SocketServerInvoker invoker, = LRUPool clientpool, LinkedList threadpool, + int timeout, int writeTimeout, String server= SocketClassName) throws Exception + { + super(socket, invoker, clientpool, threadpool, timeout, writeTime= out, serverSocketClassName); + threadCounter++; + } + protected void processInvocation(SocketWrapper socketWrapper, InputS= tream inputStream, OutputStream outputStream) throws Exception + { = + super.processInvocation(socketWrapper, inputStream, outputStream); + if (counter++ =3D=3D whenToFail) + throw new Error("TestServerThread"); + } + } + = + = + public static class TestClientFactory implements ClientFactory + { + public ClientInvoker createClientInvoker(InvokerLocator locator, Map= config) throws IOException + { + log.info("TestClientFaotory.createClientInvoker() called"); + return new SocketClientInvoker(locator, config); + } + public boolean supportsSSL() + { + return false; + } = + } + = + = + public static class TestServerFactory implements ServerFactory + { + public ServerInvoker createServerInvoker(InvokerLocator locator, Map= config) throws IOException + { + log.info("TestServerFactory.createServerInvoker() called"); + return new TestServerInvoker(locator, config); + } + public boolean supportsSSL() + { + return false; + } + } +} \ No newline at end of file --===============0032244597734949692==--