154 lines
3.9 KiB
C++
154 lines
3.9 KiB
C++
/**
|
|
* @file dnode.cpp
|
|
* @author slguan (slguan@taosdata.com)
|
|
* @brief DNODE module dnode-msg tests
|
|
* @version 0.1
|
|
* @date 2021-12-15
|
|
*
|
|
* @copyright Copyright (c) 2021
|
|
*
|
|
*/
|
|
|
|
#include "base.h"
|
|
|
|
class DndTestSnode : public ::testing::Test {
|
|
public:
|
|
void SetUp() override {}
|
|
void TearDown() override {}
|
|
|
|
public:
|
|
static void SetUpTestSuite() {
|
|
test.Init("/tmp/dnode_test_snode1", 9066);
|
|
const char* fqdn = "localhost";
|
|
const char* firstEp = "localhost:9066";
|
|
|
|
server2.Start("/tmp/dnode_test_snode2", fqdn, 9067, firstEp);
|
|
taosMsleep(300);
|
|
}
|
|
|
|
static void TearDownTestSuite() {
|
|
server2.Stop();
|
|
test.Cleanup();
|
|
}
|
|
|
|
static Testbase test;
|
|
static TestServer server2;
|
|
};
|
|
|
|
Testbase DndTestSnode::test;
|
|
TestServer DndTestSnode::server2;
|
|
|
|
TEST_F(DndTestSnode, 01_ShowSnode) {
|
|
test.SendShowMetaMsg(TSDB_MGMT_TABLE_SNODE, "");
|
|
CHECK_META("show snodes", 3);
|
|
|
|
CHECK_SCHEMA(0, TSDB_DATA_TYPE_SMALLINT, 2, "id");
|
|
CHECK_SCHEMA(1, TSDB_DATA_TYPE_BINARY, TSDB_EP_LEN + VARSTR_HEADER_SIZE, "endpoint");
|
|
CHECK_SCHEMA(2, TSDB_DATA_TYPE_TIMESTAMP, 8, "create_time");
|
|
|
|
test.SendShowRetrieveMsg();
|
|
EXPECT_EQ(test.GetShowRows(), 0);
|
|
}
|
|
|
|
TEST_F(DndTestSnode, 02_Create_Snode_Invalid_Id) {
|
|
{
|
|
int32_t contLen = sizeof(SMCreateSnodeMsg);
|
|
|
|
SMCreateSnodeMsg* pReq = (SMCreateSnodeMsg*)rpcMallocCont(contLen);
|
|
pReq->dnodeId = htonl(1);
|
|
|
|
SRpcMsg* pMsg = test.SendMsg(TDMT_MND_CREATE_SNODE, pReq, contLen);
|
|
ASSERT_NE(pMsg, nullptr);
|
|
ASSERT_EQ(pMsg->code, 0);
|
|
|
|
test.SendShowMetaMsg(TSDB_MGMT_TABLE_SNODE, "");
|
|
CHECK_META("show snodes", 3);
|
|
|
|
CHECK_SCHEMA(0, TSDB_DATA_TYPE_SMALLINT, 2, "id");
|
|
CHECK_SCHEMA(1, TSDB_DATA_TYPE_BINARY, TSDB_EP_LEN + VARSTR_HEADER_SIZE, "endpoint");
|
|
CHECK_SCHEMA(2, TSDB_DATA_TYPE_TIMESTAMP, 8, "create_time");
|
|
|
|
test.SendShowRetrieveMsg();
|
|
EXPECT_EQ(test.GetShowRows(), 1);
|
|
|
|
CheckInt16(1);
|
|
CheckBinary("localhost:9066", TSDB_EP_LEN);
|
|
CheckTimestamp();
|
|
}
|
|
}
|
|
|
|
TEST_F(DndTestSnode, 03_Create_Snode_Invalid_Id) {
|
|
{
|
|
int32_t contLen = sizeof(SMCreateSnodeMsg);
|
|
|
|
SMCreateSnodeMsg* pReq = (SMCreateSnodeMsg*)rpcMallocCont(contLen);
|
|
pReq->dnodeId = htonl(2);
|
|
|
|
SRpcMsg* pMsg = test.SendMsg(TDMT_MND_CREATE_SNODE, pReq, contLen);
|
|
ASSERT_NE(pMsg, nullptr);
|
|
ASSERT_EQ(pMsg->code, TSDB_CODE_MND_DNODE_NOT_EXIST);
|
|
}
|
|
}
|
|
|
|
TEST_F(DndTestSnode, 04_Create_Snode) {
|
|
{
|
|
// create dnode
|
|
int32_t contLen = sizeof(SCreateDnodeMsg);
|
|
|
|
SCreateDnodeMsg* pReq = (SCreateDnodeMsg*)rpcMallocCont(contLen);
|
|
strcpy(pReq->fqdn, "localhost");
|
|
pReq->port = htonl(9067);
|
|
|
|
SRpcMsg* pMsg = test.SendMsg(TDMT_MND_CREATE_DNODE, pReq, contLen);
|
|
ASSERT_NE(pMsg, nullptr);
|
|
ASSERT_EQ(pMsg->code, 0);
|
|
|
|
taosMsleep(1300);
|
|
test.SendShowMetaMsg(TSDB_MGMT_TABLE_DNODE, "");
|
|
test.SendShowRetrieveMsg();
|
|
EXPECT_EQ(test.GetShowRows(), 2);
|
|
}
|
|
|
|
{
|
|
// create snode
|
|
int32_t contLen = sizeof(SMCreateSnodeMsg);
|
|
|
|
SMCreateSnodeMsg* pReq = (SMCreateSnodeMsg*)rpcMallocCont(contLen);
|
|
pReq->dnodeId = htonl(2);
|
|
|
|
SRpcMsg* pMsg = test.SendMsg(TDMT_MND_CREATE_SNODE, pReq, contLen);
|
|
ASSERT_NE(pMsg, nullptr);
|
|
ASSERT_EQ(pMsg->code, 0);
|
|
|
|
test.SendShowMetaMsg(TSDB_MGMT_TABLE_SNODE, "");
|
|
test.SendShowRetrieveMsg();
|
|
EXPECT_EQ(test.GetShowRows(), 2);
|
|
|
|
CheckInt16(1);
|
|
CheckInt16(2);
|
|
CheckBinary("localhost:9066", TSDB_EP_LEN);
|
|
CheckBinary("localhost:9067", TSDB_EP_LEN);
|
|
CheckTimestamp();
|
|
CheckTimestamp();
|
|
}
|
|
|
|
{
|
|
// drop snode
|
|
int32_t contLen = sizeof(SMDropSnodeMsg);
|
|
|
|
SMDropSnodeMsg* pReq = (SMDropSnodeMsg*)rpcMallocCont(contLen);
|
|
pReq->dnodeId = htonl(2);
|
|
|
|
SRpcMsg* pMsg = test.SendMsg(TDMT_MND_DROP_SNODE, pReq, contLen);
|
|
ASSERT_NE(pMsg, nullptr);
|
|
ASSERT_EQ(pMsg->code, 0);
|
|
|
|
test.SendShowMetaMsg(TSDB_MGMT_TABLE_SNODE, "");
|
|
test.SendShowRetrieveMsg();
|
|
EXPECT_EQ(test.GetShowRows(), 1);
|
|
|
|
CheckInt16(1);
|
|
CheckBinary("localhost:9066", TSDB_EP_LEN);
|
|
CheckTimestamp();
|
|
}
|
|
} |