From 7922997806c83c8dc53fcee356fe2ea46e8b5fef Mon Sep 17 00:00:00 2001
From: Minghao Li <castermode@gmail.com>
Date: Tue, 31 May 2022 20:54:39 +0800
Subject: [PATCH] enh(sync): add test syncSnapshotRspTest.cpp

---
 source/libs/sync/test/CMakeLists.txt          | 14 +++
 source/libs/sync/test/syncSnapshotRspTest.cpp | 99 +++++++++++++++++++
 2 files changed, 113 insertions(+)
 create mode 100644 source/libs/sync/test/syncSnapshotRspTest.cpp

diff --git a/source/libs/sync/test/CMakeLists.txt b/source/libs/sync/test/CMakeLists.txt
index 20188d51f3..b348051eb6 100644
--- a/source/libs/sync/test/CMakeLists.txt
+++ b/source/libs/sync/test/CMakeLists.txt
@@ -39,6 +39,7 @@ add_executable(syncSnapshotTest "")
 add_executable(syncApplyMsgTest "")
 add_executable(syncConfigChangeTest "")
 add_executable(syncSnapshotSendTest "")
+add_executable(syncSnapshotRspTest "")
 
 
 target_sources(syncTest
@@ -205,6 +206,10 @@ target_sources(syncSnapshotSendTest
     PRIVATE
     "syncSnapshotSendTest.cpp"
 )
+target_sources(syncSnapshotRspTest
+    PRIVATE
+    "syncSnapshotRspTest.cpp"
+)
 
 
 target_include_directories(syncTest
@@ -412,6 +417,11 @@ target_include_directories(syncSnapshotSendTest
     "${TD_SOURCE_DIR}/include/libs/sync"
     "${CMAKE_CURRENT_SOURCE_DIR}/../inc"
 )
+target_include_directories(syncSnapshotRspTest
+    PUBLIC
+    "${TD_SOURCE_DIR}/include/libs/sync"
+    "${CMAKE_CURRENT_SOURCE_DIR}/../inc"
+)
 
 
 target_link_libraries(syncTest
@@ -578,6 +588,10 @@ target_link_libraries(syncSnapshotSendTest
     sync
     gtest_main
 )
+target_link_libraries(syncSnapshotRspTest
+    sync
+    gtest_main
+)
 
 
 enable_testing()
diff --git a/source/libs/sync/test/syncSnapshotRspTest.cpp b/source/libs/sync/test/syncSnapshotRspTest.cpp
new file mode 100644
index 0000000000..9e455c6efd
--- /dev/null
+++ b/source/libs/sync/test/syncSnapshotRspTest.cpp
@@ -0,0 +1,99 @@
+#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");
+}
+
+SyncSnapshotRsp *createMsg() {
+  SyncSnapshotRsp *pMsg = syncSnapshotRspBuild(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->ack = 44;
+  return pMsg;
+}
+
+void test1() {
+  SyncSnapshotRsp *pMsg = createMsg();
+  syncSnapshotRspLog2((char *)"test1:", pMsg);
+  syncSnapshotRspDestroy(pMsg);
+}
+
+void test2() {
+  SyncSnapshotRsp *pMsg = createMsg();
+  uint32_t           len = pMsg->bytes;
+  char *             serialized = (char *)taosMemoryMalloc(len);
+  syncSnapshotRspSerialize(pMsg, serialized, len);
+  SyncSnapshotRsp *pMsg2 = syncSnapshotRspBuild(1000);
+  syncSnapshotRspDeserialize(serialized, len, pMsg2);
+  syncSnapshotRspLog2((char *)"test2: syncSnapshotRspSerialize -> syncSnapshotRspDeserialize ", pMsg2);
+
+  taosMemoryFree(serialized);
+  syncSnapshotRspDestroy(pMsg);
+  syncSnapshotRspDestroy(pMsg2);
+}
+
+void test3() {
+  SyncSnapshotRsp *pMsg = createMsg();
+  uint32_t           len;
+  char *             serialized = syncSnapshotRspSerialize2(pMsg, &len);
+  SyncSnapshotRsp *pMsg2 = syncSnapshotRspDeserialize2(serialized, len);
+  syncSnapshotRspLog2((char *)"test3: syncSnapshotRspSerialize2 -> syncSnapshotRspDeserialize2 ", pMsg2);
+
+  taosMemoryFree(serialized);
+  syncSnapshotRspDestroy(pMsg);
+  syncSnapshotRspDestroy(pMsg2);
+}
+
+void test4() {
+  SyncSnapshotRsp *pMsg = createMsg();
+  SRpcMsg            rpcMsg;
+  syncSnapshotRsp2RpcMsg(pMsg, &rpcMsg);
+  SyncSnapshotRsp *pMsg2 = (SyncSnapshotRsp *)taosMemoryMalloc(rpcMsg.contLen);
+  syncSnapshotRspFromRpcMsg(&rpcMsg, pMsg2);
+  syncSnapshotRspLog2((char *)"test4: syncSnapshotRsp2RpcMsg -> syncSnapshotRspFromRpcMsg ", pMsg2);
+
+  rpcFreeCont(rpcMsg.pCont);
+  syncSnapshotRspDestroy(pMsg);
+  syncSnapshotRspDestroy(pMsg2);
+}
+
+void test5() {
+  SyncSnapshotRsp *pMsg = createMsg();
+  SRpcMsg            rpcMsg;
+  syncSnapshotRsp2RpcMsg(pMsg, &rpcMsg);
+  SyncSnapshotRsp *pMsg2 = syncSnapshotRspFromRpcMsg2(&rpcMsg);
+  syncSnapshotRspLog2((char *)"test5: syncSnapshotRsp2RpcMsg -> syncSnapshotRspFromRpcMsg2 ", pMsg2);
+
+  rpcFreeCont(rpcMsg.pCont);
+  syncSnapshotRspDestroy(pMsg);
+  syncSnapshotRspDestroy(pMsg2);
+}
+
+int main() {
+  tsAsyncLog = 0;
+  sDebugFlag = DEBUG_TRACE + DEBUG_SCREEN + DEBUG_FILE;
+  logTest();
+
+  test1();
+  test2();
+  test3();
+  test4();
+  test5();
+
+  return 0;
+}