Fix test

apache · Dec 27, 2024 · cd39ff9 · cd39ff9
1 parent ffeb1c1
commit cd39ff9
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...shardingsphere/infra/session/connection/transaction/TransactionConnectionContextTest.java b/...shardingsphere/infra/session/connection/transaction/TransactionConnectionContextTest.java
@@ -25,14 +25,15 @@
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.Mockito.mock;
 
 class TransactionConnectionContextTest {
 
     private final TransactionConnectionContext transactionConnectionContext = new TransactionConnectionContext();
 
     @Test
     void assertBeginTransaction() {
-        transactionConnectionContext.beginTransaction("XA");
+        transactionConnectionContext.beginTransaction("XA", mock(TransactionManager.class));
         assertThat(transactionConnectionContext.getTransactionType(), is(Optional.of("XA")));
         assertTrue(transactionConnectionContext.isInTransaction());
     }
@@ -44,19 +45,19 @@ void assertIsNotInDistributedTransactionWhenNotBegin() {
 
     @Test
     void assertIsNotInDistributedTransactionWithLocal() {
-        transactionConnectionContext.beginTransaction("LOCAL");
+        transactionConnectionContext.beginTransaction("LOCAL", mock(TransactionManager.class));
         assertFalse(transactionConnectionContext.isInDistributedTransaction());
     }
 
     @Test
     void assertIsInDistributedTransactionWithXA() {
-        transactionConnectionContext.beginTransaction("XA");
+        transactionConnectionContext.beginTransaction("XA", mock(TransactionManager.class));
         assertTrue(transactionConnectionContext.isInDistributedTransaction());
     }
 
     @Test
     void assertIsInDistributedTransactionWithBASE() {
-        transactionConnectionContext.beginTransaction("BASE");
+        transactionConnectionContext.beginTransaction("BASE", mock(TransactionManager.class));
         assertTrue(transactionConnectionContext.isInDistributedTransaction());
     }
 
@@ -68,7 +69,7 @@ void assertGetReadWriteSplitReplicaRoute() {
 
     @Test
     void assertClose() {
-        transactionConnectionContext.beginTransaction("XA");
+        transactionConnectionContext.beginTransaction("XA", mock(TransactionManager.class));
         transactionConnectionContext.close();
         assertFalse(transactionConnectionContext.getTransactionType().isPresent());
         assertFalse(transactionConnectionContext.isInTransaction());