+ * 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.DEBUG); + 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 --===============8574673847392635486==--