beast_Journal_test.cpp

#include <xrpl/beast/unit_test/suite.h>
#include <xrpl/beast/utility/Journal.h>
 
#include <string>
 
namespace beast {
 
class Journal_test : public unit_test::Suite
{
public:
    class TestSink : public Journal::Sink
    {
    private:
        int count_{0};
 
    public:
        TestSink() : Sink(Severity::Warning, false)
        {
        }
 
        [[nodiscard]] int
        count() const
        {
            return count_;
        }
 
        void
        reset()
        {
            count_ = 0;
        }
 
        void
        write(Severity level, std::string const&) override
        {
            if (level >= threshold())
                ++count_;
        }
 
        void
        writeAlways(Severity level, std::string const&) override
        {
            ++count_;
        }
    };
 
    void
    run() override
    {
        TestSink sink;
 
        sink.threshold(Severity::Info);
 
        Journal const j(sink);
 
        j.trace() << " ";
        BEAST_EXPECT(sink.count() == 0);
        j.debug() << " ";
        BEAST_EXPECT(sink.count() == 0);
        j.info() << " ";
        BEAST_EXPECT(sink.count() == 1);
        j.warn() << " ";
        BEAST_EXPECT(sink.count() == 2);
        j.error() << " ";
        BEAST_EXPECT(sink.count() == 3);
        j.fatal() << " ";
        BEAST_EXPECT(sink.count() == 4);
 
        sink.reset();
 
        sink.threshold(Severity::Debug);
 
        j.trace() << " ";
        BEAST_EXPECT(sink.count() == 0);
        j.debug() << " ";
        BEAST_EXPECT(sink.count() == 1);
        j.info() << " ";
        BEAST_EXPECT(sink.count() == 2);
        j.warn() << " ";
        BEAST_EXPECT(sink.count() == 3);
        j.error() << " ";
        BEAST_EXPECT(sink.count() == 4);
        j.fatal() << " ";
        BEAST_EXPECT(sink.count() == 5);
    }
};
 
BEAST_DEFINE_TESTSUITE(Journal, beast, beast);
 
}  // namespace beast