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)
{
#ifdef CppUnit2Gtest
#define ExpectEqual(a, b)                                                                                                                            \
    EXPECT_EQ(a, b);                                                                                                                                 \
    if (!(a == b))                                                                                                                                   \
    throw std::logic_error("bad assert")
#define Fail(msg)                                                                                                                                    \
    ADD_FAILURE() << msg;                                                                                                                            \
    throw std::logic_error("bad assert")
#else
#define ExpectEqual(a, b) CPPUNIT_ASSERT_EQUAL(a, b)
#define Fail(msg) CPPUNIT_FAIL(msg)
#endif
    m_os.rdbuf(m_regularOutputBuffer);
    const std::string actualOutput(m_buffer.str());
    if (m_customCheck) {
        m_customCheck(actualOutput);
        return;
    }
    if (m_alternativeOutput.empty()) {
        ExpectEqual(m_expectedOutput, actualOutput);
        return;
    }
    if (m_expectedOutput != actualOutput && m_alternativeOutput != actualOutput) {
        using namespace CppUtilities;
        Fail("Output is not either \"" % m_expectedOutput % "\" or \"" % m_alternativeOutput % "\". Got instead:\n" + actualOutput);
    }
#undef Fail
#undef ExpectEqual
}
 
} // namespace CppUtilities
 
#endif // TESTUTILS_OUTPUTCHECK_H