outputcheck.h

Go to the documentation of this file.

#ifndef TESTUTILS_OUTPUTCHECK_H
#define TESTUTILS_OUTPUTCHECK_H
 
#include "../conversion/stringbuilder.h"
 
#include <cppunit/extensions/HelperMacros.h>
 
#include <functional>
#include <iostream>
#include <sstream>
#include <string>
 
namespace CppUtilities {
 
class OutputCheck {
public:
    OutputCheck(const std::string &expectedOutput, std::ostream &os = std::cout);
    OutputCheck(std::string &&expectedOutput, std::string &&alternativeOutput, std::ostream &os = std::cout);
    OutputCheck(std::function<void(const std::string &output)> &&customCheck, std::ostream &os = std::cout);
    ~OutputCheck() noexcept(false);
 
private:
    std::ostream &m_os;
    const std::function<void(const std::string &output)> m_customCheck;
    const std::string m_expectedOutput;
    const std::string m_alternativeOutput;
    std::stringstream m_buffer;
    std::streambuf *const m_regularOutputBuffer;
};
 
inline OutputCheck::OutputCheck(const std::string &expectedOutput, std::ostream &os)
    : m_os(os)
    , m_expectedOutput(expectedOutput)
    , m_buffer()
    , m_regularOutputBuffer(os.rdbuf(m_buffer.rdbuf()))
{
}
 
inline OutputCheck::OutputCheck(std::string &&expectedOutput, std::string &&alternativeOutput, std::ostream &os)
    : m_os(os)
    , m_expectedOutput(expectedOutput)
    , m_alternativeOutput(alternativeOutput)
    , m_buffer()
    , m_regularOutputBuffer(os.rdbuf(m_buffer.rdbuf()))
{
}
 
inline OutputCheck::OutputCheck(std::function<void(const std::string &)> &&customCheck, std::ostream &os)
    : m_os(os)
    , m_customCheck(customCheck)
    , m_buffer()
    , m_regularOutputBuffer(os.rdbuf(m_buffer.rdbuf()))
{
}
 
inline OutputCheck::~OutputCheck() noexcept(false)
{
    m_os.rdbuf(m_regularOutputBuffer);
    const std::string actualOutput(m_buffer.str());
    if (m_customCheck) {
        m_customCheck(actualOutput);
        return;
    }
    if (m_alternativeOutput.empty()) {
        CPPUNIT_ASSERT_EQUAL(m_expectedOutput, actualOutput);
        return;
    }
    if (m_expectedOutput != actualOutput && m_alternativeOutput != actualOutput) {
        using namespace CppUtilities;
        CPPUNIT_FAIL("Output is not either \"" % m_expectedOutput % "\" or \"" % m_alternativeOutput % "\". Got instead:\n" + actualOutput);
    }
}
 
} // namespace CppUtilities
 
#endif // TESTUTILS_OUTPUTCHECK_H