/* * 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 #include 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 ); }