/** * 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 #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 start_new_shell(); void set_response_json(const char *json); void set_response_json2(const char *json); protected: pid_t service_pid_; schema::DBInitializer initer_; schema::DBInitializer initer2_; ObWebServiceRootAddr ws_; }; void TestWebServiceRootAddr::start_new_shell() { 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::SetUp() { initer_.get_config().obconfig_url.set_value("http://127.0.0.1:8657/obconfig/region1"); initer_.get_config().cluster_id.set_value("1"); int ret = ws_.init(initer_.get_config()); ASSERT_EQ(OB_SUCCESS, ret); 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-sh", "./fake_ob_config-sh", "8657", 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_json2(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); } 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 8657", 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, store) { ObSEArray rs_list; ObSEArray readonly_rs_list; ObRootAddr rs; rs.server_.set_ip_addr("127.0.0.1", 9988); rs.sql_port_ = 1; rs_list.push_back(rs); rs.server_.set_ip_addr("127.0.0.2", 9988); rs.role_ = LEADER; rs.sql_port_ = 1111; rs_list.push_back(rs); ObRootAddr readonly_rs; readonly_rs.server_.set_ip_addr("127.0.0.3", 9988); readonly_rs.sql_port_ = 1; readonly_rs_list.push_back(readonly_rs); readonly_rs.server_.set_ip_addr("127.0.0.4", 9988); readonly_rs.sql_port_ = 1111; readonly_rs_list.push_back(readonly_rs); int ret = ws_.store(rs_list, readonly_rs_list, true, LEADER); usleep(100000); rs_list.reset(); readonly_rs_list.reset(); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(2, rs_list.count()); ObAddr rs1(ObAddr::IPV4, "127.0.0.1", 9988); LOG_WARN("get server", K(rs1), K_(rs_list.at(0).server)); ASSERT_EQ(rs1, rs_list.at(0).server_); ASSERT_EQ(FOLLOWER, rs_list.at(0).role_); ASSERT_EQ(1, rs_list.at(0).sql_port_); ObAddr rs2(ObAddr::IPV4, "127.0.0.2", 9988); ASSERT_EQ(rs2, rs_list.at(1).server_); ASSERT_EQ(LEADER, rs_list.at(1).role_); ASSERT_EQ(1111, rs_list.at(1).sql_port_); ASSERT_EQ(2, readonly_rs_list.count()); ObAddr rs3(ObAddr::IPV4, "127.0.0.3", 9988); ASSERT_EQ(rs3, readonly_rs_list.at(0).server_); ASSERT_EQ(FOLLOWER, readonly_rs_list.at(0).role_); ASSERT_EQ(1, readonly_rs_list.at(0).sql_port_); ObAddr rs4(ObAddr::IPV4, "127.0.0.4", 9988); ASSERT_EQ(rs4, readonly_rs_list.at(1).server_); ASSERT_EQ(FOLLOWER, readonly_rs_list.at(1).role_); ASSERT_EQ(1111, readonly_rs_list.at(1).sql_port_); } TEST_F(TestWebServiceRootAddr, fetch) { ObSEArray rs_list; ObSEArray readonly_rs_list; set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\",\"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"}]}"); int ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(2, rs_list.count()); ASSERT_EQ(0, readonly_rs_list.count()); ASSERT_EQ(1234, rs_list[0].sql_port_); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(3, rs_list.count()); ASSERT_EQ(0, readonly_rs_list.count()); set_response_json("{}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(0, rs_list.count()); ASSERT_EQ(0, readonly_rs_list.count()); set_response_json("{\"RsList\":[{\"address_NO\":\"127.0.0.1:99\",\"role_NO\":\"FOLLOWER\"}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":{}}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\",\"sql_port\":\"abc\"}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":123,\"role\":\"FOLLOWER\",\"sql_port\":88}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":0,\"sql_port\":88}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); //check readonly_rs_list set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.5:99\",\"role\":\"FOLLOWER\"}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(3, rs_list.count()); ASSERT_EQ(2, readonly_rs_list.count()); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(3, rs_list.count()); ASSERT_EQ(1, readonly_rs_list.count()); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(3, rs_list.count()); ASSERT_EQ(0, readonly_rs_list.count()); set_response_json("{\"RsList\":[], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_EQ(OB_SUCCESS, ret); ASSERT_EQ(0, rs_list.count()); ASSERT_EQ(1, readonly_rs_list.count()); set_response_json("{\"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":123,\"role\":\"FOLLOWER\", \"sql_port\": 1234}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role_NO\":\"FOLLOWER\", \"sql_port\": 1234}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":{}}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address_NO\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role_NO\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port_NO\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":123,\"role\":\"FOLLOWER\", \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":0, \"sql_port\": 1234}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); set_response_json("{\"RsList\":[{\"address\":\"127.0.0.1:99\",\"role\":\"FOLLOWER\", \"sql_port\": 1234},{\"sql_port\": 1234,\"address\":\"127.0.0.2:99\",\"role\":\"LEADER\"},{\"sql_port\": 1234,\"address\":\"127.0.0.3:99\",\"role\":\"FOLLOWER\"}], \"ReadonlyRsList\":[{\"address\":\"127.0.0.4:99\",\"role\":\"FOLLOWER\", \"sql_port\": \"abc\"}]}"); ret = ws_.fetch(rs_list, readonly_rs_list); ASSERT_NE(OB_SUCCESS, ret); } TEST_F(TestWebServiceRootAddr, invalid) { ObWebServiceRootAddr ws; ObSEArray rs; ObSEArray readonly_rs; ASSERT_NE(OB_SUCCESS, ws.fetch(rs, readonly_rs)); ASSERT_NE(OB_SUCCESS, ws.store(rs, readonly_rs, true, LEADER)); ASSERT_EQ(OB_SUCCESS, ws.init(initer_.get_config())); initer_.get_config().obconfig_url.set_value("http://127.127.127.127:1/obconfig/region1"); ASSERT_NE(OB_SUCCESS, ws.fetch(rs, readonly_rs)); ASSERT_NE(OB_SUCCESS, ws.store(rs, readonly_rs, true, LEADER)); ObRootAddr ra; ra.server_.set_ip_addr("127.0.0.1", 9988); rs.push_back(ra); ASSERT_NE(OB_SUCCESS, ws.store(rs, readonly_rs, true, LEADER)); } } // 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(); }