Fix querycoord ut bug (#15752)

Signed-off-by: sunby <bingyi.sun@zilliz.com>

Co-authored-by: sunby <bingyi.sun@zilliz.com>
This commit is contained in:
Bingyi Sun 2022-02-28 11:27:54 +08:00 committed by GitHub
parent 0cfb6a85be
commit b41992eda1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -413,7 +413,7 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
queryCoord, err := startQueryCoord(ctx)
assert.Nil(t, err)
_, err = startQueryNodeServer(ctx)
queryNode, err := startQueryNodeServer(ctx)
assert.Nil(t, err)
taskIDAllocator := queryCoord.scheduler.taskIDAllocator
@ -423,6 +423,9 @@ func TestGrpcTaskEnqueueFail(t *testing.T) {
queryCoord.scheduler.taskIDAllocator = failedAllocator
waitQueryNodeOnline(queryCoord.cluster, queryNode.queryNodeID)
assert.NotEmpty(t, queryCoord.cluster.onlineNodeIDs())
t.Run("Test LoadPartition", func(t *testing.T) {
status, err := queryCoord.LoadPartitions(ctx, &querypb.LoadPartitionsRequest{
Base: &commonpb.MsgBase{