30 lines
761 B
Go
30 lines
761 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestConfigureWSHeartbeat(t *testing.T) {
|
|
originalPing, originalPong := pingPeriod, pongWait
|
|
t.Cleanup(func() {
|
|
pingPeriod, pongWait = originalPing, originalPong
|
|
})
|
|
|
|
if err := configureWSHeartbeat(20, 120); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if pingPeriod != 20*time.Second || pongWait != 120*time.Second {
|
|
t.Fatalf("heartbeat ping=%s pong=%s", pingPeriod, pongWait)
|
|
}
|
|
if err := configureWSHeartbeat(90, 90); err == nil {
|
|
t.Fatal("expected invalid heartbeat configuration to fail")
|
|
}
|
|
if err := configureWSHeartbeat(0, 0); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if pingPeriod != defaultWSPingPeriod || pongWait != defaultWSPongWait {
|
|
t.Fatalf("default heartbeat ping=%s pong=%s", pingPeriod, pongWait)
|
|
}
|
|
}
|