113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
package consul
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSessionTimers(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("too slow for testing.Short")
|
|
}
|
|
|
|
m := NewSessionTimers()
|
|
ch := make(chan int)
|
|
newTm := func(d time.Duration) *time.Timer {
|
|
return time.AfterFunc(d, func() { ch <- 1 })
|
|
}
|
|
|
|
waitForTimer := func() {
|
|
select {
|
|
case <-ch:
|
|
return
|
|
case <-time.After(100 * time.Millisecond):
|
|
t.Fatal("timer did not fire")
|
|
}
|
|
}
|
|
|
|
// check that non-existent id returns nil
|
|
if got, want := m.Get("foo"), (*time.Timer)(nil); got != want {
|
|
t.Fatalf("got %v want %v", got, want)
|
|
}
|
|
|
|
// add a timer and look it up and delete via Set(id, nil)
|
|
tm := newTm(time.Millisecond)
|
|
m.Set("foo", tm)
|
|
if got, want := m.Len(), 1; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
if got, want := m.Get("foo"), tm; got != want {
|
|
t.Fatalf("got %v want %v", got, want)
|
|
}
|
|
m.Set("foo", nil)
|
|
if got, want := m.Get("foo"), (*time.Timer)(nil); got != want {
|
|
t.Fatalf("got %v want %v", got, want)
|
|
}
|
|
waitForTimer()
|
|
|
|
// same thing via Del(id)
|
|
tm = newTm(time.Millisecond)
|
|
m.Set("foo", tm)
|
|
if got, want := m.Get("foo"), tm; got != want {
|
|
t.Fatalf("got %v want %v", got, want)
|
|
}
|
|
m.Del("foo")
|
|
if got, want := m.Len(), 0; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
waitForTimer()
|
|
|
|
// create timer via ResetOrCreate
|
|
m.ResetOrCreate("foo", time.Millisecond, func() { ch <- 1 })
|
|
if got, want := m.Len(), 1; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
waitForTimer()
|
|
|
|
// timer is still there
|
|
if got, want := m.Len(), 1; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
|
|
// reset the timer and check that it fires again
|
|
m.ResetOrCreate("foo", time.Millisecond, nil)
|
|
waitForTimer()
|
|
|
|
// reset the timer with a long ttl and then stop it
|
|
m.ResetOrCreate("foo", 20*time.Millisecond, func() { ch <- 1 })
|
|
m.Stop("foo")
|
|
select {
|
|
case <-ch:
|
|
t.Fatal("timer fired although it shouldn't")
|
|
case <-time.After(100 * time.Millisecond):
|
|
// want
|
|
}
|
|
|
|
// stopping a stopped timer should not break
|
|
m.Stop("foo")
|
|
|
|
// stop should also remove the timer
|
|
if got, want := m.Len(), 0; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
|
|
// create two timers and stop and then stop all
|
|
m.ResetOrCreate("foo1", 20*time.Millisecond, func() { ch <- 1 })
|
|
m.ResetOrCreate("foo2", 30*time.Millisecond, func() { ch <- 2 })
|
|
m.StopAll()
|
|
select {
|
|
case x := <-ch:
|
|
t.Fatalf("timer %d fired although it shouldn't", x)
|
|
case <-time.After(100 * time.Millisecond):
|
|
// want
|
|
}
|
|
|
|
// stopall should remove all timers
|
|
if got, want := m.Len(), 0; got != want {
|
|
t.Fatalf("got len %d want %d", got, want)
|
|
}
|
|
}
|