/*
 * $Id$
 *
 * Copyright (C) 2003-2015 JNode.org
 *
 * This library 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 library 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 library; If not, write to the Free Software Foundation, Inc., 
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
package org.jnode.test.shell.io;

import java.io.StringWriter;
import java.nio.charset.MalformedInputException;

import org.jnode.util.WriterOutputStream;
import org.junit.Assert;
import org.junit.Test;

public class WriterOutputStreamTest {

    @Test
    public void testEmpty() throws Exception {
        String LINE = "";
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        byte[] buffer = LINE.getBytes();
        wos.write(buffer);
        wos.flush();
        Assert.assertEquals(LINE, sw.getBuffer().toString());
    }

    @Test
    public void testLine() throws Exception {
        String LINE = "The quick brown fox jumped over the lazy doc";
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        byte[] buffer = LINE.getBytes();
        wos.write(buffer);
        wos.flush();
        Assert.assertEquals(LINE, sw.getBuffer().toString());
    }

    @Test
    public void testByteAtATime() throws Exception {
        String LINE = "The quick brown fox jumped over the lazy doc";
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        byte[] buffer = LINE.getBytes();
        for (byte b : buffer) {
            wos.write(b);
        }
        wos.flush();
        Assert.assertEquals(LINE, sw.getBuffer().toString());
    }

    @Test
    public void testByteAtATimeWithFlushes() throws Exception {
        String LINE = "The quick brown fox jumped over the lazy doc";
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        byte[] buffer = LINE.getBytes();
        for (int i = 0; i < buffer.length; i++) {
            wos.write(buffer[i]);
            wos.flush();
            Assert.assertEquals(LINE.charAt(i), sw.getBuffer().charAt(i));
        }
        Assert.assertEquals(LINE, sw.getBuffer().toString());
    }

    @Test
    public void testUnicode() throws Exception {
        char[] chars = new char[8192];
        for (int i = 0; i < chars.length; i++) {
            chars[i] = (char) i;
        }
        byte[] buffer = new String(chars).getBytes();
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        wos.write(buffer);
        wos.flush();
        StringBuffer sb = sw.getBuffer();
        Assert.assertEquals(chars.length, sb.length());
        for (int i = 0; i < chars.length; i++) {
            Assert.assertEquals(chars[i], sb.charAt(i));
        }
    }

    @Test
    public void testBadUnicode() throws Exception {
        byte[] BAD = new byte[] {(byte) 0x80};
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        try {
            wos.write(BAD);
            wos.flush();
            Assert.fail("no exception thrown");
        } catch (MalformedInputException ex) {
            // expected
        }
    }

    @Test
    public void testBadUnicode2() throws Exception {
        byte[] BAD = new byte[] {(byte) 'h', (byte) 'i', (byte) 0x80};
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        try {
            wos.write(BAD);
            wos.flush();
            Assert.fail("no exception thrown");
        } catch (MalformedInputException ex) {
            // expected
            Assert.assertEquals("hi", sw.getBuffer().toString());
        }
    }

    @Test
    public void testBadUnicode3() throws Exception {
        byte[] BAD = new byte[] {(byte) 'h', (byte) 'i', (byte) 0xc2, (byte) 0x00};
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        try {
            wos.write(BAD);
            wos.flush();
            Assert.fail("no exception thrown");
        } catch (MalformedInputException ex) {
            // expected
            Assert.assertEquals("hi", sw.getBuffer().toString());
        }
    }

    @Test
    public void testBadUnicode4() throws Exception {
        byte[] BAD = new byte[] {(byte) 'h', (byte) 'i', (byte) 0xc2};
        StringWriter sw = new StringWriter();
        WriterOutputStream wos = new WriterOutputStream(sw, "UTF-8", true);
        wos.write(BAD);
        wos.flush();
        try {
            wos.close();
            Assert.fail("no exception thrown");
        } catch (MalformedInputException ex) {
            // expected
            Assert.assertEquals("hi", sw.getBuffer().toString());
        }
    }
}