351 lines
7.6 KiB
Go
351 lines
7.6 KiB
Go
package service
|
||
|
||
import (
|
||
"net"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// TestYggdrasilService_Constants 测试Yggdrasil服务常量
|
||
func TestYggdrasilService_Constants(t *testing.T) {
|
||
if SessionKeyPrefix != "Join_" {
|
||
t.Errorf("SessionKeyPrefix = %s, want 'Join_'", SessionKeyPrefix)
|
||
}
|
||
|
||
if SessionTTL != 15*time.Minute {
|
||
t.Errorf("SessionTTL = %v, want 15 minutes", SessionTTL)
|
||
}
|
||
}
|
||
|
||
// TestSessionData_Structure 测试SessionData结构
|
||
func TestSessionData_Structure(t *testing.T) {
|
||
data := SessionData{
|
||
AccessToken: "test-token",
|
||
UserName: "TestUser",
|
||
SelectedProfile: "test-profile-uuid",
|
||
IP: "127.0.0.1",
|
||
}
|
||
|
||
if data.AccessToken == "" {
|
||
t.Error("AccessToken should not be empty")
|
||
}
|
||
|
||
if data.UserName == "" {
|
||
t.Error("UserName should not be empty")
|
||
}
|
||
|
||
if data.SelectedProfile == "" {
|
||
t.Error("SelectedProfile should not be empty")
|
||
}
|
||
}
|
||
|
||
// TestJoinServer_InputValidation 测试JoinServer输入验证逻辑
|
||
func TestJoinServer_InputValidation(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
serverId string
|
||
accessToken string
|
||
selectedProfile string
|
||
wantErr bool
|
||
errContains string
|
||
}{
|
||
{
|
||
name: "所有参数有效",
|
||
serverId: "test-server-123",
|
||
accessToken: "test-token",
|
||
selectedProfile: "test-profile",
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "serverId为空",
|
||
serverId: "",
|
||
accessToken: "test-token",
|
||
selectedProfile: "test-profile",
|
||
wantErr: true,
|
||
errContains: "参数不能为空",
|
||
},
|
||
{
|
||
name: "accessToken为空",
|
||
serverId: "test-server",
|
||
accessToken: "",
|
||
selectedProfile: "test-profile",
|
||
wantErr: true,
|
||
errContains: "参数不能为空",
|
||
},
|
||
{
|
||
name: "selectedProfile为空",
|
||
serverId: "test-server",
|
||
accessToken: "test-token",
|
||
selectedProfile: "",
|
||
wantErr: true,
|
||
errContains: "参数不能为空",
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
hasError := tt.serverId == "" || tt.accessToken == "" || tt.selectedProfile == ""
|
||
if hasError != tt.wantErr {
|
||
t.Errorf("Input validation failed: got %v, want %v", hasError, tt.wantErr)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestJoinServer_ServerIDValidation 测试服务器ID格式验证
|
||
func TestJoinServer_ServerIDValidation(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
serverId string
|
||
wantValid bool
|
||
}{
|
||
{
|
||
name: "有效的serverId",
|
||
serverId: "test-server-123",
|
||
wantValid: true,
|
||
},
|
||
{
|
||
name: "serverId过长",
|
||
serverId: strings.Repeat("a", 101),
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "serverId包含危险字符<",
|
||
serverId: "test<server",
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "serverId包含危险字符>",
|
||
serverId: "test>server",
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "serverId包含危险字符\"",
|
||
serverId: "test\"server",
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "serverId包含危险字符'",
|
||
serverId: "test'server",
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "serverId包含危险字符&",
|
||
serverId: "test&server",
|
||
wantValid: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
isValid := len(tt.serverId) <= 100 && !strings.ContainsAny(tt.serverId, "<>\"'&")
|
||
if isValid != tt.wantValid {
|
||
t.Errorf("ServerID validation failed: got %v, want %v", isValid, tt.wantValid)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestJoinServer_IPValidation 测试IP地址验证逻辑
|
||
func TestJoinServer_IPValidation(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
ip string
|
||
wantValid bool
|
||
}{
|
||
{
|
||
name: "有效的IPv4地址",
|
||
ip: "127.0.0.1",
|
||
wantValid: true,
|
||
},
|
||
{
|
||
name: "有效的IPv6地址",
|
||
ip: "::1",
|
||
wantValid: true,
|
||
},
|
||
{
|
||
name: "无效的IP地址",
|
||
ip: "invalid-ip",
|
||
wantValid: false,
|
||
},
|
||
{
|
||
name: "空IP地址（可选）",
|
||
ip: "",
|
||
wantValid: true, // 空IP是允许的
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
var isValid bool
|
||
if tt.ip == "" {
|
||
isValid = true // 空IP是允许的
|
||
} else {
|
||
isValid = net.ParseIP(tt.ip) != nil
|
||
}
|
||
if isValid != tt.wantValid {
|
||
t.Errorf("IP validation failed: got %v, want %v (ip=%s)", isValid, tt.wantValid, tt.ip)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestHasJoinedServer_InputValidation 测试HasJoinedServer输入验证
|
||
func TestHasJoinedServer_InputValidation(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
serverId string
|
||
username string
|
||
wantErr bool
|
||
}{
|
||
{
|
||
name: "所有参数有效",
|
||
serverId: "test-server",
|
||
username: "TestUser",
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "serverId为空",
|
||
serverId: "",
|
||
username: "TestUser",
|
||
wantErr: true,
|
||
},
|
||
{
|
||
name: "username为空",
|
||
serverId: "test-server",
|
||
username: "",
|
||
wantErr: true,
|
||
},
|
||
{
|
||
name: "两者都为空",
|
||
serverId: "",
|
||
username: "",
|
||
wantErr: true,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
hasError := tt.serverId == "" || tt.username == ""
|
||
if hasError != tt.wantErr {
|
||
t.Errorf("Input validation failed: got %v, want %v", hasError, tt.wantErr)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestHasJoinedServer_UsernameMatching 测试用户名匹配逻辑
|
||
func TestHasJoinedServer_UsernameMatching(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
sessionUser string
|
||
requestUser string
|
||
wantMatch bool
|
||
}{
|
||
{
|
||
name: "用户名匹配",
|
||
sessionUser: "TestUser",
|
||
requestUser: "TestUser",
|
||
wantMatch: true,
|
||
},
|
||
{
|
||
name: "用户名不匹配",
|
||
sessionUser: "TestUser",
|
||
requestUser: "OtherUser",
|
||
wantMatch: false,
|
||
},
|
||
{
|
||
name: "大小写敏感",
|
||
sessionUser: "TestUser",
|
||
requestUser: "testuser",
|
||
wantMatch: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
matches := tt.sessionUser == tt.requestUser
|
||
if matches != tt.wantMatch {
|
||
t.Errorf("Username matching failed: got %v, want %v", matches, tt.wantMatch)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestHasJoinedServer_IPMatching 测试IP地址匹配逻辑
|
||
func TestHasJoinedServer_IPMatching(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
sessionIP string
|
||
requestIP string
|
||
wantMatch bool
|
||
shouldCheck bool
|
||
}{
|
||
{
|
||
name: "IP匹配",
|
||
sessionIP: "127.0.0.1",
|
||
requestIP: "127.0.0.1",
|
||
wantMatch: true,
|
||
shouldCheck: true,
|
||
},
|
||
{
|
||
name: "IP不匹配",
|
||
sessionIP: "127.0.0.1",
|
||
requestIP: "192.168.1.1",
|
||
wantMatch: false,
|
||
shouldCheck: true,
|
||
},
|
||
{
|
||
name: "请求IP为空时不检查",
|
||
sessionIP: "127.0.0.1",
|
||
requestIP: "",
|
||
wantMatch: true,
|
||
shouldCheck: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
var matches bool
|
||
if tt.requestIP == "" {
|
||
matches = true // 空IP不检查
|
||
} else {
|
||
matches = tt.sessionIP == tt.requestIP
|
||
}
|
||
if matches != tt.wantMatch {
|
||
t.Errorf("IP matching failed: got %v, want %v", matches, tt.wantMatch)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestJoinServer_SessionKey 测试会话键生成
|
||
func TestJoinServer_SessionKey(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
serverId string
|
||
expected string
|
||
}{
|
||
{
|
||
name: "生成正确的会话键",
|
||
serverId: "test-server-123",
|
||
expected: "Join_test-server-123",
|
||
},
|
||
{
|
||
name: "空serverId",
|
||
serverId: "",
|
||
expected: "Join_",
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
sessionKey := SessionKeyPrefix + tt.serverId
|
||
if sessionKey != tt.expected {
|
||
t.Errorf("Session key = %s, want %s", sessionKey, tt.expected)
|
||
}
|
||
})
|
||
}
|
||
}
|