View Javadoc
   /************************************************************
   *                     Copyright                            *
   * Portions of this software are Copyright (c) 1993 - 2002, *
   * Chad Z. Hower (Kudzu) and the Indy Pit Crew              *
   *  - http://www.nevrona.com/Indy/                          *
   ************************************************************/
   package org.indy;
   
   import java.io.BufferedReader;
  import java.io.DataOutputStream;
  import java.io.File;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.InputStreamReader;
  import java.io.PrintWriter;
  
  import java.net.ServerSocket;
  import java.net.Socket;
  
  import java.util.HashMap;
  import java.util.Map;
  
  import junit.framework.TestCase;
  
  import org.indy.io.IOHandler;
  import org.indy.io.IOHandlerSocket;
  import org.indy.io.IndyIOException;
  import org.indy.io.ReadTimedOutException;
  
  import org.indy.util.StringList;
  
  
  /***
   * DOCUMENT ME!
   *
   * @version $Revision$
   * @author $author$
   */
  public class TestConnection extends TestCase {
    private Map connectionData = new HashMap();
  
    /*** DOCUMENT ME! */
    public ServerExecute dumbServer = new ServerExecute() {
      public void execute(Socket s) throws IOException {
        s.getInputStream().read();
      }
    };
  
    {
      connectionData.put("host", "127.0.0.1");
      connectionData.put("port", "6666");
    }
  
    /***
     * Creates a new TestConnection object.
     *
     * @param s DOCUMENT ME!
     */
    public TestConnection(String s) {
      super(s);
    }
  
    /***
     * DOCUMENT ME!
     */
    protected void setUp() {
    }
  
    /***
     * DOCUMENT ME!
     */
    protected void tearDown() {
    }
  
    /***
     * DOCUMENT ME!
     */
    public void testReadLineWait() {
      ServerFixture server = new ServerFixture(6666, dumbServer);
      server.start();
  
      IOHandlerSocket ioHandler = new IOHandlerSocket();
  
      Connection connection = new Connection(ioHandler);
  
      try {
        ioHandler.connectClient(connectionData, 0);
        connection.setReadTimeOut(10);
  
        long start = System.currentTimeMillis();
  
        try {
          connection.readLineWait(20);
        }
         catch (ReadTimedOutException e) {
        }
  
        long finish = System.currentTimeMillis();
  
       assertTrue("At least 20 timeout cycles should have elapsed", 
                  (finish - start) >= 200);
     }
      catch (IndyIOException ex) {
       fail(ex.getMessage());
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testCapture() {
     final String line1 = "This is Line 1";
     final String line2 = "This is Line 2";
     final String line3 = "This is Line 3";
 
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         PrintWriter pw = new PrintWriter(s.getOutputStream(), true);
         pw.println(line1);
         pw.println(line2);
         pw.println(line3);
         pw.println(":");
         s.getInputStream().read();
       }
     });
     server.start();
 
     IOHandlerSocket ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
     StringList strings = new StringList();
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       int intRet = connection.capture(strings, ":", false);
       assertEquals("String list has 3 entries", 3, strings.size());
       assertEquals("1st OK", line1, strings.get(0));
       assertEquals("2nd OK", line2, strings.get(1));
       assertEquals("3rd OK", line3, strings.get(2));
     }
      catch (Exception e) {
       fail("Exception thrown:  " + e);
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testGetResponse() {
     final int response = 123;
 
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         RFCReply reply = new RFCReply();
         reply.setReply(123, "Hello from a unit test");
         new PrintWriter(s.getOutputStream(), true).println(reply.toString());
       }
     });
     server.start();
 
     IOHandler ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       try {
         connection.getResponse(response);
       }
        catch (ProtocolException ex) {
         fail(ex.getMessage());
       }
     }
      catch (IndyIOException ex) {
       fail(ex.getMessage());
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testReadChar() {
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         DataOutputStream dos = new DataOutputStream(s.getOutputStream());
 
         for (char c = 'a'; c < 'z'; c++) {
           dos.writeChar(c);
         }
 
         dos.writeChar('a');
         dos.flush();
       }
     });
     server.start();
 
     IOHandler ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       for (char c = 'a'; c < 'z'; c++) {
         char res = connection.readChar();
         assertEquals("Read Chars", c, res);
       }
     }
      catch (Exception e) {
       fail(e.getMessage());
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testAllData() {
     final String data = "This is some data. Not very much. But It can be repeated, oh yeah";
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         PrintWriter pw = new PrintWriter(s.getOutputStream());
 
         for (int i = 0; i < 4; i++) {
           pw.print(data);
         }
 
         pw.flush();
         s.close();
       }
     });
     server.start();
 
     IOHandler ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       String res = connection.allData();
       assertEquals("Check Data Length", data.length() * 4, res.length());
 
       StringBuffer sb = new StringBuffer();
       sb.append(data);
       sb.append(data);
       sb.append(data);
       sb.append(data);
       assertEquals("Check Data", sb.toString(), res);
     }
      catch (Exception e) {
       fail("Exception thrown:  " + e);
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testReadLong() {
     final long data = System.currentTimeMillis();
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         DataOutputStream dos = new DataOutputStream(s.getOutputStream());
         dos.writeLong(data);
       }
     });
 
     IOHandler ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       long longRet = connection.readLong();
       assertEquals(data, longRet);
     }
      catch (Exception e) {
       fail("Exception thrown:  " + e);
     }
      finally {
       ioHandler.close();
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testReadStream() {
     final String data = "This is some data to go into a stream";
 
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         new PrintWriter(s.getOutputStream(), true).println(data);
       }
     });
 
     IOHandler ioHandler = new IOHandlerSocket();
     Connection connection = new Connection(ioHandler);
 
     try {
       InputStream inputStreamRet = connection.readStream(data.length() + 1, 
                                                          false);
       assertEquals(data, 
                    new BufferedReader(new InputStreamReader(inputStreamRet)).readLine());
     }
      catch (Exception e) {
       fail("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteStrings() {
     String[] strings = new String[3];
     strings[0] = "This is the first string";
     strings[1] = "This is the second string";
     strings[2] = "This is the third string";
 
     final StringList data = new StringList();
     data.add(strings[0]);
     data.add(strings[1]);
     data.add(strings[2]);
 
     ServerFixture server = new ServerFixture(6666, 
                                              new ServerExecute() {
       public void execute(Socket s) throws IOException {
         try {
           IOHandler ioHandler = new IOHandlerSocket(s);
           Connection c = new Connection(ioHandler);
           c.writeStrings(data, true);
         }
          catch (IndyIOException ex) {
           ex.printStackTrace();
         }
       }
     });
     server.start();
 
     IOHandlerSocket ioHandler = new IOHandlerSocket();
 
     try {
       ioHandler.connectClient(connectionData, 0);
 
       Connection c = new Connection(ioHandler);
 
       int lineCount = c.readInt();
       assertEquals("Test line count", 3, lineCount);
 
       for (int i = 0; i < lineCount; i++) {
         assertEquals("Test lines", strings[i], c.readLine());
       }
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteStream() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     InputStream stream1 = null
     ;
     int len2 = 0;
     boolean writeByteCount3 = true;
 
     try {
       connection.writeStream(stream1, len2, writeByteCount3);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteShort() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     short value1 = 0;
 
     try {
       connection.writeShort(value1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteRFCStrings() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     StringList strings1 = null
     ;
 
     try {
       connection.writeRFCStrings(strings1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteRFCReply() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     RFCReply reply1 = null
     ;
 
     try {
       connection.writeRFCReply(reply1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteLong() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     long value1 = 0;
 
     try {
       connection.writeLong(value1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteLine() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     String line1 = "STRING0";
 
     try {
       connection.writeLine(line1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteInt() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     int value1 = 0;
 
     try {
       connection.writeInt(value1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteHeader() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     StringList header1 = null
     ;
 
     try {
       connection.writeHeader(header1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteFile() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     File f1 = null
     ;
 
     try {
       connection.writeFile(f1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWriteBuffer() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     byte[] buf1 = null
     ;
 
     try {
       connection.writeBuffer(buf1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   /***
    * DOCUMENT ME!
    */
   public void testWrite() {
     IOHandler val1 = null
     ;
     Connection connection = new Connection(val1);
     String s1 = "STRING0";
 
     try {
       connection.write(s1);
 
       /*** @todo:  Insert test code here.  Use assertEquals(), for example. */
     }
      catch (Exception e) {
       System.err.println("Exception thrown:  " + e);
     }
   }
 
   private interface ServerExecute {
     void execute(Socket s) throws IOException;
   }
 
   private class ServerFixture extends Thread {
     ServerExecute executor;
     int port;
     private volatile boolean stopped = false;
 
     public ServerFixture(int port, ServerExecute e) {
       executor = e;
       this.port = port;
     }
 
     public void stopServer() {
       stopped = true;
     }
 
     public void run() {
       ServerSocket ss = null;
 
       try {
         ss = new ServerSocket(port);
 
         Socket s = ss.accept();
         executor.execute(s);
       }
        catch (IOException ex) {
         ex.printStackTrace();
       }
        finally {
         try {
           ss.close();
         }
          catch (IOException ex) {
         }
       }
     }
   }
 }