From cad61ea47ef155b06efe233ddd9b228e0bf5d978 Mon Sep 17 00:00:00 2001
From: bigsheeper <yihao.dai@zilliz.com>
Date: Mon, 29 Nov 2021 14:17:17 +0800
Subject: [PATCH] Fix tSafe replica unit test (#12325)
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
---
internal/querynode/tsafe_replica_test.go | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/internal/querynode/tsafe_replica_test.go b/internal/querynode/tsafe_replica_test.go
index c101dc60b2..cb1ada9a88 100644
--- a/internal/querynode/tsafe_replica_test.go
+++ b/internal/querynode/tsafe_replica_test.go
@@ -13,7 +13,6 @@ package querynode
import (
"testing"
- "time"
"github.com/stretchr/testify/assert"
)
@@ -29,7 +28,7 @@ func TestTSafeReplica_valid(t *testing.T) {
timestamp := Timestamp(1000)
err = replica.setTSafe(defaultVChannel, defaultCollectionID, timestamp)
assert.NoError(t, err)
- time.Sleep(20 * time.Millisecond)
+ <-watcher.watcherChan()
resT, err := replica.getTSafe(defaultVChannel)
assert.NoError(t, err)
assert.Equal(t, timestamp, resT)
@@ -49,7 +48,7 @@ func TestTSafeReplica_invalid(t *testing.T) {
timestamp := Timestamp(1000)
err = replica.setTSafe(defaultVChannel, defaultCollectionID, timestamp)
assert.NoError(t, err)
- time.Sleep(20 * time.Millisecond)
+ <-watcher.watcherChan()
resT, err := replica.getTSafe(defaultVChannel)
assert.NoError(t, err)
assert.Equal(t, timestamp, resT)