/*
 * 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.
 */

#ifndef ACTIVEMQ_IO_BYTEARRAYINPUTSTREAMTEST_H_
#define ACTIVEMQ_IO_BYTEARRAYINPUTSTREAMTEST_H_

#include <cppunit/TestFixture.h>
#include <cppunit/extensions/HelperMacros.h>

#include <activemq/io/ByteArrayInputStream.h>

namespace activemq{
namespace io{

   class ByteArrayInputStreamTest : public CppUnit::TestFixture 
   {
     CPPUNIT_TEST_SUITE( ByteArrayInputStreamTest );
     CPPUNIT_TEST( testStream );
     CPPUNIT_TEST_SUITE_END();

   public:
   
   	ByteArrayInputStreamTest() {}

   	virtual ~ByteArrayInputStreamTest() {}

      void testStream()
      {
         std::vector<unsigned char> testBuffer;
         
         testBuffer.push_back('t');
         testBuffer.push_back('e');
         testBuffer.push_back('s');
         testBuffer.push_back('t');

         ByteArrayInputStream stream_a(&testBuffer[0], testBuffer.size());
         
         CPPUNIT_ASSERT( stream_a.available() == 4 );
         
         char a = stream_a.read();
         char b = stream_a.read();
         char c = stream_a.read();
         char d = stream_a.read();

         //printf("a=%c, b=%c, c=%c, d=%c\n", a, b, c, d );
         CPPUNIT_ASSERT( a == 't' && b == 'e' && c == 's' && d == 't' );
         CPPUNIT_ASSERT( stream_a.available() == 0 );

         testBuffer.push_back('e');
         
         stream_a.setByteArray(&testBuffer[0], testBuffer.size());
         
         CPPUNIT_ASSERT( stream_a.available() == 5 );
         
         unsigned char* buffer = new unsigned char[6];
         
         buffer[5] = '\0';
      
         CPPUNIT_ASSERT( stream_a.read(buffer, 5) == 5 );         
         CPPUNIT_ASSERT( std::string((const char*)buffer) == std::string("teste") );
         CPPUNIT_ASSERT( stream_a.available() == 0 );
         
         stream_a.setByteArray(&testBuffer[0], testBuffer.size());

         memset(buffer, 0, 6);

         CPPUNIT_ASSERT( stream_a.read(buffer, 3) == 3 );
         CPPUNIT_ASSERT( stream_a.read(&buffer[3], 2) == 2 );
         CPPUNIT_ASSERT( std::string((const char*)buffer) == std::string("teste") );
         
         stream_a.close();
         
         delete [] buffer;
      }
   };

}}

#endif /*ACTIVEMQ_IO_BYTEARRAYINPUTSTREAMTEST_H_*/