From ed72fa84b57a5f2f07b07759ba028c079a1502be Mon Sep 17 00:00:00 2001 From: Haotian Zhang <928016560@qq.com> Date: Wed, 7 Dec 2022 16:39:24 +0800 Subject: [PATCH] fix:fix junit error. --- .../scg/PolarisReactiveLoadBalancerClientFilterTest.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/spring-cloud-starter-tencent-polaris-router/src/test/java/com/tencent/cloud/polaris/router/scg/PolarisReactiveLoadBalancerClientFilterTest.java b/spring-cloud-starter-tencent-polaris-router/src/test/java/com/tencent/cloud/polaris/router/scg/PolarisReactiveLoadBalancerClientFilterTest.java index e4c838327..e3ea6ba5b 100644 --- a/spring-cloud-starter-tencent-polaris-router/src/test/java/com/tencent/cloud/polaris/router/scg/PolarisReactiveLoadBalancerClientFilterTest.java +++ b/spring-cloud-starter-tencent-polaris-router/src/test/java/com/tencent/cloud/polaris/router/scg/PolarisReactiveLoadBalancerClientFilterTest.java @@ -48,6 +48,7 @@ import org.mockito.Mockito; import org.mockito.junit.MockitoJUnitRunner; import reactor.core.publisher.Mono; +import org.springframework.cloud.client.loadbalancer.LoadBalancerProperties; import org.springframework.cloud.gateway.config.GatewayLoadBalancerProperties; import org.springframework.cloud.gateway.filter.GatewayFilterChain; import org.springframework.cloud.loadbalancer.core.NoopServiceInstanceListSupplier; @@ -63,6 +64,7 @@ import org.springframework.web.server.ServerWebExchange; import static com.tencent.cloud.common.constant.ContextConstant.UTF_8; import static org.mockito.ArgumentMatchers.anyString; +import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_REQUEST_URL_ATTR; import static org.springframework.cloud.gateway.support.ServerWebExchangeUtils.GATEWAY_SCHEME_PREFIX_ATTR; @@ -98,7 +100,7 @@ public class PolarisReactiveLoadBalancerClientFilterTest { mockedApplicationContextAwareUtils.when(() -> ApplicationContextAwareUtils.getProperties(anyString())) .thenReturn(callerService); - MetadataContext metadataContext = Mockito.mock(MetadataContext.class); + MetadataContext metadataContext = mock(MetadataContext.class); // mock transitive metadata Map transitiveLabels = new HashMap<>(); @@ -176,6 +178,9 @@ public class PolarisReactiveLoadBalancerClientFilterTest { RoundRobinLoadBalancer roundRobinLoadBalancer = new RoundRobinLoadBalancer(new SimpleObjectProvider<>(serviceInstanceListSupplier), calleeService); when(loadBalancerClientFactory.getInstance(calleeService, ReactorServiceInstanceLoadBalancer.class)).thenReturn(roundRobinLoadBalancer); + LoadBalancerProperties loadBalancerProperties = mock(LoadBalancerProperties.class); + when(loadBalancerProperties.getHint()).thenReturn(new HashMap<>()); + when(loadBalancerClientFactory.getProperties(calleeService)).thenReturn(loadBalancerProperties); filter.filter(exchange, chain); }