test: add test cases for RaftMsgExecute under lock package (apache#6003)

changes/en-us/2.0.0.md

@@ -172,6 +172,7 @@ The version is updated as follows:
 - [[#5863](https://github.com/seata/seata/pull/5863)] fix unit test in java 21
 - [[#5986](https://github.com/seata/seata/pull/5986)] fix zookeeper UT failed
 - [[#5995](https://github.com/seata/seata/pull/5995)] add test cases for RaftClusterMetadataMsg
+- [[#6003](https://github.com/seata/seata/pull/6003)] add test cases for RaftMsgExecute under lock package
 
 
 ### Contributors:

changes/zh-cn/2.0.0.md

@@ -173,6 +173,7 @@ Seata 是一款开源的分布式事务解决方案，提供高性能和简单
 - [[#5863](https://github.com/seata/seata/pull/5863)] 修复单元测试在Java21下无法正常运行的问题。
 - [[#5986](https://github.com/seata/seata/pull/5986)] 修复 zookeeper 单测失败问题
 - [[#5995](https://github.com/seata/seata/pull/5995)] 添加 RaftClusterMetadataMsg 模块的单元测试用例
+- [[#6003](https://github.com/seata/seata/pull/6003)] 添加 RaftMsgExecute 模块 lock 包下的单元测试用例
 
 ### Contributors:
 

server/src/main/java/io/seata/server/cluster/raft/RaftServerFactory.java

@@ -160,6 +160,8 @@ public void start() {
     @Override
     public void destroy() {
         this.close();
+        rpcServer = null;
+        RAFT_SERVER_MAP.clear();
     }
 
     @Override

server/src/main/java/io/seata/server/lock/LockerManagerFactory.java

@@ -52,6 +52,9 @@ public static void init() {
         init(null);
     }
 
+    public static void destroy() {
+        LOCK_MANAGER = null;
+    }
 
     public static void init(LockMode lockMode) {
         if (LOCK_MANAGER == null) {

server/src/main/java/io/seata/server/session/SessionHolder.java

@@ -383,6 +383,7 @@ public static void destroy() {
         if (ROOT_SESSION_MANAGER != null) {
             ROOT_SESSION_MANAGER.destroy();
         }
+        SESSION_MANAGER_MAP = null;
     }
 
     @FunctionalInterface

server/src/test/java/io/seata/server/raft/execute/LockExecuteTest.java

@@ -0,0 +1,163 @@
+/*
+ *  Copyright 1999-2019 Seata.io Group.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+package io.seata.server.raft.execute;
+
+import io.seata.common.util.NetUtil;
+import io.seata.config.ConfigurationCache;
+import io.seata.core.exception.TransactionException;
+import io.seata.core.model.BranchType;
+import io.seata.server.cluster.raft.execute.lock.BranchReleaseLockExecute;
+import io.seata.server.cluster.raft.execute.lock.GlobalReleaseLockExecute;
+import io.seata.server.cluster.raft.sync.msg.RaftBranchSessionSyncMsg;
+import io.seata.server.cluster.raft.sync.msg.RaftGlobalSessionSyncMsg;
+import io.seata.server.cluster.raft.sync.msg.dto.BranchTransactionDTO;
+import io.seata.server.cluster.raft.sync.msg.dto.GlobalTransactionDTO;
+import io.seata.server.lock.LockManager;
+import io.seata.server.lock.LockerManagerFactory;
+import io.seata.server.session.BranchSession;
+import io.seata.server.session.GlobalSession;
+import io.seata.server.session.SessionHolder;
+import io.seata.server.storage.SessionConverter;
+import io.seata.server.store.StoreConfig;
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.ApplicationContext;
+
+import static io.seata.common.DefaultValues.DEFAULT_TX_GROUP;
+
+
+/**
+ * @author ptyin
+ */
+@SpringBootTest
+class LockExecuteTest {
+
+    private static GlobalSession GLOBAL_SESSION;
+
+    private static final String XID = "123:123";
+
+    private static final long BRANCH_ID = 0L;
+
+    @BeforeAll
+    public static void setUp(ApplicationContext context) throws TransactionException {
+        System.setProperty("server.raft.serverAddr", NetUtil.getLocalIp() + ":9091");
+        SessionHolder.init(StoreConfig.SessionMode.RAFT);
+        LockerManagerFactory.destroy();
+        LockerManagerFactory.init(StoreConfig.LockMode.RAFT);
+    }
+
+    @AfterAll
+    public static void destroy() throws TransactionException {
+        // Clear configuration
+        ConfigurationCache.clear();
+        System.clearProperty("server.raft.serverAddr");
+
+        // Destroy SessionHolder and LockerManagerFactory
+        SessionHolder.destroy();
+        SessionHolder.init(null);
+        LockerManagerFactory.destroy();
+    }
+
+    @BeforeEach
+    public void addGlobalSession() throws TransactionException {
+        GLOBAL_SESSION = mockGlobalSession();
+        SessionHolder.getRootSessionManager().addGlobalSession(GLOBAL_SESSION);
+    }
+
+    @AfterEach
+    public void removeTestSession() throws TransactionException {
+        SessionHolder.getRootSessionManager().removeGlobalSession(GLOBAL_SESSION);
+    }
+
+    @Test
+    public void testGlobalRelease() throws Throwable {
+        BranchSession branchSession1 = mockBranchSession("test:0");
+        BranchSession branchSession2 = mockBranchSession("test:1");
+        GLOBAL_SESSION.add(branchSession1);
+        GLOBAL_SESSION.add(branchSession2);
+
+        LockManager lockerManager = LockerManagerFactory.getLockManager();
+        Assertions.assertTrue(lockerManager.acquireLock(branchSession1));
+        Assertions.assertTrue(lockerManager.acquireLock(branchSession2));
+        Assertions.assertEquals(1, branchSession1.getLockHolder().values().size());
+        Assertions.assertEquals(1, branchSession2.getLockHolder().values().size());
+
+        GlobalReleaseLockExecute execute = new GlobalReleaseLockExecute();
+        boolean success = execute.execute(convertToGlobalSessionMsg(GLOBAL_SESSION));
+        Assertions.assertTrue(success);
+        Assertions.assertEquals(0, branchSession1.getLockHolder().values().size());
+        Assertions.assertEquals(0, branchSession2.getLockHolder().values().size());
+    }
+
+    @Test
+    public void testBranchRelease() throws Throwable {
+        BranchSession branchSession = mockBranchSession("test:0");
+        GLOBAL_SESSION.add(branchSession);
+
+        LockManager lockerManager = LockerManagerFactory.getLockManager();
+        Assertions.assertTrue(lockerManager.acquireLock(branchSession));
+        Assertions.assertEquals(1, branchSession.getLockHolder().values().size());
+
+        BranchReleaseLockExecute execute = new BranchReleaseLockExecute();
+        boolean success = execute.execute(convertToBranchSessionMsg(branchSession));
+        Assertions.assertTrue(success);
+        Assertions.assertEquals(0, branchSession.getLockHolder().values().size());
+    }
+
+    private static GlobalSession mockGlobalSession() {
+        GlobalSession session = new GlobalSession("test", "test", "test", 5000);
+        session.setXid(XID);
+        session.setApplicationData("hello, world");
+        session.setTransactionId(123);
+        session.setBeginTime(System.currentTimeMillis());
+        return session;
+    }
+
+    private static BranchSession mockBranchSession(String lockKey) {
+        BranchSession session = new BranchSession();
+        session.setXid(XID);
+        session.setTransactionId(123);
+        session.setBranchId(BRANCH_ID);
+        session.setClientId("client");
+        session.setResourceGroupId(DEFAULT_TX_GROUP);
+        session.setResourceId("db");
+        session.setLockKey(lockKey);
+        session.setBranchType(BranchType.AT);
+        session.setApplicationData("hello, world");
+        return session;
+    }
+
+    private static RaftGlobalSessionSyncMsg convertToGlobalSessionMsg(GlobalSession globalSession) {
+        RaftGlobalSessionSyncMsg sessionMsg = new RaftGlobalSessionSyncMsg();
+        GlobalTransactionDTO dto = new GlobalTransactionDTO();
+        SessionConverter.convertGlobalTransactionDO(dto, globalSession);
+        sessionMsg.setGlobalSession(dto);
+        return sessionMsg;
+    }
+
+    private static RaftBranchSessionSyncMsg convertToBranchSessionMsg(BranchSession branchSession) {
+        RaftBranchSessionSyncMsg sessionMsg = new RaftBranchSessionSyncMsg();
+        BranchTransactionDTO dto = SessionConverter.convertBranchTransactionDTO(branchSession);
+        sessionMsg.setBranchSession(dto);
+        return sessionMsg;
+    }
+}