95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
|
package notify
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetSignalReliabilityStatsDefaults(t *testing.T) {
|
||
|
|
clientStats, err := GetSignalReliabilityStatsClient(NewClient())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("GetSignalReliabilityStatsClient failed: %v", err)
|
||
|
|
}
|
||
|
|
if clientStats != (SignalReliabilityStats{}) {
|
||
|
|
t.Fatalf("client default stats mismatch: %+v", clientStats)
|
||
|
|
}
|
||
|
|
|
||
|
|
serverStats, err := GetSignalReliabilityStatsServer(NewServer())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("GetSignalReliabilityStatsServer failed: %v", err)
|
||
|
|
}
|
||
|
|
if serverStats != (SignalReliabilityStats{}) {
|
||
|
|
t.Fatalf("server default stats mismatch: %+v", serverStats)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetSignalReliabilityStatsRejectsNil(t *testing.T) {
|
||
|
|
if _, err := GetSignalReliabilityStatsClient(nil); !errors.Is(err, errSignalReliabilityStatsClientNil) {
|
||
|
|
t.Fatalf("GetSignalReliabilityStatsClient nil error = %v, want %v", err, errSignalReliabilityStatsClientNil)
|
||
|
|
}
|
||
|
|
if _, err := GetSignalReliabilityStatsServer(nil); !errors.Is(err, errSignalReliabilityStatsServerNil) {
|
||
|
|
t.Fatalf("GetSignalReliabilityStatsServer nil error = %v, want %v", err, errSignalReliabilityStatsServerNil)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSendSignalWithAckTrackedStats(t *testing.T) {
|
||
|
|
pool := newSignalAckPool()
|
||
|
|
state := newSignalReliabilityState()
|
||
|
|
err := sendSignalWithAckTracked(state, "scope", 1001, 100*time.Millisecond, pool, func() error {
|
||
|
|
go func() {
|
||
|
|
time.Sleep(time.Millisecond)
|
||
|
|
pool.deliver("scope", 1001)
|
||
|
|
}()
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("sendSignalWithAckTracked success path failed: %v", err)
|
||
|
|
}
|
||
|
|
stats := state.snapshot()
|
||
|
|
if got, want := stats.AckWaitTotal, uint64(1); got != want {
|
||
|
|
t.Fatalf("AckWaitTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
if got, want := stats.AckDeliverTotal, uint64(1); got != want {
|
||
|
|
t.Fatalf("AckDeliverTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
if got, want := stats.AckTimeoutTotal, uint64(0); got != want {
|
||
|
|
t.Fatalf("AckTimeoutTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
if got, want := stats.AckCanceledTotal, uint64(0); got != want {
|
||
|
|
t.Fatalf("AckCanceledTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSendSignalWithAckTrackedTimeoutAndCanceledStats(t *testing.T) {
|
||
|
|
timeoutPool := newSignalAckPool()
|
||
|
|
timeoutState := newSignalReliabilityState()
|
||
|
|
err := sendSignalWithAckTracked(timeoutState, "scope-timeout", 1002, time.Millisecond, timeoutPool, func() error {
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if !errors.Is(err, errSignalAckTimeout) {
|
||
|
|
t.Fatalf("timeout error = %v, want %v", err, errSignalAckTimeout)
|
||
|
|
}
|
||
|
|
timeoutStats := timeoutState.snapshot()
|
||
|
|
if got, want := timeoutStats.AckTimeoutTotal, uint64(1); got != want {
|
||
|
|
t.Fatalf("AckTimeoutTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
|
||
|
|
cancelPool := newSignalAckPool()
|
||
|
|
cancelState := newSignalReliabilityState()
|
||
|
|
err = sendSignalWithAckTracked(cancelState, "scope-cancel", 1003, 100*time.Millisecond, cancelPool, func() error {
|
||
|
|
go func() {
|
||
|
|
time.Sleep(time.Millisecond)
|
||
|
|
cancelPool.closeScope("scope-cancel")
|
||
|
|
}()
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
if !errors.Is(err, errSignalAckCanceled) {
|
||
|
|
t.Fatalf("cancel error = %v, want %v", err, errSignalAckCanceled)
|
||
|
|
}
|
||
|
|
cancelStats := cancelState.snapshot()
|
||
|
|
if got, want := cancelStats.AckCanceledTotal, uint64(1); got != want {
|
||
|
|
t.Fatalf("AckCanceledTotal mismatch: got %d want %d", got, want)
|
||
|
|
}
|
||
|
|
}
|