mirror of
https://github.com/henrygd/beszel.git
synced 2026-04-23 21:21:49 +02:00
90 lines
3.4 KiB
Go
90 lines
3.4 KiB
Go
package agent
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProbeTaskAggregateLockedUsesRawSamplesForShortWindows(t *testing.T) {
|
|
now := time.Date(2026, time.April, 21, 12, 0, 0, 0, time.UTC)
|
|
task := &probeTask{}
|
|
|
|
task.addSampleLocked(probeSample{responseMs: 10, timestamp: now.Add(-90 * time.Second)})
|
|
task.addSampleLocked(probeSample{responseMs: 20, timestamp: now.Add(-30 * time.Second)})
|
|
task.addSampleLocked(probeSample{responseMs: -1, timestamp: now.Add(-10 * time.Second)})
|
|
|
|
agg := task.aggregateLocked(time.Minute, now)
|
|
require.True(t, agg.hasData())
|
|
assert.Equal(t, 2, agg.totalCount)
|
|
assert.Equal(t, 1, agg.successCount)
|
|
assert.Equal(t, 20.0, agg.result()[0])
|
|
assert.Equal(t, 20.0, agg.result()[1])
|
|
assert.Equal(t, 20.0, agg.result()[2])
|
|
assert.Equal(t, 50.0, agg.result()[3])
|
|
}
|
|
|
|
func TestProbeTaskAggregateLockedUsesMinuteBucketsForLongWindows(t *testing.T) {
|
|
now := time.Date(2026, time.April, 21, 12, 0, 30, 0, time.UTC)
|
|
task := &probeTask{}
|
|
|
|
task.addSampleLocked(probeSample{responseMs: 10, timestamp: now.Add(-11 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: 20, timestamp: now.Add(-9 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: 40, timestamp: now.Add(-5 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: -1, timestamp: now.Add(-90 * time.Second)})
|
|
task.addSampleLocked(probeSample{responseMs: 30, timestamp: now.Add(-30 * time.Second)})
|
|
|
|
agg := task.aggregateLocked(10*time.Minute, now)
|
|
require.True(t, agg.hasData())
|
|
assert.Equal(t, 4, agg.totalCount)
|
|
assert.Equal(t, 3, agg.successCount)
|
|
assert.Equal(t, 30.0, agg.result()[0])
|
|
assert.Equal(t, 20.0, agg.result()[1])
|
|
assert.Equal(t, 40.0, agg.result()[2])
|
|
assert.Equal(t, 25.0, agg.result()[3])
|
|
}
|
|
|
|
func TestProbeTaskAddSampleLockedTrimsRawSamplesButKeepsBucketHistory(t *testing.T) {
|
|
now := time.Date(2026, time.April, 21, 12, 0, 0, 0, time.UTC)
|
|
task := &probeTask{}
|
|
|
|
task.addSampleLocked(probeSample{responseMs: 10, timestamp: now.Add(-10 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: 20, timestamp: now})
|
|
|
|
require.Len(t, task.samples, 1)
|
|
assert.Equal(t, 20.0, task.samples[0].responseMs)
|
|
|
|
agg := task.aggregateLocked(10*time.Minute, now)
|
|
require.True(t, agg.hasData())
|
|
assert.Equal(t, 2, agg.totalCount)
|
|
assert.Equal(t, 2, agg.successCount)
|
|
assert.Equal(t, 15.0, agg.result()[0])
|
|
assert.Equal(t, 10.0, agg.result()[1])
|
|
assert.Equal(t, 20.0, agg.result()[2])
|
|
assert.Equal(t, 0.0, agg.result()[3])
|
|
}
|
|
|
|
func TestProbeManagerGetResultsIncludesHourResponseRange(t *testing.T) {
|
|
now := time.Now().UTC()
|
|
task := &probeTask{}
|
|
task.addSampleLocked(probeSample{responseMs: 10, timestamp: now.Add(-30 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: 20, timestamp: now.Add(-9 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: 40, timestamp: now.Add(-5 * time.Minute)})
|
|
task.addSampleLocked(probeSample{responseMs: -1, timestamp: now.Add(-90 * time.Second)})
|
|
task.addSampleLocked(probeSample{responseMs: 30, timestamp: now.Add(-30 * time.Second)})
|
|
|
|
pm := &ProbeManager{probes: map[string]*probeTask{"icmp:example.com": task}}
|
|
|
|
results := pm.GetResults(uint16(time.Minute / time.Millisecond))
|
|
result, ok := results["icmp:example.com"]
|
|
require.True(t, ok)
|
|
require.Len(t, result, 5)
|
|
assert.Equal(t, 30.0, result[0])
|
|
assert.Equal(t, 25.0, result[1])
|
|
assert.Equal(t, 10.0, result[2])
|
|
assert.Equal(t, 40.0, result[3])
|
|
assert.Equal(t, 0.0, result[4])
|
|
}
|