(shardingsphere) branch master updated: Add test cases on TransactionConnectionContext (#33185)

[duanzhengqiang]

This is an automated email from the ASF dual-hosted git repository.

duanzhengqiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 776c400da46 Add test cases on TransactionConnectionContext (#33185)
776c400da46 is described below

commit 776c400da468c0f94ad9090c9f6035b95c05902b
Author: Liang Zhang <zhangli...@apache.org>
AuthorDate: Wed Oct 9 17:41:27 2024 +0800

    Add test cases on TransactionConnectionContext (#33185)
---
 .../TransactionConnectionContextTest.java          | 79 ++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git 
a/infra/session/src/test/java/org/apache/shardingsphere/infra/session/connection/transaction/TransactionConnectionContextTest.java
 
b/infra/session/src/test/java/org/apache/shardingsphere/infra/session/connection/transaction/TransactionConnectionContextTest.java
new file mode 100644
index 00000000000..eb4bc69cc26
--- /dev/null
+++ 
b/infra/session/src/test/java/org/apache/shardingsphere/infra/session/connection/transaction/TransactionConnectionContextTest.java
@@ -0,0 +1,79 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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 org.apache.shardingsphere.infra.session.connection.transaction;
+
+import org.junit.jupiter.api.Test;
+
+import java.util.Optional;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+class TransactionConnectionContextTest {
+    
+    private final TransactionConnectionContext transactionConnectionContext = 
new TransactionConnectionContext();
+    
+    @Test
+    void assertBeginTransaction() {
+        transactionConnectionContext.beginTransaction("XA");
+        assertThat(transactionConnectionContext.getTransactionType(), 
is(Optional.of("XA")));
+        assertTrue(transactionConnectionContext.isInTransaction());
+    }
+    
+    @Test
+    void assertIsNotInDistributedTransactionWhenNotBegin() {
+        assertFalse(transactionConnectionContext.isInDistributedTransaction());
+    }
+    
+    @Test
+    void assertIsNotInDistributedTransactionWithLocal() {
+        transactionConnectionContext.beginTransaction("LOCAL");
+        assertFalse(transactionConnectionContext.isInDistributedTransaction());
+    }
+    
+    @Test
+    void assertIsInDistributedTransactionWithXA() {
+        transactionConnectionContext.beginTransaction("XA");
+        assertTrue(transactionConnectionContext.isInDistributedTransaction());
+    }
+    
+    @Test
+    void assertIsInDistributedTransactionWithBASE() {
+        transactionConnectionContext.beginTransaction("BASE");
+        assertTrue(transactionConnectionContext.isInDistributedTransaction());
+    }
+    
+    @Test
+    void assertGetReadWriteSplitReplicaRoute() {
+        transactionConnectionContext.setReadWriteSplitReplicaRoute("foo");
+        
assertThat(transactionConnectionContext.getReadWriteSplitReplicaRoute(), 
is(Optional.of("foo")));
+    }
+    
+    @Test
+    void assertClose() {
+        transactionConnectionContext.beginTransaction("XA");
+        transactionConnectionContext.close();
+        
assertFalse(transactionConnectionContext.getTransactionType().isPresent());
+        assertFalse(transactionConnectionContext.isInTransaction());
+        assertThat(transactionConnectionContext.getBeginMills(), is(0L));
+        assertFalse(transactionConnectionContext.isExceptionOccur());
+        
assertFalse(transactionConnectionContext.getReadWriteSplitReplicaRoute().isPresent());
+    }
+}