StdoutChannelTest.h

#ifndef MANTID_STDOUTCHANNELTEST_H_
#define MANTID_STDOUTCHANNELTEST_H_

#include <cxxtest/TestSuite.h>

#include "MantidKernel/StdoutChannel.h"
#include "MantidKernel/Logger.h"
#include <iostream>
#include <Poco/Logger.h>
#include <Poco/NullChannel.h>
#include <Poco/AutoPtr.h>

using namespace Mantid::Kernel;

class StdoutChannelTest : public CxxTest::TestSuite {
public:
  void testContructor() { TS_ASSERT_THROWS_NOTHING(Poco::StdoutChannel a;) }

  void testLogMessage() {
    // Save root channel
    Poco::Channel *rootChannel = Poco::Logger::root().getChannel();

    std::stringstream obuffer, lbuffer;
    // Save cout's and clog's buffers here
    std::streambuf *obuf = std::cout.rdbuf();
    std::streambuf *lbuf = std::clog.rdbuf();
    // Redirect cout to buffer or any other ostream
    std::cout.rdbuf(obuffer.rdbuf());
    std::clog.rdbuf(lbuffer.rdbuf());
    // root logger has empty name
    Logger log("");

    // Test null channel first
    Poco::AutoPtr<Poco::NullChannel> nullChannel(new Poco::NullChannel);
    Poco::Logger::root().setChannel(nullChannel);

    log.error() << "Error Message 1" << std::endl;
    // cout and clog should be empty
    TS_ASSERT_EQUALS(obuffer.str(), "");
    TS_ASSERT_EQUALS(lbuffer.str(), "");
    obuffer.str("");
    lbuffer.str("");

    // Test console channel
    Poco::AutoPtr<Poco::ConsoleChannel> consoleChannel(
        new Poco::ConsoleChannel);
    Poco::Logger::root().setChannel(consoleChannel);
    log.error() << "Error Message 2" << std::endl;
    // the error should be in std::clog (or std:err)
    TS_ASSERT_EQUALS(obuffer.str(), "");
    TS_ASSERT_EQUALS(lbuffer.str(), "Error Message 2\n");
    obuffer.str("");
    lbuffer.str("");

    // Test std channel
    Poco::AutoPtr<Poco::StdoutChannel> stdoutChannel(new Poco::StdoutChannel);
    Poco::Logger::root().setChannel(stdoutChannel);
    log.error() << "Error Message 3" << std::endl;
    // the error should be in std::cout
    TS_ASSERT_EQUALS(obuffer.str(), "Error Message 3\n");
    TS_ASSERT_EQUALS(lbuffer.str(), "");

    // When done redirect cout to its old self
    std::cout.rdbuf(obuf);
    std::clog.rdbuf(lbuf);
    // set back the channel on root
    Poco::Logger::root().setChannel(rootChannel);
  }
};

#endif /*MANTID_STDOUTCHANNELTEST_H_*/