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
|
|
|
|
|
|
|
const IServerBrowserPingCache::CEntry *pEntries;
|
|
|
|
int NumEntries;
|
|
|
|
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 0);
|
|
|
|
|
|
|
|
pPingCache->Load();
|
|
|
|
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 0);
|
|
|
|
|
|
|
|
NETADDR Localhost4, Localhost4Tw, Localhost6, Localhost6Tw;
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost4, "127.0.0.1"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost4Tw, "127.0.0.1:8303"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost6, "[::1]"));
|
|
|
|
ASSERT_FALSE(net_addr_from_str(&Localhost6Tw, "[::1]:8304"));
|
|
|
|
EXPECT_LT(net_addr_comp(&Localhost4, &Localhost6), 0);
|
|
|
|
|
|
|
|
// Newer pings overwrite older.
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 123);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 234);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 345);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 456);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 567);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 678);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 789);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 890);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 901);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 135);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 246);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 357);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 468);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 579);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 680);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 791);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 802);
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 913);
|
|
|
|
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 1);
|
|
|
|
if(NumEntries >= 1)
|
|
|
|
{
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[0].m_Addr, &Localhost4) == 0);
|
|
|
|
EXPECT_EQ(pEntries[0].m_Ping, 913);
|
|
|
|
}
|
|
|
|
|
|
|
|
pPingCache->CachePing(Localhost4Tw, 234);
|
|
|
|
pPingCache->CachePing(Localhost6Tw, 345);
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 2);
|
|
|
|
if(NumEntries >= 2)
|
|
|
|
{
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[0].m_Addr, &Localhost4) == 0);
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[1].m_Addr, &Localhost6) == 0);
|
|
|
|
EXPECT_EQ(pEntries[0].m_Ping, 234);
|
|
|
|
EXPECT_EQ(pEntries[1].m_Ping, 345);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Port doesn't matter for overwriting.
|
|
|
|
pPingCache->CachePing(Localhost4, 1337);
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 2);
|
|
|
|
if(NumEntries >= 2)
|
|
|
|
{
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[0].m_Addr, &Localhost4) == 0);
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[1].m_Addr, &Localhost6) == 0);
|
|
|
|
EXPECT_EQ(pEntries[0].m_Ping, 1337);
|
|
|
|
EXPECT_EQ(pEntries[1].m_Ping, 345);
|
|
|
|
}
|
|
|
|
|
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();
|
|
|
|
pPingCache->GetPingCache(&pEntries, &NumEntries);
|
|
|
|
EXPECT_EQ(NumEntries, 2);
|
|
|
|
if(NumEntries >= 2)
|
|
|
|
{
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[0].m_Addr, &Localhost4) == 0);
|
|
|
|
EXPECT_TRUE(net_addr_comp(&pEntries[1].m_Addr, &Localhost6) == 0);
|
|
|
|
EXPECT_EQ(pEntries[0].m_Ping, 1337);
|
|
|
|
EXPECT_EQ(pEntries[1].m_Ping, 345);
|
|
|
|
}
|
|
|
|
}
|