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