李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func accWithWindowEnd(id int64, end *time.Time) accountWithLoad {
|
|
return accountWithLoad{
|
|
account: &Account{
|
|
ID: id,
|
|
Schedulable: true,
|
|
Status: StatusActive,
|
|
SessionWindowEnd: end,
|
|
},
|
|
loadInfo: &AccountLoadInfo{AccountID: id},
|
|
}
|
|
}
|
|
|
|
func TestFilterBySoonestReset_PicksSoonestFutureWindow(t *testing.T) {
|
|
now := time.Now()
|
|
soon := now.Add(1 * time.Hour)
|
|
later := now.Add(24 * time.Hour)
|
|
accounts := []accountWithLoad{
|
|
accWithWindowEnd(1, testTimePtr(later)),
|
|
accWithWindowEnd(2, testTimePtr(soon)),
|
|
accWithWindowEnd(3, testTimePtr(later)),
|
|
}
|
|
got := filterBySoonestReset(accounts)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, int64(2), got[0].account.ID, "重置时间最早的账号被选中")
|
|
}
|
|
|
|
func TestFilterBySoonestReset_IgnoresNilAndExpiredWindows(t *testing.T) {
|
|
now := time.Now()
|
|
expired := now.Add(-1 * time.Hour)
|
|
active := now.Add(2 * time.Hour)
|
|
accounts := []accountWithLoad{
|
|
accWithWindowEnd(1, nil), // 无活跃窗口
|
|
accWithWindowEnd(2, testTimePtr(expired)), // 已过期，视为无活跃窗口
|
|
accWithWindowEnd(3, testTimePtr(active)), // 唯一活跃窗口
|
|
}
|
|
got := filterBySoonestReset(accounts)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, int64(3), got[0].account.ID, "仅保留拥有未来重置时间的账号")
|
|
}
|
|
|
|
func TestFilterBySoonestReset_NoActiveWindowReturnsAll(t *testing.T) {
|
|
now := time.Now()
|
|
expired := now.Add(-30 * time.Minute)
|
|
accounts := []accountWithLoad{
|
|
accWithWindowEnd(1, nil),
|
|
accWithWindowEnd(2, testTimePtr(expired)),
|
|
}
|
|
got := filterBySoonestReset(accounts)
|
|
require.Len(t, got, 2, "没有任何账号拥有活跃窗口时，返回原集合不做过滤")
|
|
}
|
|
|
|
func TestFilterBySoonestReset_TiedSoonestKeepsAll(t *testing.T) {
|
|
now := time.Now()
|
|
end := now.Add(90 * time.Minute)
|
|
accounts := []accountWithLoad{
|
|
accWithWindowEnd(1, testTimePtr(end)),
|
|
accWithWindowEnd(2, testTimePtr(end)),
|
|
accWithWindowEnd(3, testTimePtr(now.Add(5*time.Hour))),
|
|
}
|
|
got := filterBySoonestReset(accounts)
|
|
require.Len(t, got, 2, "并列最早重置的账号都保留，交由后续 LRU 决定")
|
|
ids := map[int64]bool{got[0].account.ID: true, got[1].account.ID: true}
|
|
require.True(t, ids[1] && ids[2])
|
|
}
|
|
|
|
func TestFilterBySoonestReset_SingleOrEmptyUnchanged(t *testing.T) {
|
|
require.Empty(t, filterBySoonestReset(nil))
|
|
single := []accountWithLoad{accWithWindowEnd(1, nil)}
|
|
require.Len(t, filterBySoonestReset(single), 1)
|
|
}
|