2021-04-17 14:05:24 +00:00
|
|
|
#include <gtest/gtest.h>
|
2021-11-28 15:19:09 +00:00
|
|
|
#include <memory>
|
2021-04-17 14:05:24 +00:00
|
|
|
|
|
|
|
#include <engine/client/serverbrowser_ping_cache.h>
|
|
|
|
#include <engine/console.h>
|
|
|
|
#include <engine/engine.h>
|
|
|
|
#include <engine/shared/config.h>
|
|
|
|
#include <engine/storage.h>
|
|
|
|
#include <test/test.h>
|
|
|
|
|
|
|
|
TEST(ServerBrowser, PingCache)
|
|
|
|
{
|
|
|
|
CTestInfo Info;
|
2022-01-31 00:09:08 +00:00
|
|
|
Info.m_DeleteTestStorageFilesOnSuccess = true;
|
2021-11-28 15:19:09 +00:00
|
|
|
|
|
|
|
auto pConsole = std::unique_ptr<IConsole>(CreateConsole(CFGFLAG_CLIENT));
|
|
|
|
auto pStorage = std::unique_ptr<IStorage>(Info.CreateTestStorage());
|
|
|
|
auto pPingCache = std::unique_ptr<IServerBrowserPingCache>(CreateServerBrowserPingCache(pConsole.get(), pStorage.get()));
|
2021-04-17 14:05:24 +00:00
|
|
|
|
2022-05-23 18:16:18 +00:00
|
|
|
NETADDR Localhost4, Localhost6, OtherLocalhost4, OtherLocalhost6;
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost4, "127.0.0.1:8303"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost6, "[::1]:8304"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&OtherLocalhost4, "127.0.0.1:8305"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&OtherLocalhost6, "[::1]:8306"));
|
|
|
|
EXPECT_LT(net_addr_comp(&Localhost4, &Localhost6), 0);
|
|
|
|
NETADDR aLocalhostBoth[2] = {Localhost4, Localhost6};
|
2021-04-17 14:05:24 +00:00
|
|
|
|
2022-05-23 18:16:18 +00:00
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 0);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), -1);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
|
|
|
pPingCache->Load();
|
|
|
|
|
2022-05-23 18:16:18 +00:00
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 0);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), -1);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
|
|
|
// Newer pings overwrite older.
|
2022-05-23 18:16:18 +00:00
|
|
|
pPingCache->CachePing(Localhost4, 123);
|
|
|
|
pPingCache->CachePing(Localhost4, 234);
|
|
|
|
pPingCache->CachePing(Localhost4, 345);
|
|
|
|
pPingCache->CachePing(Localhost4, 456);
|
|
|
|
pPingCache->CachePing(Localhost4, 567);
|
|
|
|
pPingCache->CachePing(Localhost4, 678);
|
|
|
|
pPingCache->CachePing(Localhost4, 789);
|
|
|
|
pPingCache->CachePing(Localhost4, 890);
|
|
|
|
pPingCache->CachePing(Localhost4, 901);
|
|
|
|
pPingCache->CachePing(Localhost4, 135);
|
|
|
|
pPingCache->CachePing(Localhost4, 246);
|
|
|
|
pPingCache->CachePing(Localhost4, 357);
|
|
|
|
pPingCache->CachePing(Localhost4, 468);
|
|
|
|
pPingCache->CachePing(Localhost4, 579);
|
|
|
|
pPingCache->CachePing(Localhost4, 680);
|
|
|
|
pPingCache->CachePing(Localhost4, 791);
|
|
|
|
pPingCache->CachePing(Localhost4, 802);
|
|
|
|
pPingCache->CachePing(Localhost4, 913);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
2022-05-23 18:16:18 +00:00
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), 913);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), -1);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), 913);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), 913);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), -1);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
2022-05-23 18:16:18 +00:00
|
|
|
pPingCache->CachePing(Localhost4, 234);
|
|
|
|
pPingCache->CachePing(Localhost6, 345);
|
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 2);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), 234);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), 345);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), 234);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), 234);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), 345);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
|
|
|
// Port doesn't matter for overwriting.
|
|
|
|
pPingCache->CachePing(Localhost4, 1337);
|
2022-05-23 18:16:18 +00:00
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 2);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), 1337);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), 345);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), 345);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), 1337);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), 345);
|
2021-04-17 14:05:24 +00:00
|
|
|
|
2021-11-28 15:19:09 +00:00
|
|
|
pPingCache.reset(CreateServerBrowserPingCache(pConsole.get(), pStorage.get()));
|
2021-04-17 14:05:24 +00:00
|
|
|
|
|
|
|
// Persistence.
|
|
|
|
pPingCache->Load();
|
2022-05-23 18:16:18 +00:00
|
|
|
EXPECT_EQ(pPingCache->NumEntries(), 2);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost4, 1), 1337);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&Localhost6, 1), 345);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(aLocalhostBoth, 2), 345);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost4, 1), 1337);
|
|
|
|
EXPECT_EQ(pPingCache->GetPing(&OtherLocalhost6, 1), 345);
|
2021-04-17 14:05:24 +00:00
|
|
|
}
|