mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-09-24 16:05:44 +08:00
DaydreamCoding
af19d44327
- schema/迁移: 代理有效期、提醒天数、失败回退配置 + 账号 fallback 来源字段 - service/repo/DTO/handler: CRUD 透传新字段 + 校验 - fallback 目标解析纯函数(链式解析 + 环检测 + 兜底) - SweepExpiredProxies 到期改投账号 + outbox 失效 - ProxyExpiryService 后台到期扫描任务 + wire 注册 - 账号侧手动回切原代理 + fallback 来源徽章/按钮 - 前端: 创建/编辑表单、列表到期徽章、类型/API/i18n - ops 告警: proxy_expired_count / proxy_expiring_soon_count 指标 - 导入导出携带有效期/回退字段(备用按 name 映射) - 补全测试 stub + 集成测试 + review 问题修复
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func mkProxy(id int64, mode string, backup *int64, expiresInDays *int, now time.Time) Proxy {
|
|
p := Proxy{ID: id, FallbackMode: mode, BackupProxyID: backup}
|
|
if expiresInDays != nil {
|
|
t := now.AddDate(0, 0, *expiresInDays)
|
|
p.ExpiresAt = &t
|
|
}
|
|
return p
|
|
}
|
|
func i64(v int64) *int64 { return &v }
|
|
func di(v int) *int { return &v }
|
|
|
|
func TestResolveFallbackTarget(t *testing.T) {
|
|
now := time.Now()
|
|
t.Run("none keeps original", func(t *testing.T) {
|
|
a := mkProxy(1, FallbackModeNone, nil, di(-1), now)
|
|
by := map[int64]Proxy{1: a}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.False(t, change)
|
|
require.Nil(t, target)
|
|
})
|
|
t.Run("direct -> nil target, change", func(t *testing.T) {
|
|
a := mkProxy(1, FallbackModeDirect, nil, di(-1), now)
|
|
by := map[int64]Proxy{1: a}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.True(t, change)
|
|
require.Nil(t, target)
|
|
})
|
|
t.Run("proxy -> healthy backup", func(t *testing.T) {
|
|
b := mkProxy(2, FallbackModeNone, nil, di(30), now)
|
|
a := mkProxy(1, FallbackModeProxy, i64(2), di(-1), now)
|
|
by := map[int64]Proxy{1: a, 2: b}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.True(t, change)
|
|
require.NotNil(t, target)
|
|
require.Equal(t, int64(2), *target)
|
|
})
|
|
t.Run("chain A->B(expired)->C(healthy)", func(t *testing.T) {
|
|
c := mkProxy(3, FallbackModeNone, nil, di(30), now)
|
|
b := mkProxy(2, FallbackModeProxy, i64(3), di(-1), now)
|
|
a := mkProxy(1, FallbackModeProxy, i64(2), di(-1), now)
|
|
by := map[int64]Proxy{1: a, 2: b, 3: c}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.True(t, change)
|
|
require.Equal(t, int64(3), *target)
|
|
})
|
|
t.Run("cycle A->B->A keeps original", func(t *testing.T) {
|
|
b := mkProxy(2, FallbackModeProxy, i64(1), di(-1), now)
|
|
a := mkProxy(1, FallbackModeProxy, i64(2), di(-1), now)
|
|
by := map[int64]Proxy{1: a, 2: b}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.False(t, change)
|
|
require.Nil(t, target)
|
|
})
|
|
t.Run("chain tail direct fallback", func(t *testing.T) {
|
|
b := mkProxy(2, FallbackModeDirect, nil, di(-1), now)
|
|
a := mkProxy(1, FallbackModeProxy, i64(2), di(-1), now)
|
|
by := map[int64]Proxy{1: a, 2: b}
|
|
target, change := ResolveProxyFallbackTarget(a, by, now)
|
|
require.True(t, change)
|
|
require.Nil(t, target)
|
|
})
|
|
}
|