From 5e30641d6cf39d169fae9b98b1bf3e1b6feea577 Mon Sep 17 00:00:00 2001
From: dapan1121 <wpan@taosdata.com>
Date: Thu, 16 Dec 2021 18:50:21 +0800
Subject: [PATCH] ut test

---
 source/libs/catalog/src/catalog.c         | 13 +++++++-
 source/libs/catalog/test/catalogTests.cpp | 40 ++++++++++++++++++++++-
 2 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/source/libs/catalog/src/catalog.c b/source/libs/catalog/src/catalog.c
index 3aac02c1e4..248cdbe51c 100644
--- a/source/libs/catalog/src/catalog.c
+++ b/source/libs/catalog/src/catalog.c
@@ -54,9 +54,20 @@ int32_t ctgGetDBVgroupFromMnode(struct SCatalog* pCatalog, void *pRpc, const SEp
     return code;
   }
 
+  char *pMsg = rpcMallocCont(msgLen);
+  if (NULL == pMsg) {
+    ctgError("rpc malloc %d failed", msgLen);
+    tfree(msg);
+    return TSDB_CODE_CTG_MEM_ERROR;
+  }
+
+  memcpy(pMsg, msg, msgLen);
+
+  tfree(msg);
+  
   SRpcMsg rpcMsg = {
       .msgType = TSDB_MSG_TYPE_USE_DB,
-      .pCont   = msg,
+      .pCont   = pMsg,
       .contLen = msgLen,
   };
 
diff --git a/source/libs/catalog/test/catalogTests.cpp b/source/libs/catalog/test/catalogTests.cpp
index 0493ddfe8c..e14c58d412 100644
--- a/source/libs/catalog/test/catalogTests.cpp
+++ b/source/libs/catalog/test/catalogTests.cpp
@@ -28,6 +28,7 @@
 #include "tvariant.h"
 #include "catalog.h"
 #include "tep.h"
+#include "trpc.h"
 
 typedef struct SAppInstInfo {
   int64_t           numOfConns;
@@ -50,6 +51,41 @@ typedef struct STscObj {
 
 namespace {
 
+void sendCreateDbMsg(void *shandle, SEpSet *pEpSet) {
+  SCreateDbMsg* pReq = (SCreateDbMsg*)rpcMallocCont(sizeof(SCreateDbMsg));
+  strcpy(pReq->db, "1.db1");
+  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->minRowsPerFileBlock = htonl(100);
+  pReq->maxRowsPerFileBlock = 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 rpcMsg = {0};
+  rpcMsg.pCont = pReq;
+  rpcMsg.contLen = sizeof(SCreateDbMsg);
+  rpcMsg.msgType = TSDB_MSG_TYPE_CREATE_DB;
+
+  SRpcMsg rpcRsp = {0};
+
+  rpcSendRecv(shandle, pEpSet, &rpcMsg, &rpcRsp);
+
+  ASSERT_EQ(rpcRsp.code, 0);
+}
+
 }
 
 TEST(testCase, normalCase) {
@@ -57,13 +93,15 @@ TEST(testCase, normalCase) {
   assert(pConn != NULL);
 
   char *clusterId = "cluster1";
-  char *dbname = "db1";
+  char *dbname = "1.db1";
   char *tablename = "table1";
   struct SCatalog* pCtg = NULL;
   void *mockPointer = (void *)0x1;
   SVgroupInfo vgInfo = {0};
 
   msgInit();
+
+  sendCreateDbMsg(pConn->pTransporter, &pConn->pAppInfo->mgmtEp.epSet);
   
   int32_t code = catalogInit(NULL);
   ASSERT_EQ(code, 0);
-- 
GitLab