[jboss-remoting-commits] JBoss Remoting SVN: r6117 - in remoting2/branches/2.5.2.SP2_JBREM-1245/src: tests/org/jboss/test/remoting/transport/socket and 1 other directories.

jboss-remoting-commits at lists.jboss.org jboss-remoting-commits at lists.jboss.org
Fri Oct 1 13:09:38 EDT 2010


Author: jbertram at redhat.com
Date: 2010-10-01 13:09:37 -0400 (Fri, 01 Oct 2010)
New Revision: 6117

Added:
   remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/
   remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java
Removed:
   remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java
Modified:
   remoting2/branches/2.5.2.SP2_JBREM-1245/src/main/org/jboss/remoting/transport/socket/MicroSocketClientInvoker.java
Log:
JBPAPP-5137

Modified: remoting2/branches/2.5.2.SP2_JBREM-1245/src/main/org/jboss/remoting/transport/socket/MicroSocketClientInvoker.java
===================================================================
--- remoting2/branches/2.5.2.SP2_JBREM-1245/src/main/org/jboss/remoting/transport/socket/MicroSocketClientInvoker.java	2010-09-25 01:41:02 UTC (rev 6116)
+++ remoting2/branches/2.5.2.SP2_JBREM-1245/src/main/org/jboss/remoting/transport/socket/MicroSocketClientInvoker.java	2010-10-01 17:09:37 UTC (rev 6117)
@@ -139,7 +139,7 @@
    public static long serializeTime = 0;
    public static long deserializeTime = 0;
    
-   private static final String patternString = "^.*(?:connection.*reset|connection.*closed|broken.*pipe).*$";
+   private static final String patternString = "^.*(?:connection.*reset|connection.*closed|broken.*pipe|connection.*shutdown).*$";
    private static final Pattern RETRIABLE_ERROR_MESSAGE = Pattern.compile(patternString, Pattern.CASE_INSENSITIVE);
    
    /**

Copied: remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable (from rev 6108, remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socket/retriable)

Deleted: remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java
===================================================================
--- remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java	2010-09-23 04:08:58 UTC (rev 6108)
+++ remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java	2010-10-01 17:09:37 UTC (rev 6117)
@@ -1,340 +0,0 @@
-/*
- * JBoss, Home of Professional Open Source
- * Copyright 2010, 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.retriable;
-
-import java.io.IOException;
-import java.io.OutputStream;
-import java.net.InetAddress;
-import java.net.Socket;
-import java.net.UnknownHostException;
-import java.util.HashMap;
-import java.util.Map;
-
-import javax.management.MBeanServer;
-import javax.net.SocketFactory;
-import javax.net.ssl.SSLException;
-
-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.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 test for JBREM-1245.
- * 
- * @author <a href="mailto:ron.sigal at jboss.com">Ron Sigal</a>
- * @version $Rev$
- * <p>
- * Copyright Sep 22, 2010
- * </p>
- */
-public class SocketGeneralizedExceptionTestCase extends TestCase
-{
-   private static Logger log = Logger.getLogger(SocketGeneralizedExceptionTestCase.class);
-   
-   private static boolean firstTime = true;
-   protected static IOException exceptionToThrow;
-   
-   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 = false;
-         Logger.getLogger("org.jboss.remoting").setLevel(Level.TRACE);
-         Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO);
-         String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
-         PatternLayout layout = new PatternLayout(pattern);
-         ConsoleAppender consoleAppender = new ConsoleAppender(layout);
-         Logger.getRootLogger().addAppender(consoleAppender);  
-      }
-      
-      TestOutputStream.counter = 0;
-      TestOutputStream.threwException = false;
-   }
-
-   
-   public void tearDown()
-   {
-   }
-   
-   
-   public void testSSLException() throws Throwable
-   {
-      log.info("entering " + getName());
-      exceptionToThrow = new SSLException("Connection has been shutdown");
-      doTest();
-      log.info(getName() + " PASSES");
-   }
-   
-   
-   public void testConnectionResetException() throws Throwable
-   {
-      log.info("entering " + getName());
-      exceptionToThrow = new IOException("Connection reset by peer");
-      doTest();
-      log.info(getName() + " PASSES");
-   }
-   
-
-   public void testConnectionClosedException() throws Throwable
-   {
-      log.info("entering " + getName());
-      exceptionToThrow = new IOException("Connection is closed");
-      doTest();
-      log.info(getName() + " PASSES");
-   }
-   
-
-   public void testBrokenPipeException() throws Throwable
-   {
-      log.info("entering " + getName());
-      exceptionToThrow = new IOException("Broken pipe");
-      doTest();
-      log.info(getName() + " PASSES");
-   }
-   
-   
-   protected void doTest() throws Throwable
-   {
-      // Start server.
-      setupServer();
-      
-      // Create client.
-      InvokerLocator clientLocator = new InvokerLocator(locatorURI);
-      HashMap clientConfig = new HashMap();
-      clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
-      clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFactory(2));
-      addExtraClientConfig(clientConfig);
-      Client client = new Client(clientLocator, clientConfig);
-      client.connect();
-      log.info("client is connected");
-      
-      // Verify invocation works in spite of exception.
-      assertEquals("xyz", client.invoke("xyz"));
-      assertTrue(TestOutputStream.threwException);
-      
-      client.disconnect();
-      shutdownServer(); 
-   }
-   
-   
-   protected String getTransport()
-   {
-      return "socket";
-   }
-   
-   protected void addExtraServerConfig(Map config) {}
-   protected void addExtraClientConfig(Map config) {}
-   
-
-   protected void setupServer() throws Exception
-   {
-      host = InetAddress.getLocalHost().getHostAddress();
-      port = PortUtil.findFreePort(host);
-      locatorURI = getTransport() + "://" + host + ":" + port + "/?generalizeSocketException=true";
-      String metadata = System.getProperty("remoting.metadata");
-      if (metadata != null)
-      {
-         locatorURI += "&" + metadata;
-      }
-      serverLocator = new InvokerLocator(locatorURI);
-      log.info("Starting remoting server with locator uri of: " + locatorURI);
-      HashMap config = new HashMap();
-      config.put(InvokerLocator.FORCE_REMOTE, "true");
-      addExtraServerConfig(config);
-      connector = new Connector(serverLocator, config);
-      connector.create();
-      invocationHandler = new TestInvocationHandler();
-      connector.addInvocationHandler("test", invocationHandler);
-      connector.start();
-   }
-   
-   
-   protected void shutdownServer() throws Exception
-   {
-      if (connector != null)
-         connector.stop();
-   }
-   
-   
-   static class TestInvocationHandler implements ServerInvocationHandler
-   {
-      public void addListener(InvokerCallbackHandler callbackHandler) {}
-      public Object invoke(final InvocationRequest invocation) throws Throwable
-      {
-         return invocation.getParameter();
-      }
-      public void removeListener(InvokerCallbackHandler callbackHandler) {}
-      public void setMBeanServer(MBeanServer server) {}
-      public void setInvoker(ServerInvoker invoker) {}
-   }
-   
-   
-   public static class TestSocketFactory extends SocketFactory
-   {
-      int initialSuccesses = 1;
-
-      public TestSocketFactory()
-      {
-      }
-      public TestSocketFactory(int initialSuccesses)
-      {
-         this.initialSuccesses = initialSuccesses;
-      }
-      public Socket createSocket()
-      {
-         Socket s = new TestSocket(initialSuccesses);
-         log.info("returning " + s);
-         return s;
-      }
-      public Socket createSocket(String arg0, int arg1) throws IOException, UnknownHostException
-      {
-         Socket s = new TestSocket(arg0, arg1, initialSuccesses);
-         log.info("returning " + s);
-         return s;
-      }
-
-      public Socket createSocket(InetAddress arg0, int arg1) throws IOException
-      {
-         Socket s = new TestSocket(arg0, arg1, initialSuccesses);
-         log.info("returning " + s);
-         return s;
-      }
-
-      public Socket createSocket(String arg0, int arg1, InetAddress arg2, int arg3) throws IOException, UnknownHostException
-      {
-         Socket s = new TestSocket(arg0, arg1, arg2, arg3, initialSuccesses);
-         log.info("returning " + s);
-         return s;
-      }
-
-      public Socket createSocket(InetAddress arg0, int arg1, InetAddress arg2, int arg3) throws IOException
-      {
-         Socket s = new TestSocket(arg0, arg1, arg2, arg3, initialSuccesses);
-         log.info("returning " + s);
-         return s;
-      }
-   }
-   
-   
-   static class TestSocket extends Socket
-   {
-      int initialSuccesses;
-      
-      public TestSocket(int initialWrites)
-      {
-         this.initialSuccesses = initialWrites;
-      }
-      public TestSocket(String host, int port, int initialWrites) throws UnknownHostException, IOException
-      {
-         super(host, port);
-         this.initialSuccesses = initialWrites;
-      }
-      public TestSocket(InetAddress address, int port, int initialWrites) throws IOException
-      {
-         super(address, port);
-         this.initialSuccesses = initialWrites;
-      }
-      public TestSocket(String host, int port, InetAddress localAddr, int localPort, int initialWrites) throws IOException
-      {
-         super(host, port, localAddr, localPort);
-         this.initialSuccesses = initialWrites;
-      }
-      public TestSocket(InetAddress address, int port, InetAddress localAddr, int localPort, int initialWrites) throws IOException
-      {
-         super(address, port, localAddr, localPort);
-         this.initialSuccesses = initialWrites;
-      }
-      public OutputStream getOutputStream() throws IOException
-      {
-         return new TestOutputStream(super.getOutputStream(), initialSuccesses);
-      }
-      public String toString()
-      {
-         return "TestSocket[" + getLocalPort() + "->" + getPort() + "]";
-      }
-   }
-   
-   
-   static class TestOutputStream extends OutputStream
-   {
-      public static int counter;
-      public static boolean threwException;
-      
-      OutputStream os;
-      boolean closed;
-      int initialWrites;
-      boolean doThrow = true;
-      
-      public TestOutputStream(OutputStream os, int initialWrites)
-      {
-         this.os = os;
-         this.initialWrites = initialWrites;
-      }
-      public void write(int b) throws IOException
-      {
-         if (doThrow && ++counter == initialWrites)
-         {
-            log.info("throwing " + exceptionToThrow);
-            threwException = true;
-            throw exceptionToThrow;
-         }
-         os.write(b);
-      }
-      public void write(byte b[], int off, int len) throws IOException
-      {
-         log.info("TestOutputStream: counter = " + (counter + 1) + ", initialWrites = " + initialWrites);
-         if (++counter == initialWrites)
-         {
-            log.info("throwing " + exceptionToThrow);
-            threwException = true;
-            throw exceptionToThrow;
-         }
-         log.info(this + " calling write()");
-         doThrow = false;
-         os.write(b, off, len);
-         os.flush();
-         doThrow = true;
-         log.info(this + " back from write()");
-      }
-   }
-}
\ No newline at end of file

Copied: remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java (from rev 6108, remoting2/branches/2.x/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java)
===================================================================
--- remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java	                        (rev 0)
+++ remoting2/branches/2.5.2.SP2_JBREM-1245/src/tests/org/jboss/test/remoting/transport/socket/retriable/SocketGeneralizedExceptionTestCase.java	2010-10-01 17:09:37 UTC (rev 6117)
@@ -0,0 +1,341 @@
+/*
+ * JBoss, Home of Professional Open Source
+ * Copyright 2010, 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.retriable;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.net.InetAddress;
+import java.net.Socket;
+import java.net.UnknownHostException;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.management.MBeanServer;
+import javax.net.SocketFactory;
+import javax.net.ssl.SSLException;
+
+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 test for JBREM-1245.
+ * 
+ * @author <a href="mailto:ron.sigal at jboss.com">Ron Sigal</a>
+ * @version $Rev$
+ * <p>
+ * Copyright Sep 22, 2010
+ * </p>
+ */
+public class SocketGeneralizedExceptionTestCase extends TestCase
+{
+   private static Logger log = Logger.getLogger(SocketGeneralizedExceptionTestCase.class);
+   
+   private static boolean firstTime = true;
+   protected static IOException exceptionToThrow;
+   
+   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 = false;
+         Logger.getLogger("org.jboss.remoting").setLevel(XLevel.TRACE);
+         Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO);
+         String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
+         PatternLayout layout = new PatternLayout(pattern);
+         ConsoleAppender consoleAppender = new ConsoleAppender(layout);
+         Logger.getRootLogger().addAppender(consoleAppender);  
+      }
+      
+      TestOutputStream.counter = 0;
+      TestOutputStream.threwException = false;
+   }
+
+   
+   public void tearDown()
+   {
+   }
+   
+   
+   public void testSSLException() throws Throwable
+   {
+      log.info("entering " + getName());
+      exceptionToThrow = new SSLException("Connection has been shutdown");
+      doTest();
+      log.info(getName() + " PASSES");
+   }
+   
+   
+   public void testConnectionResetException() throws Throwable
+   {
+      log.info("entering " + getName());
+      exceptionToThrow = new IOException("Connection reset by peer");
+      doTest();
+      log.info(getName() + " PASSES");
+   }
+   
+
+   public void testConnectionClosedException() throws Throwable
+   {
+      log.info("entering " + getName());
+      exceptionToThrow = new IOException("Connection is closed");
+      doTest();
+      log.info(getName() + " PASSES");
+   }
+   
+
+   public void testBrokenPipeException() throws Throwable
+   {
+      log.info("entering " + getName());
+      exceptionToThrow = new IOException("Broken pipe");
+      doTest();
+      log.info(getName() + " PASSES");
+   }
+   
+   
+   protected void doTest() throws Throwable
+   {
+      // Start server.
+      setupServer();
+      
+      // Create client.
+      InvokerLocator clientLocator = new InvokerLocator(locatorURI);
+      HashMap clientConfig = new HashMap();
+      clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
+      clientConfig.put(Remoting.CUSTOM_SOCKET_FACTORY, new TestSocketFactory(2));
+      addExtraClientConfig(clientConfig);
+      Client client = new Client(clientLocator, clientConfig);
+      client.connect();
+      log.info("client is connected");
+      
+      // Verify invocation works in spite of exception.
+      assertEquals("xyz", client.invoke("xyz"));
+      assertTrue(TestOutputStream.threwException);
+      
+      client.disconnect();
+      shutdownServer(); 
+   }
+   
+   
+   protected String getTransport()
+   {
+      return "socket";
+   }
+   
+   protected void addExtraServerConfig(Map config) {}
+   protected void addExtraClientConfig(Map config) {}
+   
+
+   protected void setupServer() throws Exception
+   {
+      host = InetAddress.getLocalHost().getHostAddress();
+      port = PortUtil.findFreePort(host);
+      locatorURI = getTransport() + "://" + host + ":" + port + "/?generalizeSocketException=true";
+      String metadata = System.getProperty("remoting.metadata");
+      if (metadata != null)
+      {
+         locatorURI += "&" + metadata;
+      }
+      serverLocator = new InvokerLocator(locatorURI);
+      log.info("Starting remoting server with locator uri of: " + locatorURI);
+      HashMap config = new HashMap();
+      config.put(InvokerLocator.FORCE_REMOTE, "true");
+      addExtraServerConfig(config);
+      connector = new Connector(serverLocator, config);
+      connector.create();
+      invocationHandler = new TestInvocationHandler();
+      connector.addInvocationHandler("test", invocationHandler);
+      connector.start();
+   }
+   
+   
+   protected void shutdownServer() throws Exception
+   {
+      if (connector != null)
+         connector.stop();
+   }
+   
+   
+   static class TestInvocationHandler implements ServerInvocationHandler
+   {
+      public void addListener(InvokerCallbackHandler callbackHandler) {}
+      public Object invoke(final InvocationRequest invocation) throws Throwable
+      {
+         return invocation.getParameter();
+      }
+      public void removeListener(InvokerCallbackHandler callbackHandler) {}
+      public void setMBeanServer(MBeanServer server) {}
+      public void setInvoker(ServerInvoker invoker) {}
+   }
+   
+   
+   public static class TestSocketFactory extends SocketFactory
+   {
+      int initialSuccesses = 1;
+
+      public TestSocketFactory()
+      {
+      }
+      public TestSocketFactory(int initialSuccesses)
+      {
+         this.initialSuccesses = initialSuccesses;
+      }
+      public Socket createSocket()
+      {
+         Socket s = new TestSocket(initialSuccesses);
+         log.info("returning " + s);
+         return s;
+      }
+      public Socket createSocket(String arg0, int arg1) throws IOException, UnknownHostException
+      {
+         Socket s = new TestSocket(arg0, arg1, initialSuccesses);
+         log.info("returning " + s);
+         return s;
+      }
+
+      public Socket createSocket(InetAddress arg0, int arg1) throws IOException
+      {
+         Socket s = new TestSocket(arg0, arg1, initialSuccesses);
+         log.info("returning " + s);
+         return s;
+      }
+
+      public Socket createSocket(String arg0, int arg1, InetAddress arg2, int arg3) throws IOException, UnknownHostException
+      {
+         Socket s = new TestSocket(arg0, arg1, arg2, arg3, initialSuccesses);
+         log.info("returning " + s);
+         return s;
+      }
+
+      public Socket createSocket(InetAddress arg0, int arg1, InetAddress arg2, int arg3) throws IOException
+      {
+         Socket s = new TestSocket(arg0, arg1, arg2, arg3, initialSuccesses);
+         log.info("returning " + s);
+         return s;
+      }
+   }
+   
+   
+   static class TestSocket extends Socket
+   {
+      int initialSuccesses;
+      
+      public TestSocket(int initialWrites)
+      {
+         this.initialSuccesses = initialWrites;
+      }
+      public TestSocket(String host, int port, int initialWrites) throws UnknownHostException, IOException
+      {
+         super(host, port);
+         this.initialSuccesses = initialWrites;
+      }
+      public TestSocket(InetAddress address, int port, int initialWrites) throws IOException
+      {
+         super(address, port);
+         this.initialSuccesses = initialWrites;
+      }
+      public TestSocket(String host, int port, InetAddress localAddr, int localPort, int initialWrites) throws IOException
+      {
+         super(host, port, localAddr, localPort);
+         this.initialSuccesses = initialWrites;
+      }
+      public TestSocket(InetAddress address, int port, InetAddress localAddr, int localPort, int initialWrites) throws IOException
+      {
+         super(address, port, localAddr, localPort);
+         this.initialSuccesses = initialWrites;
+      }
+      public OutputStream getOutputStream() throws IOException
+      {
+         return new TestOutputStream(super.getOutputStream(), initialSuccesses);
+      }
+      public String toString()
+      {
+         return "TestSocket[" + getLocalPort() + "->" + getPort() + "]";
+      }
+   }
+   
+   
+   static class TestOutputStream extends OutputStream
+   {
+      public static int counter;
+      public static boolean threwException;
+      
+      OutputStream os;
+      boolean closed;
+      int initialWrites;
+      boolean doThrow = true;
+      
+      public TestOutputStream(OutputStream os, int initialWrites)
+      {
+         this.os = os;
+         this.initialWrites = initialWrites;
+      }
+      public void write(int b) throws IOException
+      {
+         if (doThrow && ++counter == initialWrites)
+         {
+            log.info("throwing " + exceptionToThrow);
+            threwException = true;
+            throw exceptionToThrow;
+         }
+         os.write(b);
+      }
+      public void write(byte b[], int off, int len) throws IOException
+      {
+         log.info("TestOutputStream: counter = " + (counter + 1) + ", initialWrites = " + initialWrites);
+         if (++counter == initialWrites)
+         {
+            log.info("throwing " + exceptionToThrow);
+            threwException = true;
+            throw exceptionToThrow;
+         }
+         log.info(this + " calling write()");
+         doThrow = false;
+         os.write(b, off, len);
+         os.flush();
+         doThrow = true;
+         log.info(this + " back from write()");
+      }
+   }
+}
\ No newline at end of file



More information about the jboss-remoting-commits mailing list