diff --git a/clientv3/integration/watch_test.go b/clientv3/integration/watch_test.go index 27e890ed8..efd2ef65c 100644 --- a/clientv3/integration/watch_test.go +++ b/clientv3/integration/watch_test.go @@ -487,7 +487,7 @@ func testWatchWithProgressNotify(t *testing.T, watchOnPut bool) { } else if len(resp.Events) != 0 { // wait for notification otherwise t.Fatalf("expected no events, but got %+v", resp.Events) } - case <-time.After(2 * pi): + case <-time.After(time.Duration(1.5 * float64(pi))): t.Fatalf("watch response expected in %v, but timed out", pi) } } diff --git a/integration/v3_watch_test.go b/integration/v3_watch_test.go index a3181506c..5c4250dcb 100644 --- a/integration/v3_watch_test.go +++ b/integration/v3_watch_test.go @@ -971,7 +971,7 @@ func TestWatchWithProgressNotify(t *testing.T) { } // no more notification - rok, resp := waitResponse(wStream, testInterval+time.Second) + rok, resp := waitResponse(wStream, time.Second) if !rok { t.Errorf("unexpected pb.WatchResponse is received %+v", resp) }