forked from quic-go/quic-go
utils: use synctest to make the timer tests fully deterministic (#5306)
* utils: use synctest to make the timer tests fully deterministic * Go 1.24 workaround
This commit is contained in:
@@ -4,97 +4,83 @@ import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/quic-go/quic-go/internal/synctest"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
const testDuration = 10 * time.Millisecond
|
||||
func TestTimerResets(t *testing.T) {
|
||||
synctest.Test(t, func(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
|
||||
func TestTimerCreateAndReset(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
default:
|
||||
}
|
||||
|
||||
deadline := time.Now().Add(testDuration)
|
||||
timer.Reset(deadline)
|
||||
require.Equal(t, deadline, timer.Deadline())
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(2 * testDuration):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
|
||||
timer.SetRead()
|
||||
timer.Reset(time.Now().Add(testDuration))
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(2 * testDuration):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimerMultipleResets(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
for i := 0; i < 10; i++ {
|
||||
timer.Reset(time.Now().Add(testDuration))
|
||||
if i%2 == 0 {
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(2 * testDuration):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
timer.SetRead()
|
||||
} else {
|
||||
time.Sleep(testDuration * 2)
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
default:
|
||||
}
|
||||
}
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(2 * testDuration):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
start := time.Now()
|
||||
|
||||
// timer fires immediately for a deadline in the past
|
||||
timer.Reset(time.Now().Add(-time.Second))
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
require.Zero(t, time.Since(start))
|
||||
timer.SetRead()
|
||||
case <-time.After(time.Hour): // this can be replaced with a default once we drop support for Go 1.24
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
|
||||
// timer reset without getting read
|
||||
for range 10 {
|
||||
time.Sleep(time.Second)
|
||||
timer.Reset(time.Now().Add(time.Hour))
|
||||
}
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
require.Equal(t, time.Since(start), time.Hour+10*time.Second)
|
||||
timer.SetRead()
|
||||
case <-time.After(2 * time.Hour):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
|
||||
const d = 10 * time.Minute
|
||||
for i := range 10 {
|
||||
start := time.Now()
|
||||
timer.Reset(time.Now().Add(d))
|
||||
if i%2 == 0 {
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
require.Equal(t, time.Since(start), d)
|
||||
case <-time.After(2 * d):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
timer.SetRead()
|
||||
} else {
|
||||
time.Sleep(2 * d)
|
||||
}
|
||||
}
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
default:
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestTimerResetWithoutExpiration(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
for i := 0; i < 10; i++ {
|
||||
timer.Reset(time.Now().Add(time.Hour))
|
||||
}
|
||||
timer.Reset(time.Now().Add(testDuration))
|
||||
func TestTimerClearDeadline(t *testing.T) {
|
||||
synctest.Test(t, func(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
timer.Reset(time.Time{})
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(2 * testDuration):
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimerPastDeadline(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
timer.Reset(time.Now().Add(-time.Second))
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(testDuration):
|
||||
t.Fatal("timer should have fired immediately")
|
||||
}
|
||||
}
|
||||
|
||||
func TestTimerZeroDeadline(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
timer.Reset(time.Time{})
|
||||
|
||||
// we don't expect the timer to be set for a zero deadline
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
case <-time.After(testDuration):
|
||||
}
|
||||
// we don't expect the timer to be set for a zero deadline
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
case <-time.After(time.Hour):
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestTimerSameDeadline(t *testing.T) {
|
||||
@@ -105,7 +91,7 @@ func TestTimerSameDeadline(t *testing.T) {
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(testDuration):
|
||||
default:
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
|
||||
@@ -114,7 +100,7 @@ func TestTimerSameDeadline(t *testing.T) {
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(testDuration):
|
||||
default:
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
})
|
||||
@@ -126,26 +112,28 @@ func TestTimerSameDeadline(t *testing.T) {
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
case <-time.After(testDuration):
|
||||
default:
|
||||
t.Fatal("timer should have fired")
|
||||
}
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired again")
|
||||
case <-time.After(testDuration):
|
||||
default:
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestTimerStopping(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
timer.Reset(time.Now().Add(testDuration))
|
||||
timer.Stop()
|
||||
func TestTimerStop(t *testing.T) {
|
||||
synctest.Test(t, func(t *testing.T) {
|
||||
timer := NewTimer()
|
||||
timer.Reset(time.Now().Add(time.Second))
|
||||
timer.Stop()
|
||||
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
case <-time.After(2 * testDuration):
|
||||
}
|
||||
select {
|
||||
case <-timer.Chan():
|
||||
t.Fatal("timer should not have fired")
|
||||
case <-time.After(time.Hour):
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user