mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-26 11:35:35 +00:00
782ccb5655
We use it to gate code that depends on custom Go toolchain, but it's currently only passed in the corp runners. Add a set on OSS so that we can catch regressions earlier. To specifically test sockstats this required adding a build tag to explicitly enable them -- they're normally on for iOS, macOS and Android only, and we don't run tests on those platforms normally. Signed-off-by: Mihai Parparita <mihai@tailscale.com>
74 lines
1.4 KiB
Go
74 lines
1.4 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(10 * time.Second)
|
|
},
|
|
50, // radio on 5 seconds of every 10 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(1 * time.Minute)
|
|
}
|
|
},
|
|
10, // radio on 6 seconds of every minute
|
|
},
|
|
}
|
|
|
|
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)
|
|
}
|
|
})
|
|
}
|
|
}
|