add loadbalancer unit test

CHANGELOG.md

@@ -6,3 +6,4 @@
 - [Feature: add rate limit filter debug log](https://github.com/Tencent/spring-cloud-tencent/pull/437)
 - [Add configurable heartbeat interval support](https://github.com/Tencent/spring-cloud-tencent/pull/444)
 - [feat:enhance Feign and RestTemplate and support Polaris monitor.](https://github.com/Tencent/spring-cloud-tencent/pull/447)
+- [loadBalancer module add unit test.](https://github.com/Tencent/spring-cloud-tencent/issues/421)

spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java

@@ -0,0 +1,124 @@
+/*
+ * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
+ *
+ * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
+ *
+ * Licensed under the BSD 3-Clause License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed
+ * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
+ * CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ *
+ */
+
+package com.tencent.cloud.polaris.loadbalancer;
+
+import com.tencent.cloud.common.pojo.PolarisServiceInstance;
+import com.tencent.cloud.common.util.ApplicationContextAwareUtils;
+import com.tencent.cloud.polaris.loadbalancer.config.PolarisLoadBalancerProperties;
+import com.tencent.polaris.api.pojo.Instance;
+import com.tencent.polaris.router.api.core.RouterAPI;
+import com.tencent.polaris.router.api.rpc.ProcessLoadBalanceResponse;
+import org.junit.AfterClass;
+import org.junit.Assert;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.beans.factory.ObjectProvider;
+import org.springframework.cloud.client.ServiceInstance;
+import org.springframework.cloud.client.loadbalancer.Request;
+import org.springframework.cloud.client.loadbalancer.Response;
+import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static com.tencent.cloud.common.metadata.MetadataContext.LOCAL_NAMESPACE;
+import static com.tencent.cloud.common.metadata.MetadataContext.LOCAL_SERVICE;
+import static org.mockito.Mockito.*;
+
+/**
+ * Test for {@link PolarisLoadBalancer}
+ *
+ * @author rod.xu
+ * @date 2022/7/21 5:44 下午
+ */
+@RunWith(MockitoJUnitRunner.class)
+public class PolarisLoadBalancerTest {
+
+	@Mock
+	private RouterAPI routerAPI;
+	@Mock
+	private ObjectProvider<ServiceInstanceListSupplier> supplierObjectProvider;
+	@Mock
+	private PolarisLoadBalancerProperties loadBalancerProperties;
+
+	private static MockedStatic<ApplicationContextAwareUtils> mockedApplicationContextAwareUtils;
+	private static Instance TestInstance;
+
+	@BeforeClass
+	public static void beforeClass() {
+		mockedApplicationContextAwareUtils = Mockito.mockStatic(ApplicationContextAwareUtils.class);
+		mockedApplicationContextAwareUtils.when(() -> ApplicationContextAwareUtils.getProperties(anyString()))
+				.thenReturn("unit-test");
+
+		TestInstance = Instance.createDefaultInstance("instance-id", LOCAL_NAMESPACE,
+				LOCAL_SERVICE, "host", 8090);
+	}
+
+	@AfterClass
+	public static void afterClass() {
+		mockedApplicationContextAwareUtils.close();
+	}
+
+	@Test
+	public void chooseNormalLogicTest_thenReturnAvailablePolarisInstance() {
+
+		Request request = Mockito.mock(Request.class);
+		List<ServiceInstance> mockInstanceList = new ArrayList<>();
+		mockInstanceList.add(new PolarisServiceInstance(TestInstance));
+
+		ServiceInstanceListSupplier serviceInstanceListSupplier = Mockito.mock(ServiceInstanceListSupplier.class);
+		when(serviceInstanceListSupplier.get(request)).thenReturn(Flux.just(mockInstanceList));
+
+		when(supplierObjectProvider.getIfAvailable(any())).thenReturn(serviceInstanceListSupplier);
+		when(loadBalancerProperties.getEnabled()).thenReturn(true);
+
+		ProcessLoadBalanceResponse mockLbRes = new ProcessLoadBalanceResponse(TestInstance);
+		when(routerAPI.processLoadBalance(any())).thenReturn(mockLbRes);
+
+		// request construct and execute invoke
+		PolarisLoadBalancer polarisLoadBalancer = new PolarisLoadBalancer(LOCAL_SERVICE, supplierObjectProvider,
+				loadBalancerProperties, routerAPI);
+		Mono<Response<ServiceInstance>> responseMono = polarisLoadBalancer.choose(request);
+		ServiceInstance serviceInstance = responseMono.block().getServer();
+
+		// verify method has invoked
+		verify(loadBalancerProperties).getEnabled();
+		verify(supplierObjectProvider).getIfAvailable(any());
+
+		//result assert
+		Assert.assertNotNull(serviceInstance);
+		Assert.assertTrue("polaris service instance",
+				serviceInstance instanceof PolarisServiceInstance);
+
+		PolarisServiceInstance polarisServiceInstance = (PolarisServiceInstance) serviceInstance;
+		Assert.assertEquals("instance-id", polarisServiceInstance.getPolarisInstance().getId());
+		Assert.assertEquals(LOCAL_NAMESPACE, polarisServiceInstance.getPolarisInstance().getNamespace());
+		Assert.assertEquals(LOCAL_SERVICE, polarisServiceInstance.getPolarisInstance().getService());
+		Assert.assertEquals("host", polarisServiceInstance.getPolarisInstance().getHost());
+		Assert.assertEquals(8090, polarisServiceInstance.getPolarisInstance().getPort());
+	}
+
+}

spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisServiceInstanceListSupplierTest.java

@@ -0,0 +1,89 @@
+/*
+ * Tencent is pleased to support the open source community by making Spring Cloud Tencent available.
+ *
+ * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
+ *
+ * Licensed under the BSD 3-Clause License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed
+ * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
+ * CONDITIONS OF ANY KIND, either express or implied. See the License for the
+ * specific language governing permissions and limitations under the License.
+ *
+ */
+
+package com.tencent.cloud.polaris.loadbalancer;
+
+import com.tencent.cloud.common.pojo.PolarisServiceInstance;
+import com.tencent.cloud.common.util.ApplicationContextAwareUtils;
+import org.junit.Assert;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockedStatic;
+import org.mockito.Mockito;
+import org.mockito.junit.MockitoJUnitRunner;
+import org.springframework.cloud.client.DefaultServiceInstance;
+import org.springframework.cloud.client.ServiceInstance;
+import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import static com.tencent.cloud.common.metadata.MetadataContext.LOCAL_NAMESPACE;
+import static org.mockito.ArgumentMatchers.anyString;
+
+/**
+ * Test for {@link PolarisServiceInstanceListSupplier}
+ *
+ * @author rod.xu
+ * @date 2022/7/21 5:45 下午
+ */
+@RunWith(MockitoJUnitRunner.class)
+public class PolarisServiceInstanceListSupplierTest {
+
+	@Mock
+	private ServiceInstanceListSupplier serviceInstanceListSupplier;
+
+	@Test
+	public void chooseInstancesTest() {
+		try (MockedStatic<ApplicationContextAwareUtils> mockedApplicationContextAwareUtils = Mockito.mockStatic(ApplicationContextAwareUtils.class)) {
+			mockedApplicationContextAwareUtils.when(() -> ApplicationContextAwareUtils.getProperties(anyString()))
+					.thenReturn("test-unit");
+
+			PolarisServiceInstanceListSupplier instanceListSupplier =
+					new PolarisServiceInstanceListSupplier(serviceInstanceListSupplier);
+
+			List<ServiceInstance> allServers = new ArrayList<>();
+			ServiceInstance instance1 = new DefaultServiceInstance("unit-test-instanceId-01",
+					"unit-test-serviceId", "unit-test-host-01", 8090, false);
+			ServiceInstance instance2 = new DefaultServiceInstance("unit-test-instanceId-02",
+					"unit-test-serviceId", "unit-test-host-02", 8090, false);
+
+			allServers.add(instance1);
+			allServers.add(instance2);
+
+			List<ServiceInstance> polarisInstanceList = instanceListSupplier.chooseInstances(allServers);
+			Assert.assertNotNull(polarisInstanceList);
+			Assert.assertEquals(polarisInstanceList.size(), allServers.size());
+			for (ServiceInstance serviceInstance : polarisInstanceList) {
+				Assert.assertTrue("", serviceInstance instanceof PolarisServiceInstance);
+				PolarisServiceInstance polarisServiceInstance = (PolarisServiceInstance) serviceInstance;
+
+				Assert.assertFalse(polarisServiceInstance.isSecure());
+				Assert.assertEquals("unit-test-serviceId", polarisServiceInstance.getPolarisInstance().getService());
+				Assert.assertEquals(8090, polarisServiceInstance.getPolarisInstance().getPort());
+				Assert.assertEquals(LOCAL_NAMESPACE, polarisServiceInstance.getPolarisInstance().getNamespace());
+
+				Assert.assertTrue("instance id assert",
+						polarisServiceInstance.getPolarisInstance().getId().startsWith("unit-test-instanceId"));
+				Assert.assertTrue("host assert",
+						polarisServiceInstance.getPolarisInstance().getHost().startsWith("unit-test-host"));
+			}
+		}
+	}
+}