shanshanzhong147
6157b2c571
* 修复(#23): 修复用户订阅列表未回显用户级限速 Co-authored-by: multica-agent <github@multica.ai> * 修复(#23): 移除 PR 草稿文件 Co-authored-by: multica-agent <github@multica.ai> --------- Co-authored-by: multica-agent <github@multica.ai>
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package user
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
modeluser "github.com/perfect-panel/server/internal/model/user"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
)
|
|
|
|
func TestBuildUserSubscribeListItem(t *testing.T) {
|
|
trafficLimitJSON := `[{"stat_type":"day","stat_value":1,"traffic_usage":10,"speed_limit":5}]`
|
|
now := time.Unix(1718000000, 0)
|
|
|
|
tests := []struct {
|
|
name string
|
|
item *modeluser.SubscribeDetails
|
|
wantTrafficLimit []types.TrafficLimit
|
|
}{
|
|
{
|
|
name: "non-empty user traffic limit",
|
|
item: &modeluser.SubscribeDetails{
|
|
Id: 1,
|
|
UserId: 2,
|
|
NodeGroupId: 3,
|
|
Token: "token-12345678",
|
|
TrafficLimit: &trafficLimitJSON,
|
|
StartTime: now,
|
|
ExpireTime: now,
|
|
},
|
|
wantTrafficLimit: []types.TrafficLimit{
|
|
{
|
|
StatType: "day",
|
|
StatValue: 1,
|
|
TrafficUsage: 10,
|
|
SpeedLimit: 5,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "empty user traffic limit",
|
|
item: &modeluser.SubscribeDetails{
|
|
Id: 4,
|
|
UserId: 5,
|
|
NodeGroupId: 6,
|
|
Token: "token-empty",
|
|
StartTime: now,
|
|
ExpireTime: now,
|
|
},
|
|
wantTrafficLimit: nil,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
got := buildUserSubscribeListItem(tc.item, "group-a")
|
|
if got.NodeGroupName != "group-a" {
|
|
t.Fatalf("NodeGroupName = %q, want %q", got.NodeGroupName, "group-a")
|
|
}
|
|
if len(got.Short) != 8 {
|
|
t.Fatalf("Short length = %d, want 8", len(got.Short))
|
|
}
|
|
if len(got.TrafficLimit) != len(tc.wantTrafficLimit) {
|
|
t.Fatalf("TrafficLimit length = %d, want %d", len(got.TrafficLimit), len(tc.wantTrafficLimit))
|
|
}
|
|
for i := range tc.wantTrafficLimit {
|
|
if got.TrafficLimit[i] != tc.wantTrafficLimit[i] {
|
|
t.Fatalf("TrafficLimit[%d] = %+v, want %+v", i, got.TrafficLimit[i], tc.wantTrafficLimit[i])
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|