From 40a1069de06383d0c6d4d70b5b895a4781f3bd53 Mon Sep 17 00:00:00 2001 From: Wei Liu Date: Thu, 7 Mar 2024 21:05:30 +0800 Subject: [PATCH] fix ut Signed-off-by: Wei Liu --- internal/querycoordv2/balance/rowcount_based_balancer_test.go | 1 + 1 file changed, 1 insertion(+) diff --git a/internal/querycoordv2/balance/rowcount_based_balancer_test.go b/internal/querycoordv2/balance/rowcount_based_balancer_test.go index 680c229933358..9f15de00c43a7 100644 --- 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)