fix ut

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
weiliu1031 · Mar 8, 2024 · 40a1069 · 40a1069
1 parent 716af6c
commit 40a1069
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/internal/querycoordv2/balance/rowcount_based_balancer_test.go b/internal/querycoordv2/balance/rowcount_based_balancer_test.go
@@ -986,6 +986,7 @@ func (suite *RowCountBasedBalancerTestSuite) TestDisableBalanceChannel() {
 			}
 
 			Params.Save(Params.QueryCoordCfg.AutoBalanceChannel.Key, fmt.Sprint(c.enableBalanceChannel))
+			defer Params.Reset(Params.QueryCoordCfg.AutoBalanceChannel.Key)
 			segmentPlans, channelPlans := suite.getCollectionBalancePlans(balancer, 1)
 			if !c.multiple {
 				suite.ElementsMatch(c.expectChannelPlans, channelPlans)