Fixed alternate stream bugs

This commit is contained in:
Phil Nash
2012-09-26 18:36:58 +01:00
parent 60fb60f5e0
commit c4160e9ef8
5 changed files with 43 additions and 22 deletions

View File

@@ -11,6 +11,7 @@
#include "catch_test_spec.h"
#include "catch_context.h"
#include "catch_interfaces_config.h"
#include "catch_stream.hpp"
#include <memory>
#include <vector>
@@ -79,19 +80,17 @@ namespace Catch {
public:
Config()
: m_streambuf( NULL ),
m_os( std::cout.rdbuf() )
: m_os( std::cout.rdbuf() )
{}
Config( const ConfigData& data )
: m_data( data ),
m_streambuf( NULL ),
m_os( std::cout.rdbuf() )
{}
virtual ~Config() {
m_os.rdbuf( std::cout.rdbuf() );
delete m_streambuf;
m_stream.release();
}
void setFilename( const std::string& filename ) {
@@ -131,10 +130,10 @@ namespace Catch {
}
void useStream( const std::string& streamName ) {
std::streambuf* newBuf = createStreamBuf( streamName );
setStreamBuf( newBuf );
delete m_streambuf;
m_streambuf = newBuf;
Stream stream = createStream( streamName );
setStreamBuf( stream.streamBuf );
m_stream.release();
m_stream = stream;
}
void addTestSpec( const std::string& testSpec ) {
@@ -166,7 +165,7 @@ namespace Catch {
ConfigData m_data;
// !TBD Move these out of here
std::streambuf* m_streambuf;
Stream m_stream;
mutable std::ostream m_os;
};