90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
|
#include <gtest/gtest.h>
|
||
|
|
#include <vanetza/security/peer_request_tracker.hpp>
|
||
|
|
|
||
|
|
#include "printer.hpp"
|
||
|
|
|
||
|
|
using namespace vanetza::security;
|
||
|
|
|
||
|
|
HashedId3 create_id(uint32_t id)
|
||
|
|
{
|
||
|
|
HashedId3 hid;
|
||
|
|
hid[0] = id & 0xFF;
|
||
|
|
hid[1] = (id >> 8) & 0xFF;
|
||
|
|
hid[2] = (id >> 16) & 0xFF;
|
||
|
|
return hid;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(PeerRequestTracker, is_pending)
|
||
|
|
{
|
||
|
|
PeerRequestTracker tracker;
|
||
|
|
EXPECT_FALSE(tracker.is_pending(create_id(42)));
|
||
|
|
tracker.add_request(create_id(42));
|
||
|
|
EXPECT_TRUE(tracker.is_pending(create_id(42)));
|
||
|
|
tracker.discard_request(create_id(42));
|
||
|
|
EXPECT_FALSE(tracker.is_pending(create_id(42)));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(PeerRequestTracker, bounded_capacity)
|
||
|
|
{
|
||
|
|
PeerRequestTracker tracker(2);
|
||
|
|
tracker.add_request(create_id(1));
|
||
|
|
tracker.add_request(create_id(2));
|
||
|
|
EXPECT_TRUE(tracker.is_pending(create_id(1)));
|
||
|
|
EXPECT_TRUE(tracker.is_pending(create_id(2)));
|
||
|
|
|
||
|
|
tracker.add_request(create_id(3));
|
||
|
|
EXPECT_TRUE(tracker.is_pending(create_id(3)));
|
||
|
|
EXPECT_TRUE(tracker.is_pending(create_id(2)));
|
||
|
|
// dropped oldest pending request
|
||
|
|
EXPECT_FALSE(tracker.is_pending(create_id(1)));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(PeerRequestTracker, keep_order)
|
||
|
|
{
|
||
|
|
PeerRequestTracker tracker(4);
|
||
|
|
tracker.add_request(create_id(1));
|
||
|
|
tracker.add_request(create_id(2));
|
||
|
|
tracker.add_request(create_id(3));
|
||
|
|
tracker.add_request(create_id(4));
|
||
|
|
tracker.add_request(create_id(3));
|
||
|
|
tracker.add_request(create_id(2));
|
||
|
|
|
||
|
|
std::list<HashedId3> expected = { create_id(1), create_id(2), create_id(3), create_id(4) };
|
||
|
|
EXPECT_EQ(expected, tracker.all());
|
||
|
|
|
||
|
|
// nothing left in tracker
|
||
|
|
EXPECT_FALSE(tracker.next_one());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(PeerRequestTracker, next_one)
|
||
|
|
{
|
||
|
|
PeerRequestTracker tracker(3);
|
||
|
|
tracker.add_request(create_id(0xc0));
|
||
|
|
tracker.add_request(create_id(0xff));
|
||
|
|
tracker.add_request(create_id(0xee));
|
||
|
|
tracker.add_request(create_id(0x42));
|
||
|
|
|
||
|
|
EXPECT_EQ(tracker.next_one(), create_id(0xff));
|
||
|
|
EXPECT_EQ(tracker.next_one(), create_id(0xee));
|
||
|
|
EXPECT_EQ(tracker.next_one(), create_id(0x42));
|
||
|
|
EXPECT_FALSE(tracker.next_one());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(PeerRequestTracker, next_n)
|
||
|
|
{
|
||
|
|
PeerRequestTracker tracker(6);
|
||
|
|
tracker.add_request(create_id(0x01));
|
||
|
|
tracker.add_request(create_id(0x02));
|
||
|
|
tracker.add_request(create_id(0x03));
|
||
|
|
tracker.add_request(create_id(0x04));
|
||
|
|
tracker.add_request(create_id(0x05));
|
||
|
|
|
||
|
|
std::list<HashedId3> expected = { create_id(0x01), create_id(0x02), create_id(0x03) };
|
||
|
|
EXPECT_EQ(expected, tracker.next_n(3));
|
||
|
|
|
||
|
|
expected = { create_id(0x04), create_id(0x05) };
|
||
|
|
EXPECT_EQ(expected, tracker.next_n(8));
|
||
|
|
|
||
|
|
expected = { };
|
||
|
|
EXPECT_EQ(expected, tracker.next_n(2));
|
||
|
|
}
|