/*

  * Created on Jun 28, 2004

  *

  * To change the template for this generated file go to

  * Window - Preferences - Java - Code Generation - Code and Comments

  */

 package biz.xsoftware.examples.socket;

 import java.io.IOException;

 import java.nio.ByteBuffer;

 import junit.framework.TestCase;

 import biz.xsoftware.mock.CalledMethod;

 import biz.xsoftware.mock.MockObject;

 import biz.xsoftware.mock.MockObjectFactory;

 /**

  * JUnit Suite of TestCases for demonstrating mocking out socket

  * interfaces to test network failures.

  * 

  * @author Dean Hiller

  */

 public class TestExample extends TestCase {

         private SysUnderTest sysUnderTest;

         private MockObject mockSubsys;

         private MockObject mockSocket;

         /**

          * @showcode

          */

         public TestExample(String name) {

                 super(name);

         }

         /**

          * @showcode

          */

         public void setUp() throws Exception {

                 mockSubsys = MockObjectFactory.createMock(OtherSubsystem.class);

                 mockSocket = MockObjectFactory.createMock(TCPSocket.class);

                 sysUnderTest = new SysUnderTest((TCPSocket)mockSocket, (OtherSubsystem)mockSubsys);

         }

         /**

          * @showcode

          */

         public void tearDown() throws IOException {

         }

         /**

          * Test that upon network problems like IOException when writing to server

          * that other subsystems get cleaned up properly.  The simple test below 

          * shows how this can be done and extended to more complicated network

          * recovery algorithms.

          * 

          * @showcode

          */

         public void testNetworkProblems() {

                 mockSocket.addThrowException("write", new IOException("problems writing to server"));

                 ByteBuffer buffer = ByteBuffer.allocate(100);

                 buffer.put(new byte[] { 1, 2, 3, 4});

                 buffer.flip();

                 mockSubsys.addReturnValue("getUserData", buffer);

                 int id = 10;

                 sysUnderTest.sendUserToServer(10);

                 CalledMethod[] methods = mockSubsys.expect("getUserData", "failedToSendUser");

                 CalledMethod getUserData = methods[0];

                 CalledMethod failedToSend = methods[1];

                 assertEquals(new Integer(id), getUserData.getAllParams()[0]);

                 assertEquals(new Integer(id), failedToSend.getParameter(0));

         }

 }