mirror of
https://github.com/MeshCore-Beacon/beacon-server.git
synced 2026-10-16 18:10:00 +00:00
n30nex
bcdea76c26
* feat(stats): filter scope counts by IATA and region * perf(stats): aggregate region observations before counting scopes
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
// Copyright 2026 Beacon Contributors
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
package cache
|
|
|
|
import (
|
|
"context"
|
|
"slices"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/MeshCore-Beacon/beacon-server/internal/api"
|
|
)
|
|
|
|
type scopeStatsReader struct {
|
|
api.Reader
|
|
calls int64
|
|
}
|
|
|
|
func (r *scopeStatsReader) GetScopeStats(context.Context, []string) ([]api.ScopeStats, error) {
|
|
r.calls++
|
|
return []api.ScopeStats{{Name: "#test", PacketCount: r.calls}}, nil
|
|
}
|
|
|
|
func TestScopeStatsCacheSeparatesIATAs(t *testing.T) {
|
|
c, _ := newTestClient(t)
|
|
inner := &scopeStatsReader{}
|
|
reader := NewCachedReader(inner, c, CacheTTLs{Reference: time.Minute})
|
|
for _, tc := range []struct {
|
|
iatas []string
|
|
want int64
|
|
}{
|
|
{nil, 1},
|
|
{[]string{"YVR"}, 2},
|
|
{[]string{"YYJ"}, 3},
|
|
{[]string{"YYJ", "YVR"}, 4},
|
|
{[]string{"YVR", "YYJ", "YVR"}, 4},
|
|
{[]string{"YVR"}, 2},
|
|
{[]string{}, 1},
|
|
{[]string{"ZZZ"}, 5},
|
|
{[]string{""}, 6},
|
|
{nil, 1},
|
|
} {
|
|
before := slices.Clone(tc.iatas)
|
|
rows, err := reader.GetScopeStats(context.Background(), tc.iatas)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rows) != 1 || rows[0].PacketCount != tc.want {
|
|
t.Fatalf("IATAs %v: got %+v, want cached count %d", tc.iatas, rows, tc.want)
|
|
}
|
|
if !slices.Equal(before, tc.iatas) {
|
|
t.Fatalf("caller IATAs mutated: %v -> %v", before, tc.iatas)
|
|
}
|
|
}
|
|
if inner.calls != 6 {
|
|
t.Fatalf("underlying calls = %d, want 6", inner.calls)
|
|
}
|
|
}
|