131 lines
5.8 KiB
C++
131 lines
5.8 KiB
C++
// Copyright 2024 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#include "coordination/coordinator_instance.hpp"
|
|
|
|
#include "auth/auth.hpp"
|
|
#include "flags/run_time_configurable.hpp"
|
|
#include "interpreter_faker.hpp"
|
|
#include "io/network/endpoint.hpp"
|
|
#include "license/license.hpp"
|
|
#include "replication_handler/replication_handler.hpp"
|
|
#include "storage/v2/config.hpp"
|
|
|
|
#include "utils/file.hpp"
|
|
|
|
#include <gflags/gflags.h>
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
#include "json/json.hpp"
|
|
|
|
using memgraph::coordination::CoordinatorInstance;
|
|
using memgraph::coordination::CoordinatorInstanceInitConfig;
|
|
using memgraph::coordination::CoordinatorToCoordinatorConfig;
|
|
using memgraph::coordination::CoordinatorToReplicaConfig;
|
|
using memgraph::coordination::HealthCheckClientCallback;
|
|
using memgraph::coordination::HealthCheckInstanceCallback;
|
|
using memgraph::coordination::RaftState;
|
|
using memgraph::coordination::RegisterInstanceCoordinatorStatus;
|
|
using memgraph::coordination::ReplicationClientInfo;
|
|
using memgraph::coordination::ReplicationInstanceClient;
|
|
using memgraph::coordination::ReplicationInstanceConnector;
|
|
using memgraph::io::network::Endpoint;
|
|
using memgraph::replication::ReplicationHandler;
|
|
using memgraph::replication_coordination_glue::ReplicationMode;
|
|
using memgraph::storage::Config;
|
|
|
|
using testing::_;
|
|
|
|
class ReplicationInstanceClientMock : public ReplicationInstanceClient {
|
|
public:
|
|
ReplicationInstanceClientMock(CoordinatorInstance *coord_instance, CoordinatorToReplicaConfig config)
|
|
: ReplicationInstanceClient(coord_instance, config, nullptr, nullptr) {
|
|
ON_CALL(*this, DemoteToReplica()).WillByDefault(testing::Return(true));
|
|
}
|
|
MOCK_METHOD0(DemoteToReplica, bool());
|
|
};
|
|
|
|
class CoordinatorInstanceTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {}
|
|
|
|
void TearDown() override {}
|
|
|
|
std::filesystem::path main_data_directory{std::filesystem::temp_directory_path() /
|
|
"MG_tests_unit_coordinator_instance"};
|
|
};
|
|
|
|
TEST_F(CoordinatorInstanceTest, RegisterReplicationInstance) {
|
|
auto const init_config =
|
|
CoordinatorInstanceInitConfig{.coordinator_id = 4, .coordinator_port = 10110, .bolt_port = 7686};
|
|
auto instance1 = CoordinatorInstance{init_config};
|
|
|
|
auto const coord_to_replica_config =
|
|
CoordinatorToReplicaConfig{.instance_name = "instance3",
|
|
.mgt_server = Endpoint{"127.0.0.1", 10112},
|
|
.bolt_server = Endpoint{"127.0.0.1", 7687},
|
|
.replication_client_info = {.instance_name = "instance_name",
|
|
.replication_mode = ReplicationMode::ASYNC,
|
|
.replication_server = Endpoint{"127.0.0.1", 10001}},
|
|
.instance_health_check_frequency_sec = std::chrono::seconds{1},
|
|
.instance_down_timeout_sec = std::chrono::seconds{5},
|
|
.instance_get_uuid_frequency_sec = std::chrono::seconds{10},
|
|
.ssl = std::nullopt};
|
|
|
|
auto status = instance1.RegisterReplicationInstance(coord_to_replica_config);
|
|
EXPECT_EQ(status, RegisterInstanceCoordinatorStatus::SUCCESS);
|
|
}
|
|
|
|
TEST_F(CoordinatorInstanceTest, ShowInstancesEmptyTest) {
|
|
auto const init_config =
|
|
CoordinatorInstanceInitConfig{.coordinator_id = 4, .coordinator_port = 10110, .bolt_port = 7686};
|
|
|
|
auto const instance1 = CoordinatorInstance{init_config};
|
|
auto const instances = instance1.ShowInstances();
|
|
ASSERT_EQ(instances.size(), 1);
|
|
ASSERT_EQ(instances[0].instance_name, "coordinator_4");
|
|
ASSERT_EQ(instances[0].health, "unknown");
|
|
ASSERT_EQ(instances[0].raft_socket_address, "127.0.0.1:10110");
|
|
ASSERT_EQ(instances[0].coord_socket_address, "");
|
|
ASSERT_EQ(instances[0].cluster_role, "coordinator");
|
|
}
|
|
|
|
TEST_F(CoordinatorInstanceTest, ConnectCoordinators) {
|
|
auto const init_config1 =
|
|
CoordinatorInstanceInitConfig{.coordinator_id = 1, .coordinator_port = 10111, .bolt_port = 7687};
|
|
|
|
auto instance1 = CoordinatorInstance{init_config1};
|
|
|
|
auto const init_config2 =
|
|
CoordinatorInstanceInitConfig{.coordinator_id = 2, .coordinator_port = 10112, .bolt_port = 7688};
|
|
|
|
auto const instance2 = CoordinatorInstance{init_config2};
|
|
|
|
auto const init_config3 =
|
|
CoordinatorInstanceInitConfig{.coordinator_id = 3, .coordinator_port = 10113, .bolt_port = 7689};
|
|
|
|
auto const instance3 = CoordinatorInstance{init_config3};
|
|
|
|
instance1.AddCoordinatorInstance(CoordinatorToCoordinatorConfig{.coordinator_id = 2,
|
|
.bolt_server = Endpoint{"127.0.0.1", 7688},
|
|
.coordinator_server = Endpoint{"127.0.0.1", 10112}});
|
|
|
|
instance1.AddCoordinatorInstance(CoordinatorToCoordinatorConfig{.coordinator_id = 3,
|
|
.bolt_server = Endpoint{"127.0.0.1", 7689},
|
|
.coordinator_server = Endpoint{"127.0.0.1", 10113}});
|
|
|
|
auto const instances = instance1.ShowInstances();
|
|
ASSERT_EQ(instances.size(), 3);
|
|
ASSERT_EQ(instances[0].instance_name, "coordinator_1");
|
|
ASSERT_EQ(instances[1].instance_name, "coordinator_2");
|
|
ASSERT_EQ(instances[2].instance_name, "coordinator_3");
|
|
}
|