From fc7575279cb2e8f06fdf451e5ffb6a6f4d78dad4 Mon Sep 17 00:00:00 2001
From: Shungang Li <sgli@taosdata.com>
Date: Fri, 15 Mar 2024 11:11:07 +0800
Subject: [PATCH] fix: arb unit test

---
 source/dnode/mnode/impl/test/arbgroup/arbgroup.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/source/dnode/mnode/impl/test/arbgroup/arbgroup.cpp b/source/dnode/mnode/impl/test/arbgroup/arbgroup.cpp
index 73c71c0431..887a410fdd 100644
--- a/source/dnode/mnode/impl/test/arbgroup/arbgroup.cpp
+++ b/source/dnode/mnode/impl/test/arbgroup/arbgroup.cpp
@@ -114,6 +114,8 @@ TEST_F(ArbgroupTest, 02_process_heart_beat_rsp) {
   group.assignedLeader.dnodeId = dnodeId;
   strncpy(group.assignedLeader.token, group.members[0].state.token, TSDB_ARB_TOKEN_SIZE);
 
+  taosThreadMutexInit(&group.mutex, NULL);
+
   // --------------------------------------------------------------------------------
   {  // expired hb => skip
     SVArbHbRspMember rspMember = {0};
@@ -164,6 +166,8 @@ TEST_F(ArbgroupTest, 02_process_heart_beat_rsp) {
     EXPECT_EQ(newGroup.assignedLeader.dnodeId, 0);
     EXPECT_EQ(std::string(newGroup.assignedLeader.token).size(), 0);
   }
+
+  taosThreadMutexDestroy(&group.mutex);
 }
 
 TEST_F(ArbgroupTest, 03_process_check_sync_rsp) {
@@ -184,6 +188,8 @@ TEST_F(ArbgroupTest, 03_process_check_sync_rsp) {
 
   group.isSync = 0;
 
+  taosThreadMutexInit(&group.mutex, NULL);
+
   // --------------------------------------------------------------------------------
   {  // token mismatch => skip
     char member0Token[TSDB_ARB_TOKEN_SIZE] = {0};
@@ -211,6 +217,8 @@ TEST_F(ArbgroupTest, 03_process_check_sync_rsp) {
     EXPECT_TRUE(updateIsSync);
     EXPECT_TRUE(newGroup.isSync);
   }
+
+  taosThreadMutexDestroy(&group.mutex);
 }
 
 TEST_F(ArbgroupTest, 04_process_set_assigned_leader){
@@ -233,6 +241,8 @@ TEST_F(ArbgroupTest, 04_process_set_assigned_leader){
   group.assignedLeader.dnodeId = dnodeId;
   strncpy(group.assignedLeader.token, group.members[0].state.token, TSDB_ARB_TOKEN_SIZE);
 
+  taosThreadMutexInit(&group.mutex, NULL);
+
   // --------------------------------------------------------------------------------
   {  // token mismatch => skip
     char memberToken[TSDB_ARB_TOKEN_SIZE] = {0};
@@ -269,6 +279,8 @@ TEST_F(ArbgroupTest, 04_process_set_assigned_leader){
     EXPECT_TRUE(updateAssigned);
     EXPECT_FALSE(newGroup.isSync);
   }
+
+  taosThreadMutexDestroy(&group.mutex);
 }
 
 #pragma GCC diagnostic pop