forked from quic-go/quic-go
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Timer", func() {
|
|
const d = 10 * time.Millisecond
|
|
|
|
It("works correctly for newly created and reset timers", func() {
|
|
t := NewTimer()
|
|
Consistently(t.Chan()).ShouldNot(Receive())
|
|
|
|
deadline := time.Now().Add(d)
|
|
t.Reset(deadline)
|
|
Expect(t.Deadline()).To(Equal(deadline))
|
|
Eventually(t.Chan()).Should(Receive())
|
|
|
|
t.SetRead()
|
|
t.Reset(time.Now().Add(d))
|
|
Eventually(t.Chan()).Should(Receive())
|
|
})
|
|
|
|
It("works multiple times with and without reading", func() {
|
|
t := NewTimer()
|
|
for i := 0; i < 10; i++ {
|
|
t.Reset(time.Now().Add(d))
|
|
if i%2 == 0 {
|
|
// Read on even iterations
|
|
Eventually(t.Chan()).Should(Receive())
|
|
t.SetRead()
|
|
} else {
|
|
// Don't read on odd iterations
|
|
time.Sleep(d * 2)
|
|
}
|
|
}
|
|
// Ensure the last timer fires if it wasn't read
|
|
Eventually(t.Chan()).Should(Receive())
|
|
})
|
|
|
|
It("works when resetting without expiration", func() {
|
|
t := NewTimer()
|
|
for i := 0; i < 10; i++ {
|
|
t.Reset(time.Now().Add(time.Hour))
|
|
}
|
|
t.Reset(time.Now().Add(d))
|
|
Eventually(t.Chan()).Should(Receive())
|
|
})
|
|
|
|
It("immediately fires the timer, if the deadlines has already passed", func() {
|
|
t := NewTimer()
|
|
t.Reset(time.Now().Add(-time.Second))
|
|
Eventually(t.Chan()).Should(Receive())
|
|
})
|
|
|
|
It("doesn't set a timer if the deadline is the zero value", func() {
|
|
t := NewTimer()
|
|
t.Reset(time.Time{})
|
|
Consistently(t.Chan()).ShouldNot(Receive())
|
|
})
|
|
|
|
It("fires the timer twice, if reset to the same deadline", func() {
|
|
deadline := time.Now().Add(-time.Millisecond)
|
|
t := NewTimer()
|
|
t.Reset(deadline)
|
|
Eventually(t.Chan()).Should(Receive())
|
|
t.SetRead()
|
|
t.Reset(deadline)
|
|
Eventually(t.Chan()).Should(Receive())
|
|
})
|
|
|
|
It("only fires the timer once, if it is reset to the same deadline, but not read in between", func() {
|
|
deadline := time.Now().Add(-time.Millisecond)
|
|
t := NewTimer()
|
|
t.Reset(deadline)
|
|
Eventually(t.Chan()).Should(Receive())
|
|
Consistently(t.Chan()).ShouldNot(Receive())
|
|
})
|
|
|
|
It("stops", func() {
|
|
t := NewTimer()
|
|
t.Reset(time.Now().Add(50 * time.Millisecond))
|
|
t.Stop()
|
|
Consistently(t.Chan()).ShouldNot(Receive())
|
|
})
|
|
})
|