#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