9291a5fc4d
Reviewers: teon.banek, buda Reviewed By: teon.banek Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1974
54 lines
1.7 KiB
C++
54 lines
1.7 KiB
C++
#include <optional>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "communication/rpc/server.hpp"
|
|
#include "storage/common/types/types.hpp"
|
|
#include "storage/distributed/concurrent_id_mapper_master.hpp"
|
|
#include "storage/distributed/concurrent_id_mapper_worker.hpp"
|
|
|
|
#include "test_coordination.hpp"
|
|
|
|
template <typename TId>
|
|
class DistributedConcurrentIdMapperTest : public ::testing::Test {
|
|
const std::string kLocal{"127.0.0.1"};
|
|
|
|
protected:
|
|
TestMasterCoordination coordination_;
|
|
std::optional<communication::rpc::ClientPool> master_client_pool_;
|
|
std::optional<storage::MasterConcurrentIdMapper<TId>> master_mapper_;
|
|
std::optional<storage::WorkerConcurrentIdMapper<TId>> worker_mapper_;
|
|
|
|
void SetUp() override {
|
|
master_mapper_.emplace(&coordination_);
|
|
coordination_.Start();
|
|
master_client_pool_.emplace(coordination_.GetServerEndpoint());
|
|
worker_mapper_.emplace(&master_client_pool_.value());
|
|
}
|
|
void TearDown() override {
|
|
worker_mapper_ = std::nullopt;
|
|
master_client_pool_ = std::nullopt;
|
|
coordination_.Stop();
|
|
master_mapper_ = std::nullopt;
|
|
}
|
|
};
|
|
|
|
typedef ::testing::Types<storage::Label, storage::EdgeType, storage::Property>
|
|
GraphDbTestTypes;
|
|
TYPED_TEST_CASE(DistributedConcurrentIdMapperTest, GraphDbTestTypes);
|
|
|
|
TYPED_TEST(DistributedConcurrentIdMapperTest, Basic) {
|
|
auto &master = this->master_mapper_.value();
|
|
auto &worker = this->worker_mapper_.value();
|
|
|
|
auto id1 = master.value_to_id("v1");
|
|
EXPECT_EQ(worker.id_to_value(id1), "v1");
|
|
EXPECT_EQ(worker.value_to_id("v1"), id1);
|
|
|
|
auto id2 = worker.value_to_id("v2");
|
|
EXPECT_EQ(master.id_to_value(id2), "v2");
|
|
EXPECT_EQ(master.value_to_id("v2"), id2);
|
|
|
|
EXPECT_NE(id1, id2);
|
|
}
|