diff --git a/tests/NullOutputStream.h b/tests/NullOutputStream.h new file mode 100644 index 0000000..de9f99f --- /dev/null +++ b/tests/NullOutputStream.h @@ -0,0 +1,44 @@ +#ifndef __PLASP__TESTS__NULL_OUTPUT_STREAM_H +#define __PLASP__TESTS__NULL_OUTPUT_STREAM_H + +#include + +//////////////////////////////////////////////////////////////////////////////////////////////////// +// +// NullOutputStream +// +//////////////////////////////////////////////////////////////////////////////////////////////////// + +class NullStreamBuffer : public std::streambuf +{ + private: + char dummyBuffer[64]; + + protected: + virtual int overflow(int c) + { + setp(dummyBuffer, dummyBuffer + sizeof(dummyBuffer)); + return (c == traits_type::eof()) ? '\0' : c; + } +}; + +//////////////////////////////////////////////////////////////////////////////////////////////////// + +class NullOutputStream : public std::ostream +{ + public: + NullOutputStream() + : std::ostream(&m_nullStreamBuffer) + { + } + + NullStreamBuffer *rdbuf() const + { + return &m_nullStreamBuffer; + } + + private: + mutable NullStreamBuffer m_nullStreamBuffer; +}; + +#endif diff --git a/tests/TestPDDLTranslation.cpp b/tests/TestPDDLTranslation.cpp index 4ccb89a..4dbf459 100644 --- a/tests/TestPDDLTranslation.cpp +++ b/tests/TestPDDLTranslation.cpp @@ -1,7 +1,6 @@ #include -#include -#include +#include #include @@ -11,15 +10,17 @@ #include -boost::iostreams::stream nullStream((boost::iostreams::null_sink())); +#include "NullOutputStream.h" + const pddl::Context::WarningCallback ignoreWarnings = [](const auto &, const auto &){}; //////////////////////////////////////////////////////////////////////////////////////////////////// TEST_CASE("[PDDL translation] Former issues are fixed", "[PDDL translation]") { - // TODO: refactor - colorlog::Logger logger(nullStream, nullStream); + NullOutputStream nullOutputStream; + + colorlog::Logger logger(nullOutputStream, nullOutputStream); pddl::Tokenizer tokenizer; pddl::Context context(std::move(tokenizer), ignoreWarnings);