diff --git a/tests/go_client/testcases/insert_test.go b/tests/go_client/testcases/insert_test.go index e1dfdfe2f0..0afb093437 100644 --- a/tests/go_client/testcases/insert_test.go +++ b/tests/go_client/testcases/insert_test.go @@ -3,6 +3,7 @@ package testcases import ( "math" "strconv" + "sync" "testing" "time" @@ -897,8 +898,22 @@ func TestFlushRate(t *testing.T) { _, err := mc.Insert(ctx, insertOpt) common.CheckErr(t, err, true) - _, err = mc.Flush(ctx, client.NewFlushOption(schema.CollectionName)) - common.CheckErr(t, err, true) - _, err = mc.Flush(ctx, client.NewFlushOption(schema.CollectionName)) - common.CheckErr(t, err, false, "request is rejected by grpc RateLimiter middleware, please retry later: rate limit exceeded[rate=0.1]") + cnt := 10 + errs := make([]error, cnt) + wg := &sync.WaitGroup{} + wg.Add(cnt) + for i := 0; i < cnt; i++ { + go func(i int) { + defer wg.Done() + _, err := mc.Flush(ctx, client.NewFlushOption(schema.CollectionName)) + errs[i] = err + }(i) + } + wg.Wait() + + for _, err := range errs { + if err != nil { + common.CheckErr(t, err, false, "request is rejected by grpc RateLimiter middleware, please retry later: rate limit exceeded") + } + } }