310 lines
8.9 KiB
C++
310 lines
8.9 KiB
C++
/**
|
|
* @file stb.cpp
|
|
* @author slguan (slguan@taosdata.com)
|
|
* @brief MNODE module stb tests
|
|
* @version 1.0
|
|
* @date 2022-01-12
|
|
*
|
|
* @copyright Copyright (c) 2022
|
|
*
|
|
*/
|
|
|
|
#include "sut.h"
|
|
|
|
class MndTestStb : public ::testing::Test {
|
|
protected:
|
|
static void SetUpTestSuite() { test.Init("/tmp/mnode_test_stb", 9034); }
|
|
static void TearDownTestSuite() { test.Cleanup(); }
|
|
|
|
static Testbase test;
|
|
|
|
public:
|
|
void SetUp() override {}
|
|
void TearDown() override {}
|
|
|
|
SCreateDbReq* BuildCreateDbReq(const char* dbname, int32_t* pContLen);
|
|
SMCreateStbReq* BuildCreateStbReq(const char* stbname, int32_t* pContLen);
|
|
SMAltertbReq* BuildAlterStbAddTagReq(const char* stbname, int32_t* pContLen);
|
|
};
|
|
|
|
Testbase MndTestStb::test;
|
|
|
|
SCreateDbReq* MndTestStb::BuildCreateDbReq(const char* dbname, int32_t* pContLen) {
|
|
int32_t contLen = sizeof(SCreateDbReq);
|
|
|
|
SCreateDbReq* pReq = (SCreateDbReq*)rpcMallocCont(contLen);
|
|
strcpy(pReq->db, dbname);
|
|
pReq->numOfVgroups = htonl(2);
|
|
pReq->cacheBlockSize = htonl(16);
|
|
pReq->totalBlocks = htonl(10);
|
|
pReq->daysPerFile = htonl(10);
|
|
pReq->daysToKeep0 = htonl(3650);
|
|
pReq->daysToKeep1 = htonl(3650);
|
|
pReq->daysToKeep2 = htonl(3650);
|
|
pReq->minRows = htonl(100);
|
|
pReq->maxRows = htonl(4096);
|
|
pReq->commitTime = htonl(3600);
|
|
pReq->fsyncPeriod = htonl(3000);
|
|
pReq->walLevel = 1;
|
|
pReq->precision = 0;
|
|
pReq->compression = 2;
|
|
pReq->replications = 1;
|
|
pReq->quorum = 1;
|
|
pReq->update = 0;
|
|
pReq->cacheLastRow = 0;
|
|
pReq->ignoreExist = 1;
|
|
|
|
*pContLen = contLen;
|
|
return pReq;
|
|
}
|
|
|
|
SMCreateStbReq* MndTestStb::BuildCreateStbReq(const char *stbname, int32_t* pContLen) {
|
|
int32_t cols = 2;
|
|
int32_t tags = 3;
|
|
int32_t contLen = (tags + cols) * sizeof(SSchema) + sizeof(SMCreateStbReq);
|
|
|
|
SMCreateStbReq* pReq = (SMCreateStbReq*)rpcMallocCont(contLen);
|
|
strcpy(pReq->name, stbname);
|
|
pReq->numOfTags = htonl(tags);
|
|
pReq->numOfColumns = htonl(cols);
|
|
|
|
{
|
|
SSchema* pSchema = &pReq->pSchemas[0];
|
|
pSchema->bytes = htonl(8);
|
|
pSchema->type = TSDB_DATA_TYPE_TIMESTAMP;
|
|
strcpy(pSchema->name, "ts");
|
|
}
|
|
|
|
{
|
|
SSchema* pSchema = &pReq->pSchemas[1];
|
|
pSchema->bytes = htonl(4);
|
|
pSchema->type = TSDB_DATA_TYPE_INT;
|
|
strcpy(pSchema->name, "col1");
|
|
}
|
|
|
|
{
|
|
SSchema* pSchema = &pReq->pSchemas[2];
|
|
pSchema->bytes = htonl(2);
|
|
pSchema->type = TSDB_DATA_TYPE_TINYINT;
|
|
strcpy(pSchema->name, "tag1");
|
|
}
|
|
|
|
{
|
|
SSchema* pSchema = &pReq->pSchemas[3];
|
|
pSchema->bytes = htonl(8);
|
|
pSchema->type = TSDB_DATA_TYPE_BIGINT;
|
|
strcpy(pSchema->name, "tag2");
|
|
}
|
|
|
|
{
|
|
SSchema* pSchema = &pReq->pSchemas[4];
|
|
pSchema->bytes = htonl(16);
|
|
pSchema->type = TSDB_DATA_TYPE_BINARY;
|
|
strcpy(pSchema->name, "tag3");
|
|
}
|
|
|
|
*pContLen = contLen;
|
|
return pReq;
|
|
}
|
|
|
|
// TEST_F(MndTestStb, 01_Create_Show_Meta_Drop_Restart_Stb) {
|
|
// const char *dbname = "1.d1";
|
|
// const char *stbname = "1.d1.stb";
|
|
|
|
// {
|
|
// int32_t contLen = 0;
|
|
// SCreateDbReq* pReq = BuildCreateDbReq(dbname, &contLen);
|
|
// SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_DB, pReq, contLen);
|
|
// ASSERT_NE(pRsp, nullptr);
|
|
// ASSERT_EQ(pRsp->code, 0);
|
|
// }
|
|
|
|
// {
|
|
// int32_t contLen = 0;
|
|
// SMCreateStbReq* pReq = BuildCreateStbReq(stbname, &contLen);
|
|
// SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_STB, pReq, contLen);
|
|
// ASSERT_NE(pRsp, nullptr);
|
|
// ASSERT_EQ(pRsp->code, 0);
|
|
// }
|
|
|
|
// {
|
|
// test.SendShowMetaReq(TSDB_MGMT_TABLE_STB, dbname);
|
|
// CHECK_META("show stables", 4);
|
|
// CHECK_SCHEMA(0, TSDB_DATA_TYPE_BINARY, TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE, "name");
|
|
// CHECK_SCHEMA(1, TSDB_DATA_TYPE_TIMESTAMP, 8, "create_time");
|
|
// CHECK_SCHEMA(2, TSDB_DATA_TYPE_INT, 4, "columns");
|
|
// CHECK_SCHEMA(3, TSDB_DATA_TYPE_INT, 4, "tags");
|
|
|
|
// test.SendShowRetrieveReq();
|
|
// EXPECT_EQ(test.GetShowRows(), 1);
|
|
// CheckBinary("stb", TSDB_TABLE_NAME_LEN);
|
|
// CheckTimestamp();
|
|
// CheckInt32(2);
|
|
// CheckInt32(3);
|
|
// }
|
|
|
|
// // ----- meta ------
|
|
// {
|
|
// int32_t contLen = sizeof(STableInfoReq);
|
|
// STableInfoReq* pReq = (STableInfoReq*)rpcMallocCont(contLen);
|
|
// strcpy(pReq->dbFName, dbname);
|
|
// strcpy(pReq->tbName, "stb");
|
|
|
|
// SRpcMsg* pMsg = test.SendReq(TDMT_MND_STB_META, pReq, contLen);
|
|
// ASSERT_NE(pMsg, nullptr);
|
|
// ASSERT_EQ(pMsg->code, 0);
|
|
|
|
// STableMetaRsp* pRsp = (STableMetaRsp*)pMsg->pCont;
|
|
// pRsp->numOfTags = htonl(pRsp->numOfTags);
|
|
// pRsp->numOfColumns = htonl(pRsp->numOfColumns);
|
|
// pRsp->sversion = htonl(pRsp->sversion);
|
|
// pRsp->tversion = htonl(pRsp->tversion);
|
|
// pRsp->suid = be64toh(pRsp->suid);
|
|
// pRsp->tuid = be64toh(pRsp->tuid);
|
|
// pRsp->vgId = be64toh(pRsp->vgId);
|
|
// for (int32_t i = 0; i < pRsp->numOfTags + pRsp->numOfColumns; ++i) {
|
|
// SSchema* pSchema = &pRsp->pSchema[i];
|
|
// pSchema->colId = htonl(pSchema->colId);
|
|
// pSchema->bytes = htonl(pSchema->bytes);
|
|
// }
|
|
|
|
// EXPECT_STREQ(pRsp->dbFName, dbname);
|
|
// EXPECT_STREQ(pRsp->tbName, "stb");
|
|
// EXPECT_STREQ(pRsp->stbName, "stb");
|
|
// EXPECT_EQ(pRsp->numOfColumns, 2);
|
|
// EXPECT_EQ(pRsp->numOfTags, 3);
|
|
// EXPECT_EQ(pRsp->precision, TSDB_TIME_PRECISION_MILLI);
|
|
// EXPECT_EQ(pRsp->tableType, TSDB_SUPER_TABLE);
|
|
// EXPECT_EQ(pRsp->update, 0);
|
|
// EXPECT_EQ(pRsp->sversion, 1);
|
|
// EXPECT_EQ(pRsp->tversion, 0);
|
|
// EXPECT_GT(pRsp->suid, 0);
|
|
// EXPECT_GT(pRsp->tuid, 0);
|
|
// EXPECT_EQ(pRsp->vgId, 0);
|
|
|
|
// {
|
|
// SSchema* pSchema = &pRsp->pSchema[0];
|
|
// EXPECT_EQ(pSchema->type, TSDB_DATA_TYPE_TIMESTAMP);
|
|
// EXPECT_EQ(pSchema->colId, 1);
|
|
// EXPECT_EQ(pSchema->bytes, 8);
|
|
// EXPECT_STREQ(pSchema->name, "ts");
|
|
// }
|
|
|
|
// {
|
|
// SSchema* pSchema = &pRsp->pSchema[1];
|
|
// EXPECT_EQ(pSchema->type, TSDB_DATA_TYPE_INT);
|
|
// EXPECT_EQ(pSchema->colId, 2);
|
|
// EXPECT_EQ(pSchema->bytes, 4);
|
|
// EXPECT_STREQ(pSchema->name, "col1");
|
|
// }
|
|
|
|
// {
|
|
// SSchema* pSchema = &pRsp->pSchema[2];
|
|
// EXPECT_EQ(pSchema->type, TSDB_DATA_TYPE_TINYINT);
|
|
// EXPECT_EQ(pSchema->colId, 3);
|
|
// EXPECT_EQ(pSchema->bytes, 2);
|
|
// EXPECT_STREQ(pSchema->name, "tag1");
|
|
// }
|
|
|
|
// {
|
|
// SSchema* pSchema = &pRsp->pSchema[3];
|
|
// EXPECT_EQ(pSchema->type, TSDB_DATA_TYPE_BIGINT);
|
|
// EXPECT_EQ(pSchema->colId, 4);
|
|
// EXPECT_EQ(pSchema->bytes, 8);
|
|
// EXPECT_STREQ(pSchema->name, "tag2");
|
|
// }
|
|
|
|
// {
|
|
// SSchema* pSchema = &pRsp->pSchema[4];
|
|
// EXPECT_EQ(pSchema->type, TSDB_DATA_TYPE_BINARY);
|
|
// EXPECT_EQ(pSchema->colId, 5);
|
|
// EXPECT_EQ(pSchema->bytes, 16);
|
|
// EXPECT_STREQ(pSchema->name, "tag3");
|
|
// }
|
|
// }
|
|
|
|
// // restart
|
|
// test.Restart();
|
|
|
|
// {
|
|
// test.SendShowMetaReq(TSDB_MGMT_TABLE_STB, dbname);
|
|
// CHECK_META("show stables", 4);
|
|
// test.SendShowRetrieveReq();
|
|
// EXPECT_EQ(test.GetShowRows(), 1);
|
|
|
|
// CheckBinary("stb", TSDB_TABLE_NAME_LEN);
|
|
// CheckTimestamp();
|
|
// CheckInt32(2);
|
|
// CheckInt32(3);
|
|
// }
|
|
|
|
// {
|
|
// int32_t contLen = sizeof(SMDropStbReq);
|
|
|
|
// SMDropStbReq* pReq = (SMDropStbReq*)rpcMallocCont(contLen);
|
|
// strcpy(pReq->name, stbname);
|
|
|
|
// SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_STB, pReq, contLen);
|
|
// ASSERT_NE(pRsp, nullptr);
|
|
// ASSERT_EQ(pRsp->code, 0);
|
|
// }
|
|
|
|
// test.SendShowMetaReq(TSDB_MGMT_TABLE_STB, dbname);
|
|
// CHECK_META("show stables", 4);
|
|
// test.SendShowRetrieveReq();
|
|
// EXPECT_EQ(test.GetShowRows(), 0);
|
|
// }
|
|
|
|
SMAltertbReq* MndTestStb::BuildAlterStbAddTagReq(const char* stbname, int32_t* pContLen) {
|
|
int32_t contLen = sizeof(SMAltertbReq) + sizeof(SSchema);
|
|
SMAltertbReq* pReq = (SMAltertbReq*)rpcMallocCont(contLen);
|
|
strcpy(pReq->name, stbname);
|
|
pReq->numOfSchemas = htonl(1);
|
|
pReq->alterType = TSDB_ALTER_TABLE_ADD_TAG_COLUMN;
|
|
|
|
SSchema* pSchema = &pReq->pSchemas[0];
|
|
pSchema->bytes = htonl(4);
|
|
pSchema->type = TSDB_DATA_TYPE_INT;
|
|
strcpy(pSchema->name, "tag4");
|
|
|
|
*pContLen = contLen;
|
|
return pReq;
|
|
}
|
|
|
|
TEST_F(MndTestStb, 01_Alter_Stb) {
|
|
const char *dbname = "1.d2";
|
|
const char *stbname = "1.d2.stb";
|
|
|
|
{
|
|
int32_t contLen = 0;
|
|
SCreateDbReq* pReq = BuildCreateDbReq(dbname, &contLen);
|
|
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_DB, pReq, contLen);
|
|
ASSERT_NE(pRsp, nullptr);
|
|
ASSERT_EQ(pRsp->code, 0);
|
|
}
|
|
|
|
{
|
|
int32_t contLen = 0;
|
|
SMCreateStbReq* pReq = BuildCreateStbReq(stbname, &contLen);
|
|
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_STB, pReq, contLen);
|
|
ASSERT_NE(pRsp, nullptr);
|
|
ASSERT_EQ(pRsp->code, 0);
|
|
}
|
|
|
|
{
|
|
int32_t contLen = 0;
|
|
SMAltertbReq* pReq = BuildAlterStbAddTagReq(stbname, &contLen);
|
|
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_STB, pReq, contLen);
|
|
ASSERT_NE(pRsp, nullptr);
|
|
ASSERT_EQ(pRsp->code, 0);
|
|
|
|
test.SendShowMetaReq(TSDB_MGMT_TABLE_STB, dbname);
|
|
test.SendShowRetrieveReq();
|
|
EXPECT_EQ(test.GetShowRows(), 1);
|
|
CheckBinary("stb", TSDB_TABLE_NAME_LEN);
|
|
CheckTimestamp();
|
|
CheckInt32(2);
|
|
CheckInt32(4);
|
|
}
|
|
}
|