66 lines
3.3 KiB
Go
66 lines
3.3 KiB
Go
package controller
|
|
|
|
import (
|
|
perfmetrics "github.com/QuantumNous/new-api/pkg/perf_metrics"
|
|
"github.com/QuantumNous/new-api/service"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestCurrentGroupStatus(t *testing.T) {
|
|
const now int64 = 10000
|
|
for _, tc := range []struct {
|
|
name string
|
|
success int64
|
|
probes []service.ProbeHistoryPoint
|
|
disabled bool
|
|
want int
|
|
source string
|
|
at int64
|
|
}{
|
|
{"fresh user overrides older failed probe", now, []service.ProbeHistoryPoint{{At: now - 10, Status: 0}}, false, 1, "requests", now},
|
|
{"newer failed probe cannot erase recent user success", now - 10, []service.ProbeHistoryPoint{{At: now, Status: 0}}, false, 1, "requests", now - 10},
|
|
{"working primary with failing backup stays normal", now - 60, []service.ProbeHistoryPoint{{At: now, Status: 2}}, false, 1, "requests", now - 60},
|
|
{"success remains fresh for full fifteen minutes", now - 900, nil, false, 1, "requests", now - 900},
|
|
{"expired user success uses latest partial probe", now - 901, []service.ProbeHistoryPoint{{At: now - 10, Status: 2}}, false, 2, "detection", now - 10},
|
|
{"healthy probe overrides historical hourly error", 0, []service.ProbeHistoryPoint{{At: now - 100, Status: 1}}, false, 1, "detection", now - 100},
|
|
{"latest probe determines availability", 0, []service.ProbeHistoryPoint{{At: now - 100, Status: 0}, {At: now - 200, Status: 1}}, false, 0, "detection", now - 100},
|
|
{"one missed probe remains fresh", 0, []service.ProbeHistoryPoint{{At: now - 1800, Status: 1}}, false, 1, "detection", now - 1800},
|
|
{"stale probe is unknown", 0, []service.ProbeHistoryPoint{{At: now - 1801, Status: 0}}, false, -1, "detection", 0},
|
|
{"quality failures alone do not imply an outage", 0, nil, false, -1, "detection", 0},
|
|
{"disabled remains disabled", now, []service.ProbeHistoryPoint{{At: now, Status: 1}}, true, 0, "disabled", 0},
|
|
{"future evidence is ignored", now + 1, []service.ProbeHistoryPoint{{At: now + 1, Status: 1}}, false, -1, "detection", 0},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
m := Monitor{Status: 0, StatusSource: "requests", StatusCheckedAt: now - 100, ChannelDisabled: tc.disabled, ProbeHistory: tc.probes, Uptime: 0.58, Fingerprint: &service.AstraSamplingSummary{Errors: 25}, Benchmark: &service.BenchmarkSummary{Failed: 33}}
|
|
applyCurrentStatus(&m, perfmetrics.UserRequestResult{At: tc.success * 1000, Success: true}, now)
|
|
assert.Equal(t, tc.want, m.Status)
|
|
assert.Equal(t, tc.source, m.StatusSource)
|
|
assert.Equal(t, tc.at, m.StatusCheckedAt)
|
|
assert.Equal(t, 25, m.Fingerprint.Errors)
|
|
assert.Equal(t, 33, m.Benchmark.Failed)
|
|
assert.Equal(t, 0.58, m.Uptime)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFailedUserRequestChangesGroupStatus(t *testing.T) {
|
|
const now int64 = 10000
|
|
for _, tc := range []struct {
|
|
name string
|
|
at int64
|
|
want int
|
|
source string
|
|
}{
|
|
{"user failure overrides a newer healthy probe", now - 10, 0, "requests"},
|
|
{"user failure expires after fifteen minutes", now - 901, 1, "detection"},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
m := Monitor{Status: 1, ProbeHistory: []service.ProbeHistoryPoint{{At: now, Status: 1}}}
|
|
applyCurrentStatus(&m, perfmetrics.UserRequestResult{At: tc.at * 1000, Success: false}, now)
|
|
assert.Equal(t, tc.want, m.Status)
|
|
assert.Equal(t, tc.source, m.StatusSource)
|
|
})
|
|
}
|
|
}
|