mirror of
https://github.com/tailscale/tailscale.git
synced 2024-12-03 06:56:41 +00:00
d0906cda97
Exposes some internal state of the sockstats package via the C2N and PeerAPI endpoints, so that it can be used for debugging. For now this includes the estimated radio on percentage and a second-by-second view of the times the radio was active. Also fixes another off-by-one error in the radio on percentage that was leading to >100% values (if n seconds have passed since we started to monitor, there may be n + 1 possible seconds where the radio could have been on). Updates tailscale/corp#9230 Signed-off-by: Mihai Parparita <mihai@tailscale.com>
91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build tailscale_go && (darwin || ios || android || ts_enable_sockstats)
|
|
|
|
package sockstats
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type testTime struct {
|
|
time.Time
|
|
}
|
|
|
|
func (t *testTime) now() time.Time {
|
|
return t.Time
|
|
}
|
|
|
|
func (t *testTime) Add(d time.Duration) {
|
|
t.Time = t.Time.Add(d)
|
|
}
|
|
|
|
func TestRadioMonitor(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
activity func(*testTime, *radioMonitor)
|
|
want int64
|
|
}{
|
|
{
|
|
"no activity",
|
|
func(_ *testTime, _ *radioMonitor) {},
|
|
0,
|
|
},
|
|
{
|
|
"active, 10 sec idle",
|
|
func(tt *testTime, rm *radioMonitor) {
|
|
rm.active()
|
|
tt.Add(9 * time.Second)
|
|
},
|
|
50, // radio on 5 seconds of 10 seconds
|
|
},
|
|
{
|
|
"active, spanning two seconds",
|
|
func(tt *testTime, rm *radioMonitor) {
|
|
rm.active()
|
|
tt.Add(1100 * time.Millisecond)
|
|
rm.active()
|
|
},
|
|
100, // radio on for 2 seconds
|
|
},
|
|
{
|
|
"400 iterations: 2 sec active, 1 min idle",
|
|
func(tt *testTime, rm *radioMonitor) {
|
|
// 400 iterations to ensure values loop back around rm.usage array
|
|
for i := 0; i < 400; i++ {
|
|
rm.active()
|
|
tt.Add(1 * time.Second)
|
|
rm.active()
|
|
tt.Add(59 * time.Second)
|
|
}
|
|
},
|
|
10, // radio on 6 seconds of every minute
|
|
},
|
|
{
|
|
"activity at end of time window",
|
|
func(tt *testTime, rm *radioMonitor) {
|
|
tt.Add(1 * time.Second)
|
|
rm.active()
|
|
},
|
|
50,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
tm := &testTime{time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC)}
|
|
rm := &radioMonitor{
|
|
startTime: tm.Time.Unix(),
|
|
now: tm.now,
|
|
}
|
|
tt.activity(tm, rm)
|
|
got := rm.radioHighPercent()
|
|
if got != tt.want {
|
|
t.Errorf("got radioOnPercent %d, want %d", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|