#include <xrpl/beast/unit_test/suite.h>
#include <xrpl/ledger/PendingSaves.h>

namespace xrpl::test {

struct PendingSaves_test : public beast::unit_test::Suite
{
    void
    testSaves()
    {
        PendingSaves ps;

        // Basic test
        BEAST_EXPECT(!ps.pending(0));
        BEAST_EXPECT(!ps.startWork(0));
        BEAST_EXPECT(ps.shouldWork(0, true));
        BEAST_EXPECT(ps.startWork(0));
        BEAST_EXPECT(ps.pending(0));
        BEAST_EXPECT(!ps.shouldWork(0, false));
        ps.finishWork(0);
        BEAST_EXPECT(!ps.pending(0));

        // Test work stealing
        BEAST_EXPECT(ps.shouldWork(0, false));
        BEAST_EXPECT(ps.pending(0));
        BEAST_EXPECT(ps.shouldWork(0, true));
        BEAST_EXPECT(ps.pending(0));
        BEAST_EXPECT(ps.startWork(0));
        BEAST_EXPECT(!ps.startWork(0));
        ps.finishWork(0);
        BEAST_EXPECT(!ps.pending(0));
    }

    void
    run() override
    {
        testSaves();
    }
};

BEAST_DEFINE_TESTSUITE(PendingSaves, ledger, xrpl);

}  // namespace xrpl::test