139 lines
4.4 KiB
C++
139 lines
4.4 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "ob_simple_replica.h"
|
|
#include <rapidjson/document.h>
|
|
#include <rapidjson/filewritestream.h>
|
|
#include <rapidjson/filereadstream.h>
|
|
#include <rapidjson/prettywriter.h>
|
|
#include <rapidjson/stringbuffer.h>
|
|
|
|
#define MAX_ZONE_COUNT 3
|
|
#define CLUSTER_EVENT_FILE_NAME "CLUSTER_EVENT"
|
|
#define EVENT_KV_SEPARATOR " = "
|
|
#define TEST_CASE_FINSH_EVENT_PREFIX "FINISH_TEST_CASE_FOR_"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace unittest
|
|
{
|
|
|
|
int set_trace_id(char *buf);
|
|
void init_log_and_gtest(int argc, char **argv);
|
|
void init_gtest_output(std::string >est_log_name);
|
|
|
|
struct RestartIndex
|
|
{
|
|
int restart_zone_id_;
|
|
int restart_no_;
|
|
|
|
void reset()
|
|
{
|
|
restart_zone_id_ = -1;
|
|
restart_no_ = 0;
|
|
}
|
|
|
|
RestartIndex()
|
|
{
|
|
reset();
|
|
}
|
|
};
|
|
|
|
class ObMultiReplicaTestBase : public testing::Test
|
|
{
|
|
public:
|
|
static const int64_t TRANS_TIMEOUT = 5 * 1000 * 1000;
|
|
// set_bootstrap_and_create_tenant_warn_log 默认bootstrap和创建租户使用WARN日志，加速启动
|
|
ObMultiReplicaTestBase();
|
|
virtual ~ObMultiReplicaTestBase();
|
|
|
|
static int bootstrap_multi_replica(const std::string &app_name,
|
|
const int restart_zone_id = -1,
|
|
const int restart_no = 0,
|
|
const std::string &env_prefix = "run_");
|
|
static int wait_all_test_completed();
|
|
static int restart_zone(const int zone_id, const int restart_no);
|
|
static int start();
|
|
static int close();
|
|
observer::ObServer &get_curr_observer() { return replica_->get_observer(); }
|
|
observer::ObSimpleServerReplica &get_curr_simple_server() { return *replica_; }
|
|
|
|
static int read_cur_json_document_(rapidjson::Document & json_doc);
|
|
|
|
static int wait_event_finish(const std::string &event_name,
|
|
std::string &event_content,
|
|
int64_t wait_timeout_ms,
|
|
int64_t retry_interval_ms = 1 * 1000);
|
|
static int finish_event(const std::string &event_name, const std::string &event_content);
|
|
|
|
int create_tenant(const char *tenant_name = DEFAULT_TEST_TENANT_NAME,
|
|
const char *memory_size = "2G",
|
|
const char *log_disk_size = "2G",
|
|
const bool oracle_mode = false,
|
|
const char *primary_zone = "zone1");
|
|
int delete_tenant(const char *tenant_name = DEFAULT_TEST_TENANT_NAME);
|
|
int get_tenant_id(uint64_t &tenant_id, const char *tenant_name = DEFAULT_TEST_TENANT_NAME);
|
|
int exec_write_sql_sys(const char *sql_str, int64_t &affected_rows);
|
|
int check_tenant_exist(bool &bool_ret, const char *tenant_name = DEFAULT_TEST_TENANT_NAME);
|
|
|
|
static std::string TEST_CASE_BASE_NAME;
|
|
static std::string ZONE_TEST_CASE_NAME[MAX_ZONE_COUNT];
|
|
protected:
|
|
static int init_replicas_();
|
|
static int init_test_replica_(const int zone_id);
|
|
|
|
static int is_valid_zone_id(int zone_id) { return zone_id >= 0; }
|
|
|
|
protected:
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
static void SetUpTestCase();
|
|
static void TearDownTestCase();
|
|
|
|
protected:
|
|
// 因为ob_server.h 中ObServer的使用方式导致现在只能启动单台
|
|
static std::shared_ptr<observer::ObSimpleServerReplica> replica_;
|
|
static bool is_started_;
|
|
static bool is_inited_;
|
|
static std::thread th_;
|
|
static std::string env_prefix_;
|
|
static std::string app_name_;
|
|
static std::string exec_dir_;
|
|
static std::string event_file_path_;
|
|
static std::string env_prefix_path_;
|
|
static bool enable_env_warn_log_;
|
|
|
|
static const char *log_disk_size_;
|
|
static const char *memory_size_;
|
|
|
|
static std::string local_ip_;
|
|
|
|
static int cur_zone_id_;
|
|
static int restart_zone_id_;
|
|
static int restart_no_;
|
|
static std::vector<int> zone_pids_;
|
|
|
|
static std::vector<int64_t> rpc_ports_;
|
|
static ObServerInfoList server_list_;
|
|
static std::string rs_list_;
|
|
|
|
public:
|
|
static bool block_msg_;
|
|
};
|
|
|
|
} // namespace unittest
|
|
} // namespace oceanbase
|