package org.apidesign.delegatingwriter;

import java.io.BufferedWriter;

import java.io.IOException;

import org.junit.Test;

import static org.junit.Assert.*;

/** Emulates what goes wrong when one wants to process really large character 

 * sequence.

 *

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

 */

public class BufferedWriterOnCDImageTest {

    public BufferedWriterOnCDImageTest() {

    }

    @Test

    public void testBehaviourOfRealBufferInJDKObviouslyThisIsGoingToThrowOutOfMemoryError() throws IOException {

        // BEGIN: writer.countcd

        CountingWriter writer = new CountingWriter();

        CDSequence cdImage = new CDSequence();

        BufferedWriter bufferedWriter = new BufferedWriter(writer);

        bufferedWriter.append(cdImage);

        assertEquals(

            "Correct number of writes delegated", 

            cdImage.length(), writer.getCharacterCount()

        );

        // END: writer.countcd

    }

    @Test

    public void testBehaviourOfBufferThatDelegatesToAppend() throws IOException {

        CountingWriter writer = new CountingWriter();

        CDSequence cdImage = new CDSequence();

        BufferedWriter bufferedWriter = new AltBufferedWriter(

            writer, AltBufferedWriter.Behaviour.DELEGATE_TO_OUT

        );

        bufferedWriter.append(cdImage);

        assertEquals(

            "Correct number of writes delegated", 

            cdImage.length(), 

            writer.getCharacterCount()

        );

    }

    @Test

    public void testBehaviourWhenDelegatingConditionallyIsOK() throws IOException {

        CountingWriter writer = new CountingWriter();

        CDSequence cdImage = new CDSequence();

        BufferedWriter bufferedWriter = new AltBufferedWriter(

            writer, AltBufferedWriter.Behaviour.DELEGATE_CONDITIONALLY

        );

        bufferedWriter.append(cdImage);

        assertEquals(

            "Correct number of writes delegated", 

            cdImage.length(), 

            writer.getCharacterCount()

        );

    }

// BEGIN: writer.bigseq

    /** A "lazy" sequence of characters, for example one that can represent

     * content of a CD, read it lazily, do not fit all into memory at once.

     */

    private static final class CDSequence implements CharSequence {

        private final int start;

        private final int end;

        public CDSequence() {

            this(0, 647 * 1024 * 1024);

        }

        private CDSequence(int start, int end) {

            this.start = start;

            this.end = end;

        }

        public int length() {

            return end - start;

        }

// FINISH: writer.bigseq

        public char charAt(int index) {

            int ch = index % ('Z' - 'A' + 1);

            return (char) ('A' + ch);

        }

        public CharSequence subSequence(int start, int end) {

            return new CDSequence(start, end);

        }

        @Override

        public String toString() {

            char[] arr = new char[length()];

            for (int i = 0; i < length(); i++) {

                arr[i] = charAt(i);

            }

            return new String(arr);

        }

    } // end of CharSequence

}