[doc] 测试用例调整

2023-05-30 17:54:30 +08:00 · 2023-05-30 17:54:30 +08:00 · d8f58df39a
commit d8f58df39a
parent 050c77a5f7
1 changed files with 6 additions and 1 deletions
--- a/grpcall_test.go
+++ b/grpcall_test.go
@ -92,6 +92,9 @@ func TestInvokeBidiStream(t *testing.T) {
 		flag := make(chan bool)

 		go func() {
+			timer := time.NewTimer(time.Second)
+			defer timer.Stop()
+
 			for {
 				select {
 				case msg := <-recv:
@ -102,9 +105,11 @@ func TestInvokeBidiStream(t *testing.T) {
 					}
 					flag <- true
 					return
-				case <-time.After(time.Second):
+				case <-timer.C:
 					resp.Cancel()
+					timer.Reset(time.Second)
 				}
+
 			}
 		}()