From 0eea850c3e5af263fe2a48f3c7ffad0e37e24e10 Mon Sep 17 00:00:00 2001
From: lepdou <ledouzhang@tencent.com>
Date: Mon, 7 Nov 2022 15:05:28 +0800
Subject: [PATCH] fix weight lb ut (#705)

---
 .../cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java b/spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java
index 6241318fb..6e5514c50 100644
--- a/spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java
+++ b/spring-cloud-tencent-polaris-loadbalancer/src/test/java/com/tencent/cloud/polaris/loadbalancer/PolarisLoadBalancerTest.java
@@ -80,6 +80,7 @@ public class PolarisLoadBalancerTest {
 		//mock routerAPI for rule
 		when(routerAPI.processLoadBalance(any())).thenReturn(assembleProcessLoadBalanceResp());
 		PolarisWeightedRule rule = new PolarisWeightedRule(routerAPI);
+		rule.initWithNiwsConfig(new DefaultClientConfigImpl());
 
 		// clientConfig
 		IClientConfig config = new DefaultClientConfigImpl();
@@ -111,6 +112,7 @@ public class PolarisLoadBalancerTest {
 		//mock routerAPI for rule
 		when(routerAPI.processLoadBalance(any())).thenReturn(assembleProcessLoadBalanceResp());
 		PolarisWeightedRule rule = new PolarisWeightedRule(routerAPI);
+		rule.initWithNiwsConfig(new DefaultClientConfigImpl());
 
 		// clientConfig
 		IClientConfig config = new DefaultClientConfigImpl();