TDengine/source/dnode/mnode/impl/test/user/user.cpp

632 lines
18 KiB
C++
Raw Normal View History

2021-12-16 09:05:13 +00:00
/**
2021-12-16 12:16:35 +00:00
* @file user.cpp
2021-12-16 09:05:13 +00:00
* @author slguan (slguan@taosdata.com)
2022-01-04 09:17:25 +00:00
* @brief MNODE module user tests
* @version 1.0
2022-01-04 12:18:09 +00:00
* @date 2022-01-04
2021-12-07 12:14:22 +00:00
*
2022-01-04 12:18:09 +00:00
* @copyright Copyright (c) 2022
2021-12-07 12:14:22 +00:00
*
*/
2022-01-04 12:36:54 +00:00
#include "sut.h"
2021-12-07 12:14:22 +00:00
2022-01-04 12:36:54 +00:00
class MndTestUser : public ::testing::Test {
2021-12-07 12:14:22 +00:00
protected:
2022-01-04 09:17:25 +00:00
static void SetUpTestSuite() { test.Init("/tmp/mnode_test_user", 9011); }
2021-12-22 10:39:32 +00:00
static void TearDownTestSuite() { test.Cleanup(); }
2021-12-07 12:14:22 +00:00
2021-12-22 10:39:32 +00:00
static Testbase test;
2021-12-07 12:14:22 +00:00
2021-12-13 10:19:09 +00:00
public:
void SetUp() override {}
void TearDown() override {}
};
2022-01-04 12:36:54 +00:00
Testbase MndTestUser::test;
2021-12-13 10:19:09 +00:00
2022-01-04 12:36:54 +00:00
TEST_F(MndTestUser, 01_Show_User) {
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-22 10:39:32 +00:00
CHECK_META("show users", 4);
CHECK_SCHEMA(0, TSDB_DATA_TYPE_BINARY, TSDB_USER_LEN + VARSTR_HEADER_SIZE, "name");
CHECK_SCHEMA(1, TSDB_DATA_TYPE_BINARY, 10 + VARSTR_HEADER_SIZE, "privilege");
CHECK_SCHEMA(2, TSDB_DATA_TYPE_TIMESTAMP, 8, "create_time");
CHECK_SCHEMA(3, TSDB_DATA_TYPE_BINARY, TSDB_USER_LEN + VARSTR_HEADER_SIZE, "account");
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-22 10:39:32 +00:00
EXPECT_EQ(test.GetShowRows(), 1);
2021-12-13 10:19:09 +00:00
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("super", 10);
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
2021-12-07 12:14:22 +00:00
}
2022-01-04 12:36:54 +00:00
TEST_F(MndTestUser, 02_Create_User) {
2021-12-31 10:40:49 +00:00
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "");
strcpy(createReq.pass, "p1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_INVALID_USER_FORMAT);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u1");
strcpy(createReq.pass, "");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_INVALID_PASS_FORMAT);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "root");
strcpy(createReq.pass, "1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_USER_ALREADY_EXIST);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u1");
strcpy(createReq.pass, "p1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2022-02-11 06:09:03 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 2);
CheckBinary("u1", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("normal", 10);
CheckBinary("super", 10);
CheckTimestamp();
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-31 10:40:49 +00:00
}
2022-02-11 06:09:03 +00:00
{
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 1);
}
{
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u2");
strcpy(createReq.pass, "p1");
createReq.superUser = 1;
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 2);
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("u2", TSDB_USER_LEN);
CheckBinary("super", 10);
CheckBinary("super", 10);
CheckTimestamp();
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
}
{
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u2");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 1);
}
2021-12-31 10:40:49 +00:00
}
2022-01-04 12:36:54 +00:00
TEST_F(MndTestUser, 03_Alter_User) {
2022-02-11 06:09:03 +00:00
{
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u3");
strcpy(createReq.pass, "p1");
createReq.superUser = 1;
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 2);
}
2021-12-31 10:40:49 +00:00
{
2022-02-11 05:13:32 +00:00
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_PASSWD;
strcpy(alterReq.user, "");
strcpy(alterReq.pass, "p1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_INVALID_USER_FORMAT);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_PASSWD;
2022-02-11 06:09:03 +00:00
strcpy(alterReq.user, "u3");
2022-02-11 05:13:32 +00:00
strcpy(alterReq.pass, "");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_INVALID_PASS_FORMAT);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_PASSWD;
strcpy(alterReq.user, "u4");
strcpy(alterReq.pass, "1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_USER_NOT_EXIST);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_PASSWD;
2022-02-11 06:09:03 +00:00
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_SUPERUSER;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
alterReq.superUser = 1;
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_CLEAR_WRITE_DB;
strcpy(alterReq.user, "u3");
2022-02-11 05:13:32 +00:00
strcpy(alterReq.pass, "1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-31 10:40:49 +00:00
}
2022-02-11 06:09:03 +00:00
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_CLEAR_READ_DB;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_ADD_READ_DB;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
strcpy(alterReq.dbname, "d1");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_DB_NOT_EXIST);
}
{
int32_t contLen = sizeof(SCreateDbReq);
SCreateDbReq* pReq = (SCreateDbReq*)rpcMallocCont(contLen);
strcpy(pReq->db, "1.d2");
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;
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_DB, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_ADD_READ_DB;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
strcpy(alterReq.dbname, "1.d2");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_ADD_READ_DB;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
strcpy(alterReq.dbname, "1.d2");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
2022-02-11 07:20:27 +00:00
{
SGetUserAuthReq authReq = {0};
strcpy(authReq.user, "u3");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSGetUserAuthReq(NULL, 0, &authReq);
2022-02-11 07:20:27 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSGetUserAuthReq(pReq, contLen, &authReq);
2022-02-11 07:20:27 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_GET_USER_AUTH, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
SGetUserAuthRsp authRsp = {0};
2022-02-11 09:48:26 +00:00
tDeserializeSGetUserAuthRsp(pRsp->pCont, pRsp->contLen, &authRsp);
2022-02-11 07:20:27 +00:00
EXPECT_STREQ(authRsp.user, "u3");
EXPECT_EQ(authRsp.superAuth, 1);
int32_t numOfReadDbs = taosHashGetSize(authRsp.readDbs);
int32_t numOfWriteDbs = taosHashGetSize(authRsp.writeDbs);
EXPECT_EQ(numOfReadDbs, 1);
EXPECT_EQ(numOfWriteDbs, 0);
char* dbname = (char*)taosHashGet(authRsp.readDbs, "1.d2", 5);
EXPECT_STREQ(dbname, "1.d2");
taosHashCleanup(authRsp.readDbs);
taosHashCleanup(authRsp.writeDbs);
}
2022-02-11 06:09:03 +00:00
{
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_REMOVE_READ_DB;
strcpy(alterReq.user, "u3");
strcpy(alterReq.pass, "1");
strcpy(alterReq.dbname, "1.d2");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
{
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u3");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
test.SendShowRetrieveReq();
EXPECT_EQ(test.GetShowRows(), 1);
}
2021-12-31 10:40:49 +00:00
}
2022-02-11 06:09:03 +00:00
TEST_F(MndTestUser, 05_Drop_User) {
2021-12-31 10:40:49 +00:00
{
2022-02-11 05:13:32 +00:00
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_INVALID_USER_FORMAT);
2021-12-31 10:40:49 +00:00
}
{
2022-02-11 05:13:32 +00:00
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u4");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, TSDB_CODE_MND_USER_NOT_EXIST);
2021-12-31 10:40:49 +00:00
}
2022-02-11 06:09:03 +00:00
{
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u1");
strcpy(createReq.pass, "p1");
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 06:09:03 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2022-02-11 06:09:03 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
}
2021-12-31 10:40:49 +00:00
{
2022-02-11 05:13:32 +00:00
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u1");
2021-12-31 10:40:49 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2021-12-31 10:40:49 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-31 10:40:49 +00:00
}
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-31 10:40:49 +00:00
CHECK_META("show users", 4);
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-31 10:40:49 +00:00
EXPECT_EQ(test.GetShowRows(), 1);
}
2022-02-11 06:09:03 +00:00
TEST_F(MndTestUser, 06_Create_Drop_Alter_User) {
2021-12-13 11:02:26 +00:00
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u1");
strcpy(createReq.pass, "p1");
2021-12-22 10:39:32 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-13 11:02:26 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-13 11:02:26 +00:00
}
2021-12-07 12:14:22 +00:00
2021-12-13 11:02:26 +00:00
{
2022-02-11 05:13:32 +00:00
SCreateUserReq createReq = {0};
strcpy(createReq.user, "u2");
strcpy(createReq.pass, "p2");
2021-12-22 10:39:32 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSCreateUserReq(NULL, 0, &createReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSCreateUserReq(pReq, contLen, &createReq);
2021-12-07 12:14:22 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_CREATE_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-13 11:02:26 +00:00
}
2021-12-07 12:14:22 +00:00
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-22 10:39:32 +00:00
CHECK_META("show users", 4);
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-22 10:39:32 +00:00
EXPECT_EQ(test.GetShowRows(), 3);
2021-12-13 10:19:09 +00:00
CheckBinary("u1", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("u2", TSDB_USER_LEN);
2021-12-13 10:19:09 +00:00
CheckBinary("normal", 10);
CheckBinary("super", 10);
2021-12-13 11:02:26 +00:00
CheckBinary("normal", 10);
CheckTimestamp();
2021-12-13 10:19:09 +00:00
CheckTimestamp();
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("root", TSDB_USER_LEN);
2021-12-07 13:10:36 +00:00
2021-12-16 06:02:39 +00:00
{
2022-02-11 05:13:32 +00:00
SAlterUserReq alterReq = {0};
alterReq.alterType = TSDB_ALTER_USER_PASSWD;
strcpy(alterReq.user, "u1");
strcpy(alterReq.pass, "p2");
2021-12-22 10:39:32 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSAlterUserReq(NULL, 0, &alterReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSAlterUserReq(pReq, contLen, &alterReq);
2021-12-07 13:10:36 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_ALTER_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-16 06:02:39 +00:00
}
2021-12-22 10:39:32 +00:00
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-22 10:39:32 +00:00
CHECK_META("show users", 4);
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-22 10:39:32 +00:00
EXPECT_EQ(test.GetShowRows(), 3);
2021-12-13 10:19:09 +00:00
CheckBinary("u1", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("u2", TSDB_USER_LEN);
2021-12-13 10:19:09 +00:00
CheckBinary("normal", 10);
CheckBinary("super", 10);
2021-12-13 11:02:26 +00:00
CheckBinary("normal", 10);
CheckTimestamp();
2021-12-13 10:19:09 +00:00
CheckTimestamp();
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("root", TSDB_USER_LEN);
2021-12-07 12:14:22 +00:00
2021-12-16 06:02:39 +00:00
{
2022-02-11 05:13:32 +00:00
SDropUserReq dropReq = {0};
strcpy(dropReq.user, "u1");
2021-12-07 12:14:22 +00:00
2022-02-11 09:48:26 +00:00
int32_t contLen = tSerializeSDropUserReq(NULL, 0, &dropReq);
2022-02-11 05:13:32 +00:00
void* pReq = rpcMallocCont(contLen);
2022-02-11 09:48:26 +00:00
tSerializeSDropUserReq(pReq, contLen, &dropReq);
2021-12-07 12:14:22 +00:00
2022-01-05 12:18:56 +00:00
SRpcMsg* pRsp = test.SendReq(TDMT_MND_DROP_USER, pReq, contLen);
ASSERT_NE(pRsp, nullptr);
ASSERT_EQ(pRsp->code, 0);
2021-12-16 06:02:39 +00:00
}
2021-12-22 10:39:32 +00:00
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-22 10:39:32 +00:00
CHECK_META("show users", 4);
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-22 10:39:32 +00:00
EXPECT_EQ(test.GetShowRows(), 2);
2021-12-13 10:19:09 +00:00
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("u2", TSDB_USER_LEN);
2021-12-13 10:19:09 +00:00
CheckBinary("super", 10);
2021-12-13 11:02:26 +00:00
CheckBinary("normal", 10);
CheckTimestamp();
2021-12-13 10:19:09 +00:00
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
2021-12-13 11:02:26 +00:00
CheckBinary("root", TSDB_USER_LEN);
2021-12-16 06:02:39 +00:00
// restart
2021-12-22 10:39:32 +00:00
test.Restart();
2021-12-13 11:02:26 +00:00
2022-01-05 12:18:56 +00:00
test.SendShowMetaReq(TSDB_MGMT_TABLE_USER, "");
2021-12-22 10:39:32 +00:00
CHECK_META("show users", 4);
2021-12-13 11:02:26 +00:00
2022-01-05 12:18:56 +00:00
test.SendShowRetrieveReq();
2021-12-22 10:39:32 +00:00
EXPECT_EQ(test.GetShowRows(), 2);
2021-12-13 11:02:26 +00:00
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("u2", TSDB_USER_LEN);
CheckBinary("super", 10);
CheckBinary("normal", 10);
CheckTimestamp();
CheckTimestamp();
CheckBinary("root", TSDB_USER_LEN);
CheckBinary("root", TSDB_USER_LEN);
2021-12-31 10:40:49 +00:00
}