Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/gateway_soonest_reset_test.go

@@ -0,0 +1,81 @@
+//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)
+}