@@ -0,0 +1,3974 @@
+/*%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.connection;%0A%0Aimport org.junit.Before;%0Aimport org.junit.Test;%0A%0Aimport java.lang.reflect.Field;%0Aimport java.util.concurrent.CountDownLatch;%0Aimport java.util.concurrent.ExecutorService;%0Aimport java.util.concurrent.Executors;%0Aimport java.util.concurrent.TimeUnit;%0Aimport java.util.concurrent.atomic.AtomicInteger;%0A%0Aimport static org.hamcrest.CoreMatchers.is;%0Aimport static org.junit.Assert.assertThat;%0A%0Apublic class ConnectionStatusTest %7B%0A    %0A    private int numberOfThreads = 10;%0A    %0A    private ExecutorService service;%0A    %0A    private ConnectionStatus connectionStatus;%0A    %0A    private CountDownLatch latch;%0A    %0A    private Field usingField;%0A    %0A    @Before%0A    public void setup() throws NoSuchFieldException %7B%0A        connectionStatus = new ConnectionStatus();%0A        service = Executors.newFixedThreadPool(numberOfThreads);%0A        latch = new CountDownLatch(numberOfThreads);%0A        usingField = ConnectionStatus.class.getDeclaredField(%22isUsing%22);%0A        usingField.setAccessible(true);%0A    %7D%0A    %0A    @Test%0A    public void assertSwitchToUsing() throws InterruptedException, IllegalAccessException %7B%0A        AtomicInteger counter = new AtomicInteger();%0A        for (int i = 0; i %3C numberOfThreads; i++) %7B%0A            service.submit(() -%3E %7B%0A                connectionStatus.switchToUsing();%0A                counter.incrementAndGet();%0A                latch.countDown();%0A            %7D);%0A        %7D%0A        latch.await();%0A        assertThat(numberOfThreads, is(counter.get()));%0A        assertThat(usingField.getBoolean(connectionStatus), is(true));%0A    %7D%0A    %0A    @Test%0A    public void assertSwitchToReleased() throws InterruptedException, IllegalAccessException %7B%0A        AtomicInteger counter = new AtomicInteger();%0A        for (int i = 0; i %3C numberOfThreads; i++) %7B%0A            service.submit(() -%3E %7B%0A                connectionStatus.switchToUsing();%0A                counter.incrementAndGet();%0A                connectionStatus.switchToReleased();%0A                counter.decrementAndGet();%0A                latch.countDown();%0A            %7D);%0A        %7D%0A        latch.await();%0A        assertThat(counter.get(), is(0));%0A        assertThat(usingField.getBoolean(connectionStatus), is(false));%0A    %7D%0A    %0A    @Test%0A    public void assertWaitUntilConnectionRelease() throws InterruptedException, IllegalAccessException %7B%0A        AtomicInteger counter = new AtomicInteger();%0A        for (int i = 0; i %3C numberOfThreads; i++) %7B%0A            service.submit(() -%3E %7B%0A                connectionStatus.switchToUsing();%0A                counter.incrementAndGet();%0A                connectionStatus.waitUntilConnectionRelease();%0A                counter.decrementAndGet();%0A                latch.countDown();%0A            %7D);%0A        %7D%0A        latch.await(200, TimeUnit.MILLISECONDS);%0A        assertThat(counter.get(), is(10));%0A        assertThat(usingField.getBoolean(connectionStatus), is(true));%0A        connectionStatus.switchToReleased();%0A        assertThat(usingField.getBoolean(connectionStatus), is(false));%0A        latch.await(300, TimeUnit.MILLISECONDS);%0A        assertThat(counter.get(), is(0));%0A    %7D%0A%7D%0A

Add ConnectionStatus Test Case (#7615)