/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "BufferedOutputStreamTest.h"
#include <decaf/io/ByteArrayOutputStream.h>
#include <decaf/io/ByteArrayInputStream.h>

using namespace std;
using namespace decaf;
using namespace decaf::lang;
using namespace decaf::lang::exceptions;
using namespace decaf::io;
using namespace decaf::util;

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testConstructor1() {

    try {
        MyOutputStream myStream;
        BufferedOutputStream os( &myStream );
        os.write( (unsigned char*)&testString[0], 0, 500 );
    } catch( IOException& e ) {
        CPPUNIT_FAIL("Constrcutor test failed");
    }
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testConstructor2() {

    try {
        MyOutputStream myStream;
        BufferedOutputStream os( &myStream, (std::size_t)1024 );
        os.write( (unsigned char*)&testString[0], 0, 500 );
    } catch( IOException& e) {
        CPPUNIT_FAIL("IOException during Constrcutor test");
    }
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testFlush() {

    try {

        ByteArrayOutputStream myStream;
        BufferedOutputStream os( &myStream, (std::size_t)600 );
        os.write( (unsigned char*)&testString[0], 0, 500 );
        os.flush();
        CPPUNIT_ASSERT_MESSAGE("Bytes not written after flush",
                500 == myStream.size() );
    } catch( IOException& e) {
        CPPUNIT_FAIL("Flush test failed");
    }
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWrite() {

    try {

        ByteArrayOutputStream baos;
        BufferedOutputStream os( &baos, (std::size_t)512 );
        os.write( (unsigned char*)&testString[0], 0, 500 );

        ByteArrayInputStream bais1( baos.toByteArray(), baos.size() );
        CPPUNIT_ASSERT_MESSAGE( "Bytes written, not buffered", 0 == bais1.available());
        os.flush();
        ByteArrayInputStream bais2( baos.toByteArray(), baos.size() );
        CPPUNIT_ASSERT_MESSAGE( "Bytes not written after flush", 500 == bais2.available() );
        os.write( (unsigned char*)&testString[500], 0, 514 );
        ByteArrayInputStream bais3( baos.toByteArray(), baos.size() );
        CPPUNIT_ASSERT_MESSAGE( "Bytes not written when buffer full",
                                bais3.available() >= 1000);
        unsigned char wbytes[1014] = {0};
        bais3.read( wbytes, 0, 1013 );

        CPPUNIT_ASSERT_MESSAGE(
            "Incorrect bytes written",
            testString.substr(0, 1012) == string( (const char*)wbytes ) );

    } catch( IOException& e) {
        CPPUNIT_FAIL("write test failed: ");
    }
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteException() {

    BufferedOutputStream bos( new ByteArrayOutputStream(), true );
    unsigned char* nullByteArray = NULL;
    unsigned char byteArray[10];

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw NullPointerException",
        bos.write( nullByteArray, 0, (std::size_t)1 ),
        NullPointerException );

    bos.write( byteArray, 0, 0 );
    bos.write( byteArray, 0, 1 );
    bos.write( byteArray, 0, 10 );
    bos.close();

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw IOException",
        bos.write( nullByteArray, 0, (std::size_t)1 ),
        IOException );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteNullStreamNullArray() {

    BufferedOutputStream bos( NULL );
    unsigned char* nullByteArray = NULL;

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw NullPointerException",
        bos.write( nullByteArray, 0, (std::size_t)1 ),
        IOException );

    CPPUNIT_ASSERT_NO_THROW_MESSAGE(
        "should not throw NullPointerException",
        bos.write( nullByteArray, 0, (std::size_t)0 ) );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteNullStreamNullArraySize() {

    BufferedOutputStream bos( NULL, (std::size_t)1 );
    unsigned char* nullByteArray = NULL;

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw IOException",
        bos.write( nullByteArray, 0, (std::size_t)1 ),
        IOException );

    CPPUNIT_ASSERT_NO_THROW_MESSAGE(
        "should not throw NullPointerException",
        bos.write( nullByteArray, 0, (std::size_t)0 ) );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteNullStream() {

    BufferedOutputStream bos( NULL );
    unsigned char byteArray[10];

    CPPUNIT_ASSERT_NO_THROW_MESSAGE(
        "should not throw IOException",
        bos.write( byteArray, 0, (std::size_t)0 ) );

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw IOException",
        bos.write( byteArray, 0, (std::size_t)1 ),
        IOException );

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw IOException",
        bos.write( byteArray, 0, (std::size_t)10 ),
        IOException );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteNullStreamSize() {

    BufferedOutputStream bos( NULL, (std::size_t)1 );
    unsigned char byteArray[10];

    bos.write( byteArray, 0, 0 );

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw NullPointerException",
        bos.write( byteArray, 0, 2 ),
        IOException );

    CPPUNIT_ASSERT_THROW_MESSAGE(
        "should throw NullPointerException",
        bos.write( byteArray, 0, 10 ),
        IOException );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testWriteI() {

    try {

        ByteArrayOutputStream baos;
        BufferedOutputStream os( &baos );
        os.write('t');
        ByteArrayInputStream bais1( baos.toByteArray(), baos.size() );
        CPPUNIT_ASSERT_MESSAGE( "Byte written, not buffered", 0 == bais1.available() );
        os.flush();

        ByteArrayInputStream bais2( baos.toByteArray(), baos.size() );
        CPPUNIT_ASSERT_MESSAGE( "Byte not written after flush", 1 == bais2.available() );
        unsigned char wbytes[10];
        bais2.read( wbytes, 0, 1 );
        CPPUNIT_ASSERT_MESSAGE( "Incorrect byte written", 't' == wbytes[0] );

    } catch( IOException& e) {
        CPPUNIT_FAIL("Write test failed");
    }
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testSmallerBuffer(){

    MyOutputStream myStream;
    BufferedOutputStream bufStream( &myStream, (std::size_t)1 );

    const char* buffer = myStream.getBuffer();

    bufStream.write( (unsigned char)'T' );
    // Should not be written yet.
    CPPUNIT_ASSERT( strcmp( buffer, "" ) == 0 );

    bufStream.write( (unsigned char)'E' );
    // This time the T should have been written.
    CPPUNIT_ASSERT( strcmp( buffer, "T" ) == 0 );

    bufStream.write( (unsigned char*)"ST", 0, 2 );
    // This time the ES should have been written.
    CPPUNIT_ASSERT( strcmp( buffer, "TES" ) == 0 );

    bufStream.flush();
    CPPUNIT_ASSERT( strcmp( buffer, "TEST" ) == 0 );
}

////////////////////////////////////////////////////////////////////////////////
void BufferedOutputStreamTest::testBiggerBuffer(){

    MyOutputStream myStream;
    BufferedOutputStream bufStream( &myStream, (std::size_t)10 );

    const char* buffer = myStream.getBuffer();

    bufStream.write( (unsigned char*)"TEST", 0, 4 );

    // Should not be written yet.
    CPPUNIT_ASSERT( strcmp( buffer, "" ) == 0 );

    bufStream.flush();
    CPPUNIT_ASSERT( strcmp( buffer, "TEST" ) == 0 );

    bufStream.write( (unsigned char*)"TEST", 0, 4 );
    bufStream.write( (unsigned char*)"12345678910", 0, 11);

    CPPUNIT_ASSERT( strcmp( buffer, "TESTTEST123456" ) == 0 );

    bufStream.flush();
    CPPUNIT_ASSERT( strcmp( buffer, "TESTTEST12345678910" ) == 0 );
}