From 995317bf9cca443d2740c4ee413c213f6e86da8c Mon Sep 17 00:00:00 2001
From: Minghao Li <castermode@gmail.com>
Date: Tue, 31 May 2022 20:38:59 +0800
Subject: [PATCH] enh(sync): add test syncSnapshotSendTest.cpp

---
 source/libs/sync/test/CMakeLists.txt          |  14 +++
 .../libs/sync/test/syncSnapshotSendTest.cpp   | 100 ++++++++++++++++++
 2 files changed, 114 insertions(+)
 create mode 100644 source/libs/sync/test/syncSnapshotSendTest.cpp

diff --git a/source/libs/sync/test/CMakeLists.txt b/source/libs/sync/test/CMakeLists.txt
index cfbdf0e961..20188d51f3 100644
--- a/source/libs/sync/test/CMakeLists.txt
+++ b/source/libs/sync/test/CMakeLists.txt
@@ -38,6 +38,7 @@ add_executable(syncRespMgrTest "")
 add_executable(syncSnapshotTest "")
 add_executable(syncApplyMsgTest "")
 add_executable(syncConfigChangeTest "")
+add_executable(syncSnapshotSendTest "")
 
 
 target_sources(syncTest
@@ -200,6 +201,10 @@ target_sources(syncConfigChangeTest
     PRIVATE
     "syncConfigChangeTest.cpp"
 )
+target_sources(syncSnapshotSendTest
+    PRIVATE
+    "syncSnapshotSendTest.cpp"
+)
 
 
 target_include_directories(syncTest
@@ -402,6 +407,11 @@ target_include_directories(syncConfigChangeTest
     "${TD_SOURCE_DIR}/include/libs/sync"
     "${CMAKE_CURRENT_SOURCE_DIR}/../inc"
 )
+target_include_directories(syncSnapshotSendTest
+    PUBLIC
+    "${TD_SOURCE_DIR}/include/libs/sync"
+    "${CMAKE_CURRENT_SOURCE_DIR}/../inc"
+)
 
 
 target_link_libraries(syncTest
@@ -564,6 +574,10 @@ target_link_libraries(syncConfigChangeTest
     sync
     gtest_main
 )
+target_link_libraries(syncSnapshotSendTest
+    sync
+    gtest_main
+)
 
 
 enable_testing()
diff --git a/source/libs/sync/test/syncSnapshotSendTest.cpp b/source/libs/sync/test/syncSnapshotSendTest.cpp
new file mode 100644
index 0000000000..8985a3f1a2
--- /dev/null
+++ b/source/libs/sync/test/syncSnapshotSendTest.cpp
@@ -0,0 +1,100 @@
+#include <gtest/gtest.h>
+#include <stdio.h>
+#include "syncIO.h"
+#include "syncInt.h"
+#include "syncMessage.h"
+#include "syncUtil.h"
+
+void logTest() {
+  sTrace("--- sync log test: trace");
+  sDebug("--- sync log test: debug");
+  sInfo("--- sync log test: info");
+  sWarn("--- sync log test: warn");
+  sError("--- sync log test: error");
+  sFatal("--- sync log test: fatal");
+}
+
+SyncSnapshotSend *createMsg() {
+  SyncSnapshotSend *pMsg = syncSnapshotSendBuild(20, 1000);
+  pMsg->srcId.addr = syncUtilAddr2U64("127.0.0.1", 1234);
+  pMsg->srcId.vgId = 100;
+  pMsg->destId.addr = syncUtilAddr2U64("127.0.0.1", 5678);
+  pMsg->destId.vgId = 100;
+  pMsg->term = 11;
+  pMsg->lastIndex = 22;
+  pMsg->lastTerm = 33;
+  pMsg->seq = 44;
+  strcpy(pMsg->data, "hello world");
+  return pMsg;
+}
+
+void test1() {
+  SyncSnapshotSend *pMsg = createMsg();
+  syncSnapshotSendLog2((char *)"test1:", pMsg);
+  syncSnapshotSendDestroy(pMsg);
+}
+
+void test2() {
+  SyncSnapshotSend *pMsg = createMsg();
+  uint32_t           len = pMsg->bytes;
+  char *             serialized = (char *)taosMemoryMalloc(len);
+  syncSnapshotSendSerialize(pMsg, serialized, len);
+  SyncSnapshotSend *pMsg2 = syncSnapshotSendBuild(pMsg->dataLen, 1000);
+  syncSnapshotSendDeserialize(serialized, len, pMsg2);
+  syncSnapshotSendLog2((char *)"test2: syncSnapshotSendSerialize -> syncSnapshotSendDeserialize ", pMsg2);
+
+  taosMemoryFree(serialized);
+  syncSnapshotSendDestroy(pMsg);
+  syncSnapshotSendDestroy(pMsg2);
+}
+
+void test3() {
+  SyncSnapshotSend *pMsg = createMsg();
+  uint32_t           len;
+  char *             serialized = syncSnapshotSendSerialize2(pMsg, &len);
+  SyncSnapshotSend *pMsg2 = syncSnapshotSendDeserialize2(serialized, len);
+  syncSnapshotSendLog2((char *)"test3: syncSnapshotSendSerialize2 -> syncSnapshotSendDeserialize2 ", pMsg2);
+
+  taosMemoryFree(serialized);
+  syncSnapshotSendDestroy(pMsg);
+  syncSnapshotSendDestroy(pMsg2);
+}
+
+void test4() {
+  SyncSnapshotSend *pMsg = createMsg();
+  SRpcMsg            rpcMsg;
+  syncSnapshotSend2RpcMsg(pMsg, &rpcMsg);
+  SyncSnapshotSend *pMsg2 = (SyncSnapshotSend *)taosMemoryMalloc(rpcMsg.contLen);
+  syncSnapshotSendFromRpcMsg(&rpcMsg, pMsg2);
+  syncSnapshotSendLog2((char *)"test4: syncSnapshotSend2RpcMsg -> syncSnapshotSendFromRpcMsg ", pMsg2);
+
+  rpcFreeCont(rpcMsg.pCont);
+  syncSnapshotSendDestroy(pMsg);
+  syncSnapshotSendDestroy(pMsg2);
+}
+
+void test5() {
+  SyncSnapshotSend *pMsg = createMsg();
+  SRpcMsg            rpcMsg;
+  syncSnapshotSend2RpcMsg(pMsg, &rpcMsg);
+  SyncSnapshotSend *pMsg2 = syncSnapshotSendFromRpcMsg2(&rpcMsg);
+  syncSnapshotSendLog2((char *)"test5: syncSnapshotSend2RpcMsg -> syncSnapshotSendFromRpcMsg2 ", pMsg2);
+
+  rpcFreeCont(rpcMsg.pCont);
+  syncSnapshotSendDestroy(pMsg);
+  syncSnapshotSendDestroy(pMsg2);
+}
+
+int main() {
+  tsAsyncLog = 0;
+  sDebugFlag = DEBUG_TRACE + DEBUG_SCREEN + DEBUG_FILE;
+  logTest();
+
+  test1();
+  test2();
+  test3();
+  test4();
+  test5();
+
+  return 0;
+}
-- 
GitLab