mirror of https://github.com/oxen-io/lokinet
Tests for dht::ServiceAddressLookup
Michael
5 years ago
parent
b1213c7a7c
commit
1ba6507a2f
@ -0,0 +1,221 @@
|
||||
#include <dht/serviceaddresslookup.hpp>
|
||||
|
||||
#include <crypto/mock_crypto.hpp>
|
||||
#include <dht/mock_context.hpp>
|
||||
#include <dht/messages/gotintro.hpp>
|
||||
#include <service/IntroSet.hpp>
|
||||
#include <test_util.hpp>
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
using namespace llarp;
|
||||
using namespace ::testing;
|
||||
|
||||
using test::makeBuf;
|
||||
|
||||
struct MockIntroSetHandler
|
||||
{
|
||||
MOCK_METHOD1(call, void(const std::vector< service::IntroSet > &));
|
||||
};
|
||||
|
||||
static constexpr uint64_t EXPIRY = 1548503831ull;
|
||||
|
||||
struct TestDhtServiceAddressLookup : public ::testing::Test
|
||||
{
|
||||
test::MockCrypto crypto;
|
||||
MockIntroSetHandler introsetHandler;
|
||||
|
||||
dht::Key_t ourKey;
|
||||
dht::Key_t txKey;
|
||||
uint64_t txId;
|
||||
dht::TXOwner txOwner;
|
||||
service::Address address;
|
||||
|
||||
test::MockContext context;
|
||||
uint64_t r;
|
||||
std::unique_ptr< dht::ServiceAddressLookup > serviceAddressLookup;
|
||||
|
||||
TestDhtServiceAddressLookup()
|
||||
: ourKey(makeBuf< dht::Key_t >(0xFF))
|
||||
, txKey(makeBuf< dht::Key_t >(0x01))
|
||||
, txId(2)
|
||||
, txOwner(txKey, txId)
|
||||
, address(makeBuf< service::Address >(0x03))
|
||||
, r(4)
|
||||
{
|
||||
EXPECT_CALL(context, OurKey()).WillOnce(ReturnRef(ourKey));
|
||||
|
||||
serviceAddressLookup = std::make_unique< dht::ServiceAddressLookup >(
|
||||
txOwner, address, &context, r,
|
||||
std::bind(&MockIntroSetHandler::call, &introsetHandler,
|
||||
std::placeholders::_1));
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(TestDhtServiceAddressLookup, validate)
|
||||
{
|
||||
// Concerns
|
||||
// - introset fails to verify
|
||||
// - introset topic is not the target
|
||||
// - happy path
|
||||
|
||||
{
|
||||
service::IntroSet introset;
|
||||
EXPECT_CALL(context, Crypto()).WillOnce(Return(&crypto));
|
||||
EXPECT_CALL(context, Now()).WillOnce(Return(EXPIRY));
|
||||
EXPECT_CALL(crypto, verify(_, _, _)).WillOnce(Return(false));
|
||||
|
||||
ASSERT_FALSE(serviceAddressLookup->Validate(introset));
|
||||
}
|
||||
|
||||
{
|
||||
service::IntroSet introset;
|
||||
|
||||
// Fiddle with the introset so we pass the Verify call
|
||||
introset.I.emplace_back();
|
||||
introset.I.front().expiresAt =
|
||||
EXPIRY + service::MAX_INTROSET_TIME_DELTA + 1;
|
||||
|
||||
// Set expectations
|
||||
EXPECT_CALL(context, Crypto()).WillOnce(Return(&crypto));
|
||||
EXPECT_CALL(context, Now()).WillOnce(Return(EXPIRY));
|
||||
EXPECT_CALL(crypto, verify(_, _, _)).WillOnce(Return(true));
|
||||
|
||||
ASSERT_FALSE(serviceAddressLookup->Validate(introset));
|
||||
}
|
||||
|
||||
{
|
||||
service::IntroSet introset;
|
||||
// Set the current address of the lookup to be equal to the default address.
|
||||
// This is easier than manipulating the ServiceInfo address.
|
||||
serviceAddressLookup->target.Zero();
|
||||
|
||||
// Fiddle with the introset so we pass the Verify call
|
||||
introset.I.emplace_back();
|
||||
introset.I.front().expiresAt =
|
||||
EXPIRY + service::MAX_INTROSET_TIME_DELTA + 1;
|
||||
|
||||
// Set expectations
|
||||
EXPECT_CALL(context, Crypto()).WillOnce(Return(&crypto));
|
||||
EXPECT_CALL(context, Now()).WillOnce(Return(EXPIRY));
|
||||
EXPECT_CALL(crypto, verify(_, _, _)).WillOnce(Return(true));
|
||||
|
||||
ASSERT_TRUE(serviceAddressLookup->Validate(introset));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(TestDhtServiceAddressLookup, start)
|
||||
{
|
||||
// Verify input arguments are passed correctly.
|
||||
// The actual logic is inside the `dht::AbstractContext` implementation.
|
||||
|
||||
// clang-format off
|
||||
EXPECT_CALL(context, DHTSendTo(
|
||||
Eq(txKey.as_array()),
|
||||
WhenDynamicCastTo< dht::FindIntroMessage* >(NotNull()),
|
||||
true)
|
||||
).Times(1);
|
||||
// clang-format off
|
||||
|
||||
ASSERT_NO_THROW(serviceAddressLookup->Start(txOwner));
|
||||
}
|
||||
|
||||
TEST_F(TestDhtServiceAddressLookup, get_next_peer)
|
||||
{
|
||||
// Concerns
|
||||
// - Nodes returns nullptr
|
||||
// - Happy path
|
||||
|
||||
dht::Key_t key = makeBuf< dht::Key_t >(0x02);
|
||||
std::set< dht::Key_t > exclude;
|
||||
{
|
||||
EXPECT_CALL(context, Nodes()).WillOnce(ReturnNull());
|
||||
ASSERT_FALSE(serviceAddressLookup->GetNextPeer(key, exclude));
|
||||
}
|
||||
|
||||
{
|
||||
uint64_t randVal = 0;
|
||||
|
||||
dht::Bucket< dht::RCNode > nodes(ourKey, [&]() { return randVal++; });
|
||||
nodes.nodes.emplace(makeBuf< dht::Key_t >(0x03), dht::RCNode());
|
||||
EXPECT_CALL(context, Nodes()).WillOnce(Return(&nodes));
|
||||
ASSERT_TRUE(serviceAddressLookup->GetNextPeer(key, exclude));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(TestDhtServiceAddressLookup, do_next)
|
||||
{
|
||||
// Concerns:
|
||||
// - R != 0
|
||||
// - R = 0
|
||||
|
||||
const dht::Key_t key = makeBuf< dht::Key_t >(0x02);
|
||||
|
||||
{
|
||||
// R != 0
|
||||
EXPECT_CALL(context, LookupIntroSetRecursive(address, txKey, txId, key, r - 1, _));
|
||||
ASSERT_NO_THROW(serviceAddressLookup->DoNextRequest(key));
|
||||
}
|
||||
|
||||
{
|
||||
// R = 0
|
||||
serviceAddressLookup->R = 0;
|
||||
EXPECT_CALL(context, LookupIntroSetIterative(address, txKey, txId, key, _));
|
||||
ASSERT_NO_THROW(serviceAddressLookup->DoNextRequest(key));
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(TestDhtServiceAddressLookup, send_reply)
|
||||
{
|
||||
// Concerns
|
||||
// - handle result is set, is called
|
||||
// - handle result is not set
|
||||
|
||||
{
|
||||
serviceAddressLookup->valuesFound.emplace_back();
|
||||
EXPECT_CALL(introsetHandler, call(SizeIs(1)));
|
||||
|
||||
// clang-format off
|
||||
EXPECT_CALL(
|
||||
context,
|
||||
DHTSendTo(
|
||||
Eq(txKey.as_array()),
|
||||
WhenDynamicCastTo<dht::GotIntroMessage *>(
|
||||
AllOf(
|
||||
NotNull(),
|
||||
Field(&dht::GotIntroMessage::I, SizeIs(1))
|
||||
)
|
||||
),
|
||||
true
|
||||
)
|
||||
);
|
||||
// clang-format on
|
||||
|
||||
ASSERT_NO_THROW(serviceAddressLookup->SendReply());
|
||||
}
|
||||
|
||||
{
|
||||
serviceAddressLookup->valuesFound.clear();
|
||||
serviceAddressLookup->valuesFound.emplace_back();
|
||||
serviceAddressLookup->handleResult =
|
||||
decltype(serviceAddressLookup->handleResult)();
|
||||
|
||||
// clang-format off
|
||||
EXPECT_CALL(
|
||||
context,
|
||||
DHTSendTo(
|
||||
Eq(txKey.as_array()),
|
||||
WhenDynamicCastTo<dht::GotIntroMessage *>(
|
||||
AllOf(
|
||||
NotNull(),
|
||||
Field(&dht::GotIntroMessage::I, SizeIs(1))
|
||||
)
|
||||
),
|
||||
true
|
||||
)
|
||||
);
|
||||
// clang-format on
|
||||
|
||||
ASSERT_NO_THROW(serviceAddressLookup->SendReply());
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue