@@ -0,0 +1,2969 @@
+/*%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.driver.governance.internal.state;%0A%0Aimport org.apache.shardingsphere.driver.governance.internal.circuit.connection.CircuitBreakerConnection;%0Aimport org.apache.shardingsphere.driver.jdbc.core.connection.ShardingSphereConnection;%0Aimport org.apache.shardingsphere.infra.context.metadata.MetaDataContexts;%0Aimport org.apache.shardingsphere.infra.state.StateContext;%0Aimport org.apache.shardingsphere.infra.state.StateEvent;%0Aimport org.apache.shardingsphere.infra.state.StateType;%0Aimport org.apache.shardingsphere.transaction.context.TransactionContexts;%0Aimport org.apache.shardingsphere.transaction.core.TransactionType;%0Aimport static org.hamcrest.CoreMatchers.instanceOf;%0Aimport static org.junit.Assert.assertThat;%0Aimport org.junit.Test;%0Aimport static org.mockito.Mockito.RETURNS_DEEP_STUBS;%0Aimport static org.mockito.Mockito.mock;%0A%0Aimport javax.sql.DataSource;%0Aimport java.sql.Connection;%0Aimport java.util.Collections;%0A%0Apublic class DriverStateContextTest %7B%0A    %0A    @Test%0A    public void assertGetConnection() %7B%0A        Connection actual1 = DriverStateContext.getConnection(%0A                Collections.singletonMap(%22ds%22, mock(DataSource.class, RETURNS_DEEP_STUBS)), mock(MetaDataContexts.class), mock(TransactionContexts.class, RETURNS_DEEP_STUBS), TransactionType.LOCAL);%0A        assertThat(actual1, instanceOf(ShardingSphereConnection.class));%0A        StateContext.switchState(new StateEvent(StateType.CIRCUIT_BREAK, true));%0A        Connection actual2 = DriverStateContext.getConnection(%0A                Collections.emptyMap(), mock(MetaDataContexts.class), mock(TransactionContexts.class, RETURNS_DEEP_STUBS), TransactionType.LOCAL);%0A        assertThat(actual2, instanceOf(CircuitBreakerConnection.class));%0A        StateContext.switchState(new StateEvent(StateType.CIRCUIT_BREAK, false));%0A        Connection actual3 = DriverStateContext.getConnection(%0A                Collections.singletonMap(%22ds%22, mock(DataSource.class, RETURNS_DEEP_STUBS)), mock(MetaDataContexts.class), mock(TransactionContexts.class, RETURNS_DEEP_STUBS), TransactionType.LOCAL);%0A        assertThat(actual3, instanceOf(ShardingSphereConnection.class));%0A    %7D%0A%7D%0A

Add test cases for DriverStateContext (#8235)