From jboss-remoting-commits at lists.jboss.org Wed Apr 22 23:43:15 2009 Content-Type: multipart/mixed; boundary="===============5693087962438556927==" MIME-Version: 1.0 From: jboss-remoting-commits at lists.jboss.org To: jboss-remoting-commits at lists.jboss.org Subject: [jboss-remoting-commits] JBoss Remoting SVN: r5088 - remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socket/timeout. Date: Wed, 22 Apr 2009 23:39:14 -0400 Message-ID: --===============5693087962438556927== Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: quoted-printable Author: ron.sigal(a)jboss.com Date: 2009-04-22 23:39:13 -0400 (Wed, 22 Apr 2009) New Revision: 5088 Added: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socke= t/timeout/SocketWriteTimeoutTestCase.java remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socke= t/timeout/WriteTimeoutTestParent.java Log: JBREM-1120: New unit tests. Added: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/s= ocket/timeout/SocketWriteTimeoutTestCase.java =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D= =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D= =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D --- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/sock= et/timeout/SocketWriteTimeoutTestCase.java (rev 0) +++ remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/sock= et/timeout/SocketWriteTimeoutTestCase.java 2009-04-23 03:39:13 UTC (rev 508= 8) @@ -0,0 +1,39 @@ +/* +* JBoss, Home of Professional Open Source +* Copyright 2009, JBoss Inc., 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.transport.socket.timeout; + +/** + * Unit tests for JBREM-1120. + * = + * @author Ron Sigal + * @version $Rev$ + *

+ * Copyright Apr 22, 2009 + *

+ */ +public class SocketWriteTimeoutTestCase extends WriteTimeoutTestParent +{ + protected String getTransport() + { + return "socket"; + } +} Added: remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/s= ocket/timeout/WriteTimeoutTestParent.java =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D= =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D= =3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D=3D --- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/sock= et/timeout/WriteTimeoutTestParent.java (rev 0) +++ remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/sock= et/timeout/WriteTimeoutTestParent.java 2009-04-23 03:39:13 UTC (rev 5088) @@ -0,0 +1,652 @@ +/* +* JBoss, Home of Professional Open Source +* Copyright 2009, JBoss Inc., 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.transport.socket.timeout; + +import java.io.EOFException; +import java.io.IOException; +import java.io.OutputStream; +import java.net.InetAddress; +import java.net.ServerSocket; +import java.net.Socket; +import java.net.UnknownHostException; +import java.util.HashMap; +import java.util.Map; +import java.util.Set; + +import javax.management.MBeanServer; +import javax.net.ServerSocketFactory; +import javax.net.SocketFactory; + +import junit.framework.TestCase; + +import org.apache.log4j.ConsoleAppender; +import org.apache.log4j.Level; +import org.apache.log4j.Logger; +import org.apache.log4j.PatternLayout; +import org.jboss.remoting.Client; +import org.jboss.remoting.InvocationFailureException; +import org.jboss.remoting.InvocationRequest; +import org.jboss.remoting.InvokerLocator; +import org.jboss.remoting.Remoting; +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 org.jboss.remoting.transport.PortUtil; +import org.jboss.remoting.transport.bisocket.Bisocket; +import org.jboss.remoting.transport.socket.SocketServerInvoker; +import org.jboss.remoting.transport.socket.SocketWrapper; + + +/** + * Unit tests for JBREM-1120. + * = + * @author Ron Sigal + * @version $Rev$ + *

+ * Copyright Apr 22, 2009 + *

+ */ +public abstract class WriteTimeoutTestParent extends TestCase +{ + private static Logger log =3D Logger.getLogger(WriteTimeoutTestParent.c= lass); + = + private static boolean firstTime =3D true; + = + protected static int SECONDARY_SERVER_SOCKET_PORT =3D 8765; + protected static String SECONDARY_SERVER_SOCKET_PORT_STRING =3D "8765"; + protected static boolean callbackTest; + = + 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.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); = + } + } + + = + public void tearDown() + { + } + = + = + public void testClientWriteTimeout() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(false, false, "", -1); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(SocketWrapper.WRITE_TIMEOUT, "1000"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(5000)); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + log.info("**************************************"); + log.info("*** WorkerThread error is expected ***"); + log.info("**************************************"); + = + // Test client side write timeout. + try + { + client.invoke("abc"); + } + catch (InvocationFailureException e) + { + log.info(e.getMessage()); + assertNotNull(e.getMessage()); + assertTrue(e.getMessage().startsWith("Unable to perform invocatio= n")); + assertTrue(e.getCause() instanceof IOException); + IOException ioe =3D (IOException) e.getCause(); + assertEquals("closed", ioe.getMessage()); + log.info("got expected Exception"); + } + catch (Throwable t) + { + log.error("got unexpected Exception", t); + fail("got unexpected Exception"); + } + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testServerWriteTimeout() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(true, false, "1000", 5000); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put("numberOfCallRetries", "1"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + log.info("**************************************"); + log.info("*** WorkerThread error is expected ***"); + log.info("**************************************"); + = + // Test server side write timeout. + try + { + client.invoke("abc"); + } + catch (InvocationFailureException e) + { + log.info(e.getMessage()); + assertNotNull(e.getMessage()); + assertTrue(e.getMessage().startsWith("Unable to perform invocatio= n")); + assertTrue(e.getCause() instanceof EOFException); + log.info("got expected Exception"); + } + catch (Throwable t) + { + log.error("got unexpected Exception", t); + fail("got unexpected Exception"); + } + = + // Test server invoker state. + Thread.sleep(4000); + SocketServerInvoker serverInvoker =3D (SocketServerInvoker) connecto= r.getServerInvoker(); + assertEquals(0, serverInvoker.getCurrentClientPoolSize()); + assertEquals(1, serverInvoker.getCurrentThreadPoolSize()); + log.info("used ServerThread has returned to threadPool"); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testClientCallbackWriteTimeout() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(false, false, "", -1); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + 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 + client.invoke("abc"); + log.info("connection is good"); + = + // Test client callback write timeout. + log.info("registering callback handler"); + log.info("**************************************"); + log.info("*** WorkerThread error is expected ***"); + log.info("**************************************"); + TestCallbackHandler callbackHandler =3D new TestCallbackHandler(); + HashMap metadata =3D new HashMap(); + if (isBisocket(getTransport())) + { + metadata.put(SocketWrapper.WRITE_TIMEOUT, "1000"); + metadata.put(Remoting.SOCKET_FACTORY_NAME, TestSocketFactory.clas= s.getName()); + metadata.put("numberOfCallRetries", "1"); + metadata.put(Bisocket.IS_CALLBACK_SERVER, "true"); + metadata.put(Bisocket.PING_FREQUENCY, "11111111"); = + } + else + { + metadata.put(SocketWrapper.WRITE_TIMEOUT, "1000"); + metadata.put(ServerInvoker.SERVER_SOCKET_FACTORY, TestServerSocke= tFactory.class.getName()); + metadata.put("numberOfCallRetries", "1"); + } + client.addListener(callbackHandler, metadata, null, true); + = + // Test server invoker state. + Thread.sleep(4000); + Set callbackConnectors =3D client.getCallbackConnectors(callbackHand= ler); + assertEquals(1, callbackConnectors.size()); + Connector callbackConnector =3D (Connector) callbackConnectors.itera= tor().next(); + SocketServerInvoker serverInvoker =3D (SocketServerInvoker) callback= Connector.getServerInvoker(); + assertEquals(0, serverInvoker.getCurrentClientPoolSize()); + assertEquals(1, serverInvoker.getCurrentThreadPoolSize()); + log.info("used ServerThread has returned to threadPool"); + + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testServerCallbackWriteTimeout() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + if (isBisocket(getTransport())) + { + callbackTest =3D true; + setupServer(true, false, "1000", 5000); + } + else + { + setupServer(false, true, "1000", 5000); + } + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put("numberOfCallRetries", "1"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connection + client.invoke("abc"); + log.info("connection is good"); + = + // Test server callback write timeout. + log.info("registering callback handler"); + log.info("**************************************"); + log.info("*** WorkerThread error is expected ***"); + log.info("**************************************"); + HashMap metadata =3D new HashMap(); + if (isBisocket(getTransport())) + { + metadata.put(Bisocket.IS_CALLBACK_SERVER, "true"); + } + TestCallbackHandler callbackHandler =3D new TestCallbackHandler(); + client.addListener(callbackHandler, null, null, true); + = + // Test server invoker state. + Thread.sleep(4000); + Throwable t =3D invocationHandler.t; + assertTrue(t instanceof HandleCallbackException); + assertTrue(t.getCause() instanceof InvocationFailureException); + InvocationFailureException e =3D (InvocationFailureException) t.getC= ause(); + assertNotNull(e.getMessage()); + assertTrue(e.getMessage().startsWith("Unable to perform invocation")= ); + assertTrue(e.getCause() instanceof IOException); + IOException ioe =3D (IOException) e.getCause(); + assertEquals("closed", ioe.getMessage()); + log.info("got expected Exception"); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected abstract String getTransport(); + = + protected boolean isBisocket(String transport) + { + return transport.indexOf("bisocket") >=3D 0; + } + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + protected void setupServer(boolean setWriteTimeout, boolean setCallback= WriteTimeout, + String writeTimeout, int blockingTime) throw= s 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"); + if (isBisocket(getTransport())) + { + config.put(Bisocket.SECONDARY_BIND_PORT, SECONDARY_SERVER_SOCKET_= PORT_STRING); + config.put(Bisocket.PING_FREQUENCY, "11111111"); = + } + if (setWriteTimeout) + { + config.put(SocketWrapper.WRITE_TIMEOUT, writeTimeout); + config.put(Remoting.CUSTOM_SERVER_SOCKET_FACTORY, new TestServerS= ocketFactory(blockingTime)); + } + if (setCallbackWriteTimeout) + { + config.put(SocketWrapper.WRITE_TIMEOUT, writeTimeout); + config.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFactory(= blockingTime)); + } + if (callbackTest) + { + config.put("numberOfCallRetries", "1"); + } + 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 + { + Throwable t; + = + public void addListener(InvokerCallbackHandler callbackHandler) + { + try + { + callbackHandler.handleCallback(new Callback("callback")); + } + catch (Throwable t) + { + this.t =3D t; + } + } + 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 TestCallbackHandler implements InvokerCallbackHandler + { + public void handleCallback(Callback callback) throws HandleCallbackE= xception + { + log.info("received callback"); + } = + } + = + = + static public class TestServerSocketFactory extends ServerSocketFactory + { + int timeout; + public TestServerSocketFactory() + { + this.timeout =3D 5000; + } = + public TestServerSocketFactory(int timeout) + { + this.timeout =3D timeout; + } + public ServerSocket createServerSocket() throws IOException + { + ServerSocket ss =3D null; + if (callbackTest) + { + ss =3D ServerSocketFactory.getDefault().createServerSocket(); + } + else + { + ss =3D new TestServerSocket(timeout); + } + log.info("returning: " + ss); + return ss; + } + public ServerSocket createServerSocket(int port) throws IOException + { + ServerSocket ss =3D null; + if (callbackTest && port !=3D SECONDARY_SERVER_SOCKET_PORT) + { + ss =3D ServerSocketFactory.getDefault().createServerSocket(por= t); + } + else + { + ss =3D new TestServerSocket(port, timeout); + } + log.info("returning: " + ss); + return ss; + } + + public ServerSocket createServerSocket(int port, int backlog) throws= IOException + { + ServerSocket ss =3D null; + if (callbackTest && port !=3D SECONDARY_SERVER_SOCKET_PORT) + { + ss =3D ServerSocketFactory.getDefault().createServerSocket(por= t, backlog); + } + else + { + ss =3D new TestServerSocket(port, backlog, timeout); + } + log.info("returning: " + ss); + return ss; + } + + public ServerSocket createServerSocket(int port, int backlog, InetAd= dress ifAddress) throws IOException + { + ServerSocket ss =3D null; + if (callbackTest && port !=3D SECONDARY_SERVER_SOCKET_PORT) + { + ss =3D ServerSocketFactory.getDefault().createServerSocket(por= t, backlog, ifAddress); + } + else + { + ss =3D new TestServerSocket(port, backlog, ifAddress, timeout); + } + log.info("returning: " + ss); + return ss; + } + } + = + = + static class TestServerSocket extends ServerSocket + { + int timeout; + + public TestServerSocket(int timeout) throws IOException + { + super(); + this.timeout =3D timeout; + } + public TestServerSocket(int port, int timeout) throws IOException + { + super(port); + this.timeout =3D timeout; + } + public TestServerSocket(int port, int backlog, int timeout) throws I= OException + { + super(port, backlog); + this.timeout =3D timeout; + } + public TestServerSocket(int port, int backlog, InetAddress bindAddr,= int timeout) throws IOException + { + super(port, backlog, bindAddr); + this.timeout =3D timeout; + } + public Socket accept() throws IOException + { + Socket s =3D new TestSocket(timeout); + implAccept(s); + return s; + } + public String toString() + { + return "TestServerSocket[" + getLocalPort() + "]"; + } + } + = + = + public static class TestSocketFactory extends SocketFactory + { + int timeout; + = + public TestSocketFactory() + { + timeout =3D 5000; + } + public TestSocketFactory(int timeout) + { + this.timeout =3D timeout; + } + public Socket createSocket() + { + return new TestSocket(timeout); + } + public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException + { + return new TestSocket(arg0, arg1, timeout); + } + + public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption + { + return new TestSocket(arg0, arg1, timeout); + } + + public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, UnknownHostException + { + return new TestSocket(arg0, arg1, arg2, arg3, timeout); + } + + public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException + { + return new TestSocket(arg0, arg1, arg2, arg3, timeout); + } + } + = + static class TestSocket extends Socket + { + int timeout; + = + public TestSocket(int timeout) + { + this.timeout =3D timeout; + } + public TestSocket(String host, int port, int timeout) throws Unknown= HostException, IOException + { + super(host, port); + this.timeout =3D timeout; + } + public TestSocket(InetAddress address, int port, int timeout) throws= IOException + { + super(address, port); + this.timeout =3D timeout; + } + public TestSocket(String host, int port, InetAddress localAddr, int = localPort, int timeout) throws IOException + { + super(host, port, localAddr, localPort); + this.timeout =3D timeout; + } + public TestSocket(InetAddress address, int port, InetAddress localAd= dr, int localPort, int timeout) throws IOException + { + super(address, port, localAddr, localPort); + this.timeout =3D timeout; + } + public OutputStream getOutputStream() throws IOException + { + return new TestOutputStream(super.getOutputStream(), timeout); + } + } + = + static class TestOutputStream extends OutputStream + { + OutputStream os; + int timeout; + boolean closed; + = + public TestOutputStream(OutputStream os, int timeout) + { + this.os =3D os; + this.timeout =3D timeout; + } + public void close()throws IOException + { + closed =3D true; + super.close(); + log.info("closed"); + } + public void write(int b) throws IOException + { + if (closed) + { + log.info("TestOutputStream closed, cannot write"); + throw new IOException("closed"); + } + os.write(b); + } + public void write(byte b[], int off, int len) throws IOException + { + if (closed) + { + log.info("TestOutputStream closed, cannot write"); + throw new IOException("closed"); + } + try + { + log.info("TestOutputStream.write() sleeping: " + timeout); + Thread.sleep(timeout); + } + catch (InterruptedException e) + { + e.printStackTrace(); + } + log.info("TestOutputStream writing"); + os.write(b, off, len); + } + } +} \ No newline at end of file --===============5693087962438556927==--