fix flaky ut in TestCause contexthelper_test

This commit is contained in:
Paco Xu 2024-02-21 18:58:01 +08:00
parent 11785bb815
commit f6682370b1

View File

@ -47,13 +47,13 @@ func TestCause(t *testing.T) {
}{
"nothing": {
parentCtx: context.Background(),
timeout: 5 * time.Millisecond,
timeout: 10 * time.Millisecond,
sleep: time.Millisecond,
},
"timeout": {
parentCtx: context.Background(),
timeout: time.Millisecond,
sleep: 5 * time.Millisecond,
sleep: 10 * time.Millisecond,
expectErr: context.Canceled,
expectCause: canceledError(timeoutCause),
},
@ -64,7 +64,7 @@ func TestCause(t *testing.T) {
return ctx
}(),
timeout: time.Millisecond,
sleep: 5 * time.Millisecond,
sleep: 10 * time.Millisecond,
expectErr: context.Canceled,
expectCause: context.Canceled,
},
@ -112,7 +112,7 @@ func TestCause(t *testing.T) {
if tt.expectDeadline != 0 {
actualDeadline, ok := ctx.Deadline()
if assert.True(t, ok, "should have had a deadline") {
assert.InDelta(t, time.Until(actualDeadline), tt.expectDeadline, float64(time.Second), "remaining time till Deadline()")
assert.InDelta(t, time.Until(actualDeadline), tt.expectDeadline, float64(5*time.Second), "remaining time till Deadline()")
}
}
time.Sleep(tt.sleep)