mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-09 13:10:25 +00:00
89 lines
2.3 KiB
C++
89 lines
2.3 KiB
C++
|
#include <dht/kademlia.hpp>
|
||
|
|
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
using llarp::dht::Key_t;
|
||
|
|
|
||
|
|
using Array = std::array< byte_t, Key_t::SIZE >;
|
||
|
|
|
||
|
|
struct XorMetricData
|
||
|
|
{
|
||
|
|
Array us;
|
||
|
|
Array left;
|
||
|
|
Array right;
|
||
|
|
bool result;
|
||
|
|
|
||
|
|
XorMetricData(const Array& u, const Array& l, const Array& r, bool res)
|
||
|
|
: us(u), left(l), right(r), result(res)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
std::ostream&
|
||
|
|
operator<<(std::ostream& stream, const XorMetricData& x)
|
||
|
|
{
|
||
|
|
stream << int(x.us[0]) << " " << int(x.left[0]) << " " << int(x.right[0])
|
||
|
|
<< " " << std::boolalpha << x.result;
|
||
|
|
|
||
|
|
return stream;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct XorMetric : public ::testing::TestWithParam< XorMetricData >
|
||
|
|
{
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_P(XorMetric, test)
|
||
|
|
{
|
||
|
|
auto d = GetParam();
|
||
|
|
ASSERT_EQ(llarp::dht::XorMetric{Key_t{d.us}}(Key_t{d.left}, Key_t{d.right}),
|
||
|
|
d.result);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector< XorMetricData >
|
||
|
|
makeData()
|
||
|
|
{
|
||
|
|
std::vector< XorMetricData > result;
|
||
|
|
|
||
|
|
Array zero;
|
||
|
|
zero.fill(0);
|
||
|
|
Array one;
|
||
|
|
one.fill(1);
|
||
|
|
Array two;
|
||
|
|
two.fill(2);
|
||
|
|
Array three;
|
||
|
|
three.fill(3);
|
||
|
|
|
||
|
|
result.emplace_back(zero, zero, zero, false);
|
||
|
|
result.emplace_back(zero, zero, one, true);
|
||
|
|
result.emplace_back(zero, zero, two, true);
|
||
|
|
result.emplace_back(zero, one, zero, false);
|
||
|
|
result.emplace_back(zero, one, one, false);
|
||
|
|
result.emplace_back(zero, one, two, true);
|
||
|
|
result.emplace_back(zero, two, zero, false);
|
||
|
|
result.emplace_back(zero, two, one, false);
|
||
|
|
result.emplace_back(zero, two, two, false);
|
||
|
|
result.emplace_back(one, zero, zero, false);
|
||
|
|
result.emplace_back(one, zero, one, false);
|
||
|
|
result.emplace_back(one, zero, two, true);
|
||
|
|
result.emplace_back(one, one, zero, true);
|
||
|
|
result.emplace_back(one, one, one, false);
|
||
|
|
result.emplace_back(one, one, two, true);
|
||
|
|
result.emplace_back(one, two, zero, false);
|
||
|
|
result.emplace_back(one, two, one, false);
|
||
|
|
result.emplace_back(one, two, two, false);
|
||
|
|
result.emplace_back(two, zero, zero, false);
|
||
|
|
result.emplace_back(two, zero, one, true);
|
||
|
|
result.emplace_back(two, zero, two, false);
|
||
|
|
result.emplace_back(two, one, zero, false);
|
||
|
|
result.emplace_back(two, one, one, false);
|
||
|
|
result.emplace_back(two, one, two, false);
|
||
|
|
result.emplace_back(two, two, zero, true);
|
||
|
|
result.emplace_back(two, two, one, true);
|
||
|
|
result.emplace_back(two, two, two, false);
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
INSTANTIATE_TEST_CASE_P(TestDhtXorMetric, XorMetric,
|
||
|
|
::testing::ValuesIn(makeData()));
|