root/library/tests/unittest-cpp/MemoryOutStream.h @ 822

#ifndef UNITTEST_MEMORYOUTSTREAM_H
#define UNITTEST_MEMORYOUTSTREAM_H

#include "Config.h"

#ifndef UNITTEST_USE_CUSTOM_STREAMS

#include <sstream>

namespace UnitTest {

class MemoryOutStream : public std::ostringstream {
public:
        MemoryOutStream() {}
        ~MemoryOutStream() {}
        char const* GetText() const;

private:
        MemoryOutStream ( MemoryOutStream const& );
        void operator = ( MemoryOutStream const& );

        mutable std::string m_text;
};

}

#else

#include <cstddef>

namespace UnitTest {

class MemoryOutStream {
public:
        explicit MemoryOutStream ( int const size = 256 );
        ~MemoryOutStream();

        char const* GetText() const;

        MemoryOutStream& operator << ( char const* txt );
        MemoryOutStream& operator << ( int n );
        MemoryOutStream& operator << ( long n );
        MemoryOutStream& operator << ( unsigned long n );
        MemoryOutStream& operator << ( float f );
        MemoryOutStream& operator << ( double d );
        MemoryOutStream& operator << ( void const* p );
        MemoryOutStream& operator << ( unsigned int s );

        enum { GROW_CHUNK_SIZE = 32 };
        int GetCapacity() const;

private:
        void operator= ( MemoryOutStream const& );
        void GrowBuffer ( int capacity );

        int m_capacity;
        char* m_buffer;
};

}

#endif

#endif