- * Copyright Sep 2, 2009 - *
- */ -public class GeneralizeSocketExceptionTestCase extends TestCase -{ - private static Logger log =3D Logger.getLogger(GeneralizeSocketExceptio= nTestCase.class); - private static String connectionResetMessage =3D "yada yada connection = reset yada yada"; - private static String connectionClosedMessage =3D "yada yada connection= closed yada yada"; - private static String brokenPipeMessage =3D "yada yada broken pipe yada= yada"; - = - 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.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); = - } - = - TestOutputStream.counter =3D 0; - } - - = - public void tearDown() - { - } - = - = - public void testConnectionResetNotGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionResetMessage), 2)); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation fails. - try - { - client.invoke("abc"); - } - catch (MarshalException e) - { - Throwable t =3D e.getCause(); - assertEquals(connectionResetMessage, t.getMessage()); - } - catch (Throwable e) - { - fail("Expecteded message: " + connectionResetMessage + ", got: " = + e.getMessage()); - } - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - - public void testConnectionResetGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionResetMessage), 2)); - clientConfig.put("generalizeSocketException", "true"); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation succeeds. - assertEquals("abc", client.invoke("abc")); - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - = - public void testConnectionClosedNotGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionClosedMessage), 2)); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation fails. - try - { - client.invoke("abc"); - } - catch (MarshalException e) - { - Throwable t =3D e.getCause(); - assertEquals(connectionClosedMessage, t.getMessage()); - } - catch (Throwable e) - { - fail("Expecteded message: " + connectionClosedMessage + ", got: "= + e.getMessage()); - } - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - - public void testConnectionClosedGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionClosedMessage), 2)); - clientConfig.put("generalizeSocketException", "true"); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation succeeds. - assertEquals("abc", client.invoke("abc")); - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - = - public void testBrokedPipedNotGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(brokenPipeMessage), 2)); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation fails. - try - { - client.invoke("abc"); - } - catch (MarshalException e) - { - Throwable t =3D e.getCause(); - assertEquals(brokenPipeMessage, t.getMessage()); - } - catch (Throwable e) - { - fail("Expecteded message: " + brokenPipeMessage + ", got: " + e.g= etMessage()); - } - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - - public void testBrokedPipeGeneralized() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(brokenPipeMessage), 2)); - clientConfig.put("generalizeSocketException", "true"); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Verify invocation succeeds. - assertEquals("abc", client.invoke("abc")); - = - 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() 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) {} - } - = - public static class TestSocketFactory extends SocketFactory - { - IOException exception; - int throwOn =3D -1; - = - public TestSocketFactory() - { - exception =3D new SocketException("default"); - } - public TestSocketFactory(IOException exception, int throwOn) - { - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public Socket createSocket() - { - return new TestSocket(exception, throwOn); - } - public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException - { - return new TestSocket(arg0, arg1, exception, throwOn); - } - public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption - { - return new TestSocket(arg0, arg1, exception, throwOn); - } - public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, UnknownHostException - { - return new TestSocket(arg0, arg1, arg2, arg3, exception, throwOn); - } - - public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException - { - return new TestSocket(arg0, arg1, arg2, arg3, exception, throwOn); - } - } - = - static class TestSocket extends Socket - { - IOException exception; - int throwOn; - = - public TestSocket(IOException exception, int throwOn) - { - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public TestSocket(String host, int port, IOException exception, int = throwOn) throws UnknownHostException, IOException - { - super(host, port); - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public TestSocket(InetAddress address, int port, IOException excepti= on, int throwOn) throws IOException - { - super(address, port); - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public TestSocket(String host, int port, InetAddress localAddr, int = localPort, IOException exception, int throwOn) throws IOException - { - super(host, port, localAddr, localPort); - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public TestSocket(InetAddress address, int port, InetAddress localAd= dr, int localPort, IOException exception, int throwOn) throws IOException - { - super(address, port, localAddr, localPort); - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public OutputStream getOutputStream() throws IOException - { - return new TestOutputStream(super.getOutputStream(), exception, t= hrowOn); - } - public String toString() - { - return "TestSocket[" + getLocalPort() + "->" + getPort() + "]"; - } - } - = - static class TestOutputStream extends OutputStream - { - OutputStream os; - IOException exception; - boolean closed; - int throwOn; - boolean doThrowException =3D true; - public static int counter; - = - public TestOutputStream(OutputStream os, IOException exception, int = throwOn) - { - this.os =3D os; - this.exception =3D exception; - this.throwOn =3D throwOn; - } - public void close()throws IOException - { - closed =3D true; - super.close(); - log.info(this + " closed"); - } - public void write(int b) throws IOException - { - System.out.print("b: " + b); - if (closed) - { - log.info("TestOutputStream closed, cannot write"); - throw new IOException("closed"); - } - if (doThrowException && ++counter =3D=3D throwOn) - { - throw exception; - } - os.write(b); - } - public void write(byte b[], int off, int len) throws IOException - { - System.out.print("b: "); - for (int i =3D 0; i < len; i++) - { - System.out.print(b[i] + " "); - } - System.out.println(""); - if (closed) - { - log.info("TestOutputStream closed, cannot write"); - throw new IOException("closed"); - } - log.info("TestOutputStream: counter =3D " + counter + ", throwOn = =3D " + throwOn); - if (++counter =3D=3D throwOn) - { - throw exception; - } - if (closed) - { - log.info("TestOutputStream closed, cannot write"); - throw new IOException("closed"); - } - try - { - log.info(this + " calling write()"); - doThrowException =3D false; - os.write(b, off, len); - os.flush(); - doThrowException =3D true; - log.info(this + " back from write()"); - } - catch (IOException e) - { - log.info("exception: ", e); - throw e; - } - } - } +/* + * JBoss, Home of Professional Open Source. + * Copyright 2009, Red Hat Middleware LLC, and individual contributors + * as indicated by the @author tags. See the copyright.txt file 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.socketexception; + +import java.io.IOException; +import java.io.OutputStream; +import java.net.InetAddress; +import java.net.Socket; +import java.net.SocketException; +import java.net.UnknownHostException; +import java.rmi.MarshalException; +import java.util.HashMap; +import java.util.Map; + +import javax.management.MBeanServer; +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.logging.XLevel; +import org.jboss.remoting.Client; +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.InvokerCallbackHandler; +import org.jboss.remoting.transport.Connector; +import org.jboss.remoting.transport.PortUtil; + + +/** + * Unit tests for JBREM-1146. + * = + * @author Ron Sigal + * @version = + *+ * Copyright Sep 2, 2009 + *
+ */ +public class GeneralizeSocketExceptionTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(GeneralizeSocketExceptio= nTestCase.class); + private static String connectionResetMessage =3D "yada yada connection = reset yada yada"; + private static String connectionClosedMessage =3D "yada yada connection= closed yada yada"; + private static String brokenPipeMessage =3D "yada yada broken pipe yada= yada"; + = + 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.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); = + } + = + TestOutputStream.counter =3D 0; + } + + = + public void tearDown() + { + } + = + = + public void testConnectionResetNotGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionResetMessage), 2)); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation fails. + try + { + client.invoke("abc"); + } + catch (MarshalException e) + { + Throwable t =3D e.getCause(); + assertEquals(connectionResetMessage, t.getMessage()); + } + catch (Throwable e) + { + fail("Expecteded message: " + connectionResetMessage + ", got: " = + e.getMessage()); + } + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + + public void testConnectionResetGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionResetMessage), 2)); + clientConfig.put("generalizeSocketException", "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation succeeds. + assertEquals("abc", client.invoke("abc")); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testConnectionClosedNotGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionClosedMessage), 2)); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation fails. + try + { + client.invoke("abc"); + } + catch (MarshalException e) + { + Throwable t =3D e.getCause(); + assertEquals(connectionClosedMessage, t.getMessage()); + } + catch (Throwable e) + { + fail("Expecteded message: " + connectionClosedMessage + ", got: "= + e.getMessage()); + } + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + + public void testConnectionClosedGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(connectionClosedMessage), 2)); + clientConfig.put("generalizeSocketException", "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation succeeds. + assertEquals("abc", client.invoke("abc")); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testBrokedPipedNotGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(brokenPipeMessage), 2)); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation fails. + try + { + client.invoke("abc"); + } + catch (MarshalException e) + { + Throwable t =3D e.getCause(); + assertEquals(brokenPipeMessage, t.getMessage()); + } + catch (Throwable e) + { + fail("Expecteded message: " + brokenPipeMessage + ", got: " + e.g= etMessage()); + } + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + + public void testBrokedPipeGeneralized() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException(brokenPipeMessage), 2)); + clientConfig.put("generalizeSocketException", "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Verify invocation succeeds. + assertEquals("abc", client.invoke("abc")); + = + 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() 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) {} + } + = + public static class TestSocketFactory extends SocketFactory + { + IOException exception; + int throwOn =3D -1; + = + public TestSocketFactory() + { + exception =3D new SocketException("default"); + } + public TestSocketFactory(IOException exception, int throwOn) + { + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public Socket createSocket() + { + return new TestSocket(exception, throwOn); + } + public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException + { + return new TestSocket(arg0, arg1, exception, throwOn); + } + public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption + { + return new TestSocket(arg0, arg1, exception, throwOn); + } + public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, UnknownHostException + { + return new TestSocket(arg0, arg1, arg2, arg3, exception, throwOn); + } + + public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException + { + return new TestSocket(arg0, arg1, arg2, arg3, exception, throwOn); + } + } + = + static class TestSocket extends Socket + { + IOException exception; + int throwOn; + = + public TestSocket(IOException exception, int throwOn) + { + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public TestSocket(String host, int port, IOException exception, int = throwOn) throws UnknownHostException, IOException + { + super(host, port); + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public TestSocket(InetAddress address, int port, IOException excepti= on, int throwOn) throws IOException + { + super(address, port); + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public TestSocket(String host, int port, InetAddress localAddr, int = localPort, IOException exception, int throwOn) throws IOException + { + super(host, port, localAddr, localPort); + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public TestSocket(InetAddress address, int port, InetAddress localAd= dr, int localPort, IOException exception, int throwOn) throws IOException + { + super(address, port, localAddr, localPort); + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public OutputStream getOutputStream() throws IOException + { + return new TestOutputStream(super.getOutputStream(), exception, t= hrowOn); + } + public String toString() + { + return "TestSocket[" + getLocalPort() + "->" + getPort() + "]"; + } + } + = + static class TestOutputStream extends OutputStream + { + OutputStream os; + IOException exception; + boolean closed; + int throwOn; + boolean doThrowException =3D true; + public static int counter; + = + public TestOutputStream(OutputStream os, IOException exception, int = throwOn) + { + this.os =3D os; + this.exception =3D exception; + this.throwOn =3D throwOn; + } + public void close()throws IOException + { + closed =3D true; + super.close(); + log.info(this + " closed"); + } + public void write(int b) throws IOException + { + System.out.print("b: " + b); + if (closed) + { + log.info("TestOutputStream closed, cannot write"); + throw new IOException("closed"); + } + if (doThrowException && ++counter =3D=3D throwOn) + { + throw exception; + } + os.write(b); + } + public void write(byte b[], int off, int len) throws IOException + { + System.out.print("b: "); + for (int i =3D 0; i < len; i++) + { + System.out.print(b[i] + " "); + } + System.out.println(""); + if (closed) + { + log.info("TestOutputStream closed, cannot write"); + throw new IOException("closed"); + } + log.info("TestOutputStream: counter =3D " + counter + ", throwOn = =3D " + throwOn); + if (++counter =3D=3D throwOn) + { + throw exception; + } + if (closed) + { + log.info("TestOutputStream closed, cannot write"); + throw new IOException("closed"); + } + try + { + log.info(this + " calling write()"); + doThrowException =3D false; + os.write(b, off, len); + os.flush(); + doThrowException =3D true; + log.info(this + " back from write()"); + } + catch (IOException e) + { + log.info("exception: ", e); + throw e; + } + } + } } \ No newline at end of file Property changes on: remoting2/branches/2.2/src/tests/org/jboss/test/remoti= ng/transport/socket/socketexception/GeneralizeSocketExceptionTestCase.java ___________________________________________________________________ Name: svn:eol-style + native Modified: remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transpor= t/socket/socketexception/SocketCreationExceptionTestCase.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.2/src/tests/org/jboss/test/remoting/transport/sock= et/socketexception/SocketCreationExceptionTestCase.java 2010-08-05 02:10:57= UTC (rev 6042) +++ remoting2/branches/2.2/src/tests/org/jboss/test/remoting/transport/sock= et/socketexception/SocketCreationExceptionTestCase.java 2010-08-05 02:11:36= UTC (rev 6043) @@ -1,300 +1,300 @@ -/* -* 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.socketexception; - -import java.io.IOException; -import java.net.InetAddress; -import java.net.Socket; -import java.net.SocketException; -import java.net.UnknownHostException; -import java.util.HashMap; -import java.util.Map; - -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.logging.XLevel; -import org.jboss.remoting.Client; -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; - -/** - * Unit tests for JBREM-1152. - * = - * @author Ron Sigal - * @version $Rev$ - *- * Copyright Sep 9, 2009 - *
- */ -public class SocketCreationExceptionTestCase extends TestCase -{ - protected static String SEND_CALLBACK =3D "sendCallback"; - private static Logger log =3D Logger.getLogger(SocketCreationExceptionT= estCase.class); - 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); = - } - } - - = - public void tearDown() - { - } - = - = - public void testInvocationWithSocketException() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(null); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new SocketException(getName()), 2)); - 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"); - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - = - public void testInvocationWithIOException() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(null); - = - // Create client. - InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException("Connection reset"), 2)); - clientConfig.put("generalizeSocketException", "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"); - = - 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(ServerSocketFactory ssf) 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("timeout", "5000"); - if (ssf !=3D null) - { - config.put(Remoting.CUSTOM_SERVER_SOCKET_FACTORY, ssf); - } - 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 - { - InvokerCallbackHandler callbackHandler; - = - public void addListener(InvokerCallbackHandler callbackHandler) - { - this.callbackHandler =3D callbackHandler; - } - public Object invoke(final InvocationRequest invocation) throws Thro= wable - { - Object parameter =3D invocation.getParameter(); - if (SEND_CALLBACK.equals(parameter)) - { - try - { - callbackHandler.handleCallback(new Callback("callback")); - } - catch (HandleCallbackException e) - { - log.error("unable to send callback", e); - } - } - return parameter; - } - public void removeListener(InvokerCallbackHandler callbackHandler) {} - public void setMBeanServer(MBeanServer server) {} - public void setInvoker(ServerInvoker invoker) {} - } - = - = - public static class TestSocketFactory extends SocketFactory - { - int counter; - int limit; = - IOException exception; - = - public TestSocketFactory() - { - } - public TestSocketFactory(IOException exception, int limit) - { - this.exception =3D exception; - this.limit =3D limit; - } - public Socket createSocket() throws IOException - { - counter++; - log.info("counter: " + counter); - if (counter <=3D limit) - { - log.info("throwing exception"); - throw exception; - } - log.info("returning socket"); - return new Socket(); - } - public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException - { - counter++; - log.info("counter: " + counter); - if (counter <=3D limit) - { - throw exception; - } - log.info("returning socket"); - return new Socket(arg0, arg1); - } - public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption - { - counter++; - log.info("counter: " + counter); - if (counter <=3D limit) - { - throw exception; - } - log.info("returning socket"); - return new Socket(arg0, arg1); - } - public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, UnknownHostException - { - counter++; - log.info("counter: " + counter); - if (counter <=3D limit) - { - throw exception; - } - log.info("returning socket"); - return new Socket(arg0, arg1, arg2, arg3); - } - - public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException - { - counter++; - log.info("counter: " + counter); - if (counter <=3D limit) - { - throw exception; - } - log.info("returning socket"); - return new Socket(arg0, arg1, arg2, arg3); - } - } +/* +* 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.socketexception; + +import java.io.IOException; +import java.net.InetAddress; +import java.net.Socket; +import java.net.SocketException; +import java.net.UnknownHostException; +import java.util.HashMap; +import java.util.Map; + +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.logging.XLevel; +import org.jboss.remoting.Client; +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; + +/** + * Unit tests for JBREM-1152. + * = + * @author Ron Sigal + * @version $Rev$ + *+ * Copyright Sep 9, 2009 + *
+ */ +public class SocketCreationExceptionTestCase extends TestCase +{ + protected static String SEND_CALLBACK =3D "sendCallback"; + private static Logger log =3D Logger.getLogger(SocketCreationExceptionT= estCase.class); + 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); = + } + } + + = + public void tearDown() + { + } + = + = + public void testInvocationWithSocketException() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(null); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new SocketException(getName()), 2)); + 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"); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testInvocationWithIOException() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(null); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFacto= ry(new IOException("Connection reset"), 2)); + clientConfig.put("generalizeSocketException", "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"); + = + 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(ServerSocketFactory ssf) 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("timeout", "5000"); + if (ssf !=3D null) + { + config.put(Remoting.CUSTOM_SERVER_SOCKET_FACTORY, ssf); + } + 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 + { + InvokerCallbackHandler callbackHandler; + = + public void addListener(InvokerCallbackHandler callbackHandler) + { + this.callbackHandler =3D callbackHandler; + } + public Object invoke(final InvocationRequest invocation) throws Thro= wable + { + Object parameter =3D invocation.getParameter(); + if (SEND_CALLBACK.equals(parameter)) + { + try + { + callbackHandler.handleCallback(new Callback("callback")); + } + catch (HandleCallbackException e) + { + log.error("unable to send callback", e); + } + } + return parameter; + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + = + public static class TestSocketFactory extends SocketFactory + { + int counter; + int limit; = + IOException exception; + = + public TestSocketFactory() + { + } + public TestSocketFactory(IOException exception, int limit) + { + this.exception =3D exception; + this.limit =3D limit; + } + public Socket createSocket() throws IOException + { + counter++; + log.info("counter: " + counter); + if (counter <=3D limit) + { + log.info("throwing exception"); + throw exception; + } + log.info("returning socket"); + return new Socket(); + } + public Socket createSocket(String arg0, int arg1) throws IOException= , UnknownHostException + { + counter++; + log.info("counter: " + counter); + if (counter <=3D limit) + { + throw exception; + } + log.info("returning socket"); + return new Socket(arg0, arg1); + } + public Socket createSocket(InetAddress arg0, int arg1) throws IOExce= ption + { + counter++; + log.info("counter: " + counter); + if (counter <=3D limit) + { + throw exception; + } + log.info("returning socket"); + return new Socket(arg0, arg1); + } + public Socket createSocket(String arg0, int arg1, InetAddress arg2, = int arg3) throws IOException, UnknownHostException + { + counter++; + log.info("counter: " + counter); + if (counter <=3D limit) + { + throw exception; + } + log.info("returning socket"); + return new Socket(arg0, arg1, arg2, arg3); + } + + public Socket createSocket(InetAddress arg0, int arg1, InetAddress a= rg2, int arg3) throws IOException + { + counter++; + log.info("counter: " + counter); + if (counter <=3D limit) + { + throw exception; + } + log.info("returning socket"); + return new Socket(arg0, arg1, arg2, arg3); + } + } } \ No newline at end of file Property changes on: remoting2/branches/2.2/src/tests/org/jboss/test/remoti= ng/transport/socket/socketexception/SocketCreationExceptionTestCase.java ___________________________________________________________________ Name: svn:eol-style + native --===============0022336448915152436==--