@@ -0,0 +1,2671 @@
+/*%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 */%0Apackage org.apache.rocketmq.client.latency;%0A%0Aimport java.util.concurrent.TimeUnit;%0Aimport org.junit.Before;%0Aimport org.junit.Test;%0A%0Aimport static org.assertj.core.api.Assertions.assertThat;%0A%0Apublic class LatencyFaultToleranceImplTest %7B%0A    private LatencyFaultTolerance%3CString%3E latencyFaultTolerance;%0A    private String brokerName = %22BrokerA%22;%0A    private String anotherBrokerName = %22BrokerB%22;%0A%0A    @Before%0A    public void init() %7B%0A        latencyFaultTolerance = new LatencyFaultToleranceImpl();%0A    %7D%0A%0A    @Test%0A    public void testUpdateFaultItem() throws Exception %7B%0A        latencyFaultTolerance.updateFaultItem(brokerName, 3000, 3000);%0A        assertThat(latencyFaultTolerance.isAvailable(brokerName)).isFalse();%0A        assertThat(latencyFaultTolerance.isAvailable(anotherBrokerName)).isTrue();%0A    %7D%0A%0A    @Test%0A    public void testIsAvailable() throws Exception %7B%0A        latencyFaultTolerance.updateFaultItem(brokerName, 3000, 50);%0A        assertThat(latencyFaultTolerance.isAvailable(brokerName)).isFalse();%0A%0A        TimeUnit.MILLISECONDS.sleep(70);%0A        assertThat(latencyFaultTolerance.isAvailable(brokerName)).isTrue();%0A    %7D%0A%0A    @Test%0A    public void testRemove() throws Exception %7B%0A        latencyFaultTolerance.updateFaultItem(brokerName, 3000, 3000);%0A        assertThat(latencyFaultTolerance.isAvailable(brokerName)).isFalse();%0A        latencyFaultTolerance.remove(brokerName);%0A        assertThat(latencyFaultTolerance.isAvailable(brokerName)).isTrue();%0A    %7D%0A%0A    @Test%0A    public void testPickOneAtLeast() throws Exception %7B%0A        latencyFaultTolerance.updateFaultItem(brokerName, 1000, 3000);%0A        assertThat(latencyFaultTolerance.pickOneAtLeast()).isEqualTo(brokerName);%0A%0A        latencyFaultTolerance.updateFaultItem(anotherBrokerName, 1001, 3000);%0A        assertThat(latencyFaultTolerance.pickOneAtLeast()).isEqualTo(brokerName);%0A    %7D%0A%7D

Add unit tests for LatencyFaultTolerance