mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
1.9 KiB
C++
77 lines
1.9 KiB
C++
#include <peerstats/peer_db.hpp>
|
|
#include <test_util.hpp>
|
|
|
|
#include <numeric>
|
|
#include <catch2/catch.hpp>
|
|
|
|
TEST_CASE("Test PeerDb PeerStats memory storage", "[PeerDb]")
|
|
{
|
|
const llarp::RouterID id = llarp::test::makeBuf<llarp::RouterID>(0x01);
|
|
const llarp::PeerStats empty(id);
|
|
|
|
llarp::PeerDb db;
|
|
CHECK(db.getCurrentPeerStats(id).has_value() == false);
|
|
|
|
llarp::PeerStats delta(id);
|
|
delta.numConnectionAttempts = 4;
|
|
delta.peakBandwidthBytesPerSec = 5;
|
|
db.accumulatePeerStats(id, delta);
|
|
CHECK(db.getCurrentPeerStats(id).value() == delta);
|
|
|
|
delta = llarp::PeerStats(id);
|
|
delta.numConnectionAttempts = 5;
|
|
delta.peakBandwidthBytesPerSec = 6;
|
|
db.accumulatePeerStats(id, delta);
|
|
|
|
llarp::PeerStats expected(id);
|
|
expected.numConnectionAttempts = 9;
|
|
expected.peakBandwidthBytesPerSec = 6;
|
|
CHECK(db.getCurrentPeerStats(id).value() == expected);
|
|
}
|
|
|
|
TEST_CASE("Test PeerDb flush before load", "[PeerDb]")
|
|
{
|
|
llarp::PeerDb db;
|
|
CHECK_THROWS_WITH(db.flushDatabase(), "Cannot flush database before it has been loaded");
|
|
}
|
|
|
|
TEST_CASE("Test PeerDb load twice", "[PeerDb]")
|
|
{
|
|
llarp::PeerDb db;
|
|
CHECK_NOTHROW(db.loadDatabase(std::nullopt));
|
|
CHECK_THROWS_WITH(db.loadDatabase(std::nullopt), "Reloading database not supported");
|
|
}
|
|
|
|
TEST_CASE("Test PeerDb nukes stats on load", "[PeerDb]")
|
|
{
|
|
const llarp::RouterID id = llarp::test::makeBuf<llarp::RouterID>(0x01);
|
|
|
|
llarp::PeerDb db;
|
|
|
|
llarp::PeerStats stats(id);
|
|
stats.numConnectionAttempts = 1;
|
|
|
|
db.accumulatePeerStats(id, stats);
|
|
CHECK(db.getCurrentPeerStats(id).value() == stats);
|
|
|
|
db.loadDatabase(std::nullopt);
|
|
|
|
CHECK(db.getCurrentPeerStats(id).has_value() == false);
|
|
}
|
|
|
|
/*
|
|
TEST_CASE("Test file-backed database", "[PeerDb]")
|
|
{
|
|
llarp::PeerDb db;
|
|
db.loadDatabase("/tmp/peerdb_test_tmp.db.sqlite");
|
|
|
|
const llarp::RouterID id = llarp::test::makeBuf<llarp::RouterID>(0x01);
|
|
llarp::PeerStats stats(id);
|
|
stats.numConnectionAttempts = 42;
|
|
|
|
db.accumulatePeerStats(id, stats);
|
|
|
|
db.flushDatabase();
|
|
}
|
|
*/
|