@@ -0,0 +1,3091 @@
+%0A/*%0A * Licensed to the Apache Software Foundation (ASF) under one or more%0A * contributor license agreements.  See the NOTICE file distributed with%0A * this work for additional information regarding copyright ownership.%0A * The ASF licenses this file to You under the Apache License, Version 2.0%0A * (the %22License%22); you may not use this file except in compliance with%0A * the License.  You may obtain a copy of the License at%0A *%0A *     http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0A%0Apackage org.apache.shardingsphere.proxy.backend.communication.jdbc.transaction;%0A%0Aimport com.google.common.collect.HashMultimap;%0Aimport com.google.common.collect.Multimap;%0Aimport lombok.SneakyThrows;%0Aimport org.apache.shardingsphere.proxy.backend.communication.jdbc.connection.BackendConnection;%0Aimport org.junit.Before;%0Aimport org.junit.Test;%0Aimport org.junit.runner.RunWith;%0Aimport org.mockito.Mock;%0Aimport org.mockito.junit.MockitoJUnitRunner;%0A%0Aimport java.sql.Connection;%0Aimport java.sql.SQLException;%0Aimport java.util.ArrayList;%0Aimport java.util.List;%0A%0Aimport static org.mockito.Mockito.verify;%0Aimport static org.mockito.Mockito.when;%0A%0A@RunWith(MockitoJUnitRunner.class)%0Apublic final class LocalTransactionManagerTest %7B%0A    %0A    @Mock%0A    private BackendConnection backendConnection;%0A    %0A    @Mock%0A    private TransactionStatus transactionStatus;%0A    %0A    @Mock%0A    private Connection connection;%0A    %0A    private LocalTransactionManager localTransactionManager;%0A    %0A    @Before%0A    public void setUp() throws SQLException %7B%0A        when(backendConnection.getTransactionStatus()).thenReturn(transactionStatus);%0A        when(backendConnection.getCachedConnections()).thenReturn(setCachedConnections());%0A        when(transactionStatus.isInTransaction()).thenReturn(true);%0A        localTransactionManager = new LocalTransactionManager(backendConnection);%0A    %7D%0A    %0A    private Multimap%3CString, Connection%3E setCachedConnections() %7B%0A        Multimap%3CString, Connection%3E result = HashMultimap.create();%0A        List%3CConnection%3E connections = new ArrayList%3C%3E(1);%0A        connections.add(connection);%0A        result.putAll(%22ds1%22, connections);%0A        return result;%0A    %7D%0A    %0A    @Test%0A    public void assertBegin() %7B%0A        localTransactionManager.begin();%0A        verify(backendConnection).getConnectionPostProcessors();%0A    %7D%0A    %0A    @Test%0A    @SneakyThrows(SQLException.class)%0A    public void assertCommit() %7B%0A        localTransactionManager.commit();%0A        verify(transactionStatus).isInTransaction();%0A        verify(connection).commit();%0A    %7D%0A    %0A    @Test%0A    @SneakyThrows(SQLException.class)%0A    public void assertRollback() %7B%0A        localTransactionManager.rollback();%0A        verify(transactionStatus).isInTransaction();%0A        verify(connection).rollback();%0A    %7D%0A%7D%0A

Add test for LocalTransactionManager (#10701)