From a9a5f76f7483d5f21fecd7dff0fd7865b6bcaca0 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Thu, 13 Apr 2023 20:46:29 +0800
Subject: [PATCH] Fix AddDistribution unit test (#23418)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/querynodev2/delegator/distribution_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/internal/querynodev2/delegator/distribution_test.go b/internal/querynodev2/delegator/distribution_test.go
index cae37f154b..8dae155810 100644
--- a/internal/querynodev2/delegator/distribution_test.go
+++ b/internal/querynodev2/delegator/distribution_test.go
@@ -145,10 +145,12 @@ func (s *DistributionSuite) TestAddDistribution() {
 			s.SetupTest()
 			defer s.TearDownTest()
 			s.dist.AddGrowing(tc.growing...)
+			_, _, version := s.dist.GetCurrent()
 			_, signal := s.dist.AddDistributions(tc.input...)
 			sealed, _ := s.dist.Peek()
 			s.compareSnapshotItems(tc.expected, sealed)
 			s.Equal(tc.expectedSignalClosed, s.isClosedCh(signal))
+			s.dist.FinishUsage(version)
 		})
 	}
 }