From d1dead500c7d3a381dab05d4d6c3061b6404b6a2 Mon Sep 17 00:00:00 2001
From: sunby <bingyi.sun@zilliz.com>
Date: Mon, 27 Sep 2021 10:46:08 +0800
Subject: [PATCH] Check error in allocator test (#8621)

Signed-off-by: sunby <bingyi.sun@zilliz.com>
---
 internal/datacoord/allocator_test.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/internal/datacoord/allocator_test.go b/internal/datacoord/allocator_test.go
index 46d9f40369..4e72e2b167 100644
--- a/internal/datacoord/allocator_test.go
+++ b/internal/datacoord/allocator_test.go
@@ -36,9 +36,10 @@ func TestAllocator_Basic(t *testing.T) {
 	t.Run("Test Unhealthy Root", func(t *testing.T) {
 		ms := newMockRootCoordService()
 		allocator := newRootCoordAllocator(ms)
-		ms.Stop()
+		err := ms.Stop()
+		assert.Nil(t, err)
 
-		_, err := allocator.allocTimestamp(ctx)
+		_, err = allocator.allocTimestamp(ctx)
 		assert.Error(t, err)
 		_, err = allocator.allocID(ctx)
 		assert.Error(t, err)