package org.apidesign.delegatingwriter;

 import java.io.IOException;

 import java.io.StringWriter;

 import java.io.StringWriter;

 import org.junit.Before;

 import org.junit.Test;

 import static org.junit.Assert.*;

 /** Emulates what goes wrong when delegating directly

  *

  * @author Jaroslav Tulach <jaroslav.tulach@apidesign.org>

  */

 public class BufferedWriterCryptoTest {

     private StringWriter writer;

     public BufferedWriterCryptoTest() {

     }

     @Before

     public void setUp() {

         writer = new StringWriter();

     }

     @Test

     public void testBehaviourOfRealBufferInJDKWorksFine() throws IOException {

         // BEGIN: writer.crypto.ok

         CryptoWriter bufferedWriter = new CryptoWriter(writer);

         bufferedWriter.append("VMS");

         bufferedWriter.flush();

         assertEquals("Converted", "WNT", writer.toString());

         // END: writer.crypto.ok

     }

     @Test

     public void testBehaviourOfBufferThatDelegatesToAppendFails() throws IOException {

         CryptoWriter bufferedWriter = new CryptoWriter(writer, CryptoWriter.Behaviour.DELEGATE_TO_OUT);

         bufferedWriter.append("VMS");

         bufferedWriter.flush();

         assertEquals("This will fail, as the direct delegation from append to " +

             "the underlaying writer will skip all the crypto methods", 

             "WNT", writer.toString()

         );

     }

 }