KeyCache_test.cpp

#include <test/unit_test/SuiteJournal.h>
 
#include <xrpl/basics/TaggedCache.h>
#include <xrpl/basics/TaggedCache.ipp>  // IWYU pragma: keep
#include <xrpl/basics/chrono.h>
#include <xrpl/beast/unit_test/suite.h>
#include <xrpl/protocol/Protocol.h>
 
#include <string>
 
namespace xrpl {
 
class KeyCache_test : public beast::unit_test::Suite
{
public:
    void
    run() override
    {
        using namespace std::chrono_literals;
        TestStopwatch clock;
        clock.set(0);
 
        using Key = std::string;
        using Cache = TaggedCache<Key, int, true>;
 
        test::SuiteJournal j("KeyCacheTest", *this);
 
        // Insert an item, retrieve it, and age it so it gets purged.
        {
            Cache c("test", LedgerIndex(1), 2s, clock, j);
 
            BEAST_EXPECT(c.size() == 0);
            BEAST_EXPECT(c.insert("one"));
            BEAST_EXPECT(!c.insert("one"));
            BEAST_EXPECT(c.size() == 1);
            BEAST_EXPECT(c.touchIfExists("one"));
            ++clock;
            c.sweep();
            BEAST_EXPECT(c.size() == 1);
            ++clock;
            c.sweep();
            BEAST_EXPECT(c.size() == 0);
            BEAST_EXPECT(!c.touchIfExists("one"));
        }
 
        // Insert two items, have one expire
        {
            Cache c("test", LedgerIndex(2), 2s, clock, j);
 
            BEAST_EXPECT(c.insert("one"));
            BEAST_EXPECT(c.size() == 1);
            BEAST_EXPECT(c.insert("two"));
            BEAST_EXPECT(c.size() == 2);
            ++clock;
            c.sweep();
            BEAST_EXPECT(c.size() == 2);
            BEAST_EXPECT(c.touchIfExists("two"));
            ++clock;
            c.sweep();
            BEAST_EXPECT(c.size() == 1);
        }
 
        // Insert three items (1 over limit), sweep
        {
            Cache c("test", LedgerIndex(2), 3s, clock, j);
 
            BEAST_EXPECT(c.insert("one"));
            ++clock;
            BEAST_EXPECT(c.insert("two"));
            ++clock;
            BEAST_EXPECT(c.insert("three"));
            ++clock;
            BEAST_EXPECT(c.size() == 3);
            c.sweep();
            BEAST_EXPECT(c.size() < 3);
        }
    }
};
 
BEAST_DEFINE_TESTSUITE(KeyCache, basics, xrpl);
 
}  // namespace xrpl