126 lines
4.4 KiB
C++
126 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.
|
|
*/
|
|
|
|
#define USING_LOG_PREFIX SHARE
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "share/ob_web_service_root_addr.h"
|
|
#include "schema/db_initializer.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace share
|
|
{
|
|
using namespace common;
|
|
|
|
class TestWebServiceRootAddr : public ::testing::Test
|
|
{
|
|
public:
|
|
TestWebServiceRootAddr() : service_pid_(0) {}
|
|
|
|
virtual void SetUp();
|
|
virtual void TearDown();
|
|
|
|
void set_response_json(const char *json);
|
|
|
|
protected:
|
|
pid_t service_pid_;
|
|
schema::DBInitializer initer_;
|
|
schema::DBInitializer initer2_;
|
|
ObWebServiceRootAddr ws_;
|
|
};
|
|
|
|
void TestWebServiceRootAddr::SetUp()
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
initer2_.get_config().obconfig_url.set_value("http://127.0.0.1:8658/obconfig/region1");
|
|
initer2_.get_config().cluster_id.set_value("1");
|
|
|
|
pid_t pid = fork();
|
|
if (0 == pid) {
|
|
ret = setpgid(pid, pid);
|
|
if (ret < 0) {
|
|
LOG_ERROR("setpgid failed", K(errno));
|
|
} else if (-1 == (ret = execl(
|
|
"/bin/bash", "./fake_ob_config-sh2", "./fake_ob_config-sh2", "8658", NULL))) {
|
|
LOG_ERROR("execl failed", K(errno));
|
|
}
|
|
exit(1);
|
|
} else if (-1 == pid) {
|
|
LOG_ERROR("fork failed", K(errno));
|
|
} else {
|
|
LOG_INFO("create child", K(pid));
|
|
service_pid_ = pid;
|
|
|
|
// wait ./fake_ob_config-sh execute.
|
|
usleep(100000);
|
|
}
|
|
}
|
|
|
|
void TestWebServiceRootAddr::TearDown()
|
|
{
|
|
int status = 0;
|
|
kill(-service_pid_, SIGINT);
|
|
pid_t pid = wait(&status);
|
|
LOG_INFO("child exit", K(pid));
|
|
}
|
|
|
|
void TestWebServiceRootAddr::set_response_json(const char *json)
|
|
{
|
|
usleep(50000);
|
|
ObSqlString cmd;
|
|
int ret = cmd.assign_fmt("echo -n 'POST / HTTP/1.1\r\nContent-Length: %ld\r\n%s' | nc 127.0.0.1 8658",
|
|
strlen(json), json);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
ret = system(cmd.ptr());
|
|
LOG_INFO("set response json", K(cmd), K(ret), K(errno), K(json));
|
|
usleep(50000);
|
|
}
|
|
|
|
TEST_F(TestWebServiceRootAddr, fetch_version2)
|
|
{
|
|
ObSEArray<ObRootAddr, 16> rs_list;
|
|
ObSEArray<ObRootAddr, 16> readonly_rs_list;
|
|
|
|
//No master
|
|
set_response_json("[{\"ObRegion\":\"ob2.rongxuan.lc\",\"ObRegionId\":2,\"Type\":\"SLAVE\",\"RsList\":[{\"address\":\"10.101.67.165:16825\",\"role\":\"LEADER\",\"sql_port\":16860},{\"address\":\"10.218.253.100:16826\",\"role\":\"FOLLOWER\",\"sql_port\":16861}],\"ReadonlyRsList\":[]}]");
|
|
int ret = ws_.fetch_master_cluster_info(&(initer_.get_config()), rs_list, readonly_rs_list);
|
|
ASSERT_EQ(OB_ENTRY_NOT_EXIST, ret);
|
|
|
|
//Have master
|
|
set_response_json("[{\"ObRegion\":\"ob2.rongxuan.lc\",\"ObRegionId\":2,\"Type\":\"MASTER\",\"RsList\":[{\"address\":\"10.101.67.165:16825\",\"role\":\"LEADER\",\"sql_port\":16860}],\"ReadonlyRsList\":[]}]");
|
|
ret = ws_.fetch_master_cluster_info(&(initer_.get_config()), rs_list, readonly_rs_list);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
ASSERT_EQ(rs_list.count(), 1);
|
|
ASSERT_EQ(readonly_rs_list.count(), 0);
|
|
ASSERT_EQ(16860, rs_list.at(0).sql_port_);
|
|
|
|
//Active and standby two clusters
|
|
set_response_json("[{\"ObRegion\":\"ob2.rongxuan.lc\",\"ObRegionId\":2,\"Type\":\"SLAVE\",\"RsList\":[{\"address\":\"10.101.67.165:16825\",\"role\":\"LEADER\",\"sql_port\":16860},{\"address\":\"10.218.253.100:16826\",\"role\":\"FOLLOWER\",\"sql_port\":16861}],\"ReadonlyRsList\":[]}"
|
|
",{\"ObRegion\":\"ob2.rongxuan.lc\",\"ObRegionId\":3,\"Type\":\"MASTER\",\"RsList\":[{\"address\":\"10.101.67.160:16820\",\"role\":\"LEADER\",\"sql_port\":16870},"
|
|
"{\"address\":\"10.218.253.170:16825\",\"role\":\"FOLLOWER\",\"sql_port\":16865}],\"ReadonlyRsList\":[]}]");
|
|
ret = ws_.fetch_master_cluster_info(&(initer_.get_config()), rs_list, readonly_rs_list);
|
|
ASSERT_EQ(OB_SUCCESS, ret);
|
|
ASSERT_EQ(rs_list.count(), 2);
|
|
ASSERT_EQ(readonly_rs_list.count(), 0);
|
|
}
|
|
} // end namespace share
|
|
} // end namespace oceanbase
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
|
|
OB_LOGGER.set_log_level("INFO");
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|