milvus/internal/querynode/tsafe_test.go
bigsheeper b98b226d8c Refactor query node and implement channelRemove, channelAdd
Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
2021-01-15 15:28:54 +08:00

22 lines
358 B
Go

package querynodeimp
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestTSafe_GetAndSet(t *testing.T) {
tSafe := newTSafe()
watcher := newTSafeWatcher()
tSafe.registerTSafeWatcher(watcher)
go func() {
watcher.hasUpdate()
timestamp := tSafe.get()
assert.Equal(t, timestamp, Timestamp(1000))
}()
tSafe.set(Timestamp(1000))
}