280 lines
6.8 KiB
Go
280 lines
6.8 KiB
Go
package service
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// TestUploadService_FileTypes 测试文件类型常量
|
||
func TestUploadService_FileTypes(t *testing.T) {
|
||
if FileTypeAvatar == "" {
|
||
t.Error("FileTypeAvatar should not be empty")
|
||
}
|
||
|
||
if FileTypeTexture == "" {
|
||
t.Error("FileTypeTexture should not be empty")
|
||
}
|
||
|
||
if FileTypeAvatar == FileTypeTexture {
|
||
t.Error("FileTypeAvatar and FileTypeTexture should be different")
|
||
}
|
||
}
|
||
|
||
// TestGetUploadConfig 测试获取上传配置
|
||
func TestGetUploadConfig(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
fileType FileType
|
||
wantConfig bool
|
||
}{
|
||
{
|
||
name: "头像类型返回配置",
|
||
fileType: FileTypeAvatar,
|
||
wantConfig: true,
|
||
},
|
||
{
|
||
name: "材质类型返回配置",
|
||
fileType: FileTypeTexture,
|
||
wantConfig: true,
|
||
},
|
||
{
|
||
name: "无效类型返回nil",
|
||
fileType: FileType("invalid"),
|
||
wantConfig: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
config := GetUploadConfig(tt.fileType)
|
||
hasConfig := config != nil
|
||
if hasConfig != tt.wantConfig {
|
||
t.Errorf("GetUploadConfig() = %v, want %v", hasConfig, tt.wantConfig)
|
||
}
|
||
|
||
if config != nil {
|
||
// 验证配置字段
|
||
if config.MinSize <= 0 {
|
||
t.Error("MinSize should be greater than 0")
|
||
}
|
||
if config.MaxSize <= 0 {
|
||
t.Error("MaxSize should be greater than 0")
|
||
}
|
||
if config.MaxSize < config.MinSize {
|
||
t.Error("MaxSize should be greater than or equal to MinSize")
|
||
}
|
||
if config.Expires <= 0 {
|
||
t.Error("Expires should be greater than 0")
|
||
}
|
||
if len(config.AllowedExts) == 0 {
|
||
t.Error("AllowedExts should not be empty")
|
||
}
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestGetUploadConfig_AvatarConfig 测试头像配置详情
|
||
func TestGetUploadConfig_AvatarConfig(t *testing.T) {
|
||
config := GetUploadConfig(FileTypeAvatar)
|
||
if config == nil {
|
||
t.Fatal("Avatar config should not be nil")
|
||
}
|
||
|
||
// 验证允许的扩展名
|
||
expectedExts := []string{".jpg", ".jpeg", ".png", ".gif", ".webp"}
|
||
for _, ext := range expectedExts {
|
||
if !config.AllowedExts[ext] {
|
||
t.Errorf("Avatar config should allow %s extension", ext)
|
||
}
|
||
}
|
||
|
||
// 验证文件大小限制
|
||
if config.MinSize != 1024 {
|
||
t.Errorf("Avatar MinSize = %d, want 1024", config.MinSize)
|
||
}
|
||
|
||
if config.MaxSize != 5*1024*1024 {
|
||
t.Errorf("Avatar MaxSize = %d, want 5MB", config.MaxSize)
|
||
}
|
||
|
||
// 验证过期时间
|
||
if config.Expires != 15*time.Minute {
|
||
t.Errorf("Avatar Expires = %v, want 15 minutes", config.Expires)
|
||
}
|
||
}
|
||
|
||
// TestGetUploadConfig_TextureConfig 测试材质配置详情
|
||
func TestGetUploadConfig_TextureConfig(t *testing.T) {
|
||
config := GetUploadConfig(FileTypeTexture)
|
||
if config == nil {
|
||
t.Fatal("Texture config should not be nil")
|
||
}
|
||
|
||
// 验证允许的扩展名（材质只允许PNG）
|
||
if !config.AllowedExts[".png"] {
|
||
t.Error("Texture config should allow .png extension")
|
||
}
|
||
|
||
// 验证文件大小限制
|
||
if config.MinSize != 1024 {
|
||
t.Errorf("Texture MinSize = %d, want 1024", config.MinSize)
|
||
}
|
||
|
||
if config.MaxSize != 10*1024*1024 {
|
||
t.Errorf("Texture MaxSize = %d, want 10MB", config.MaxSize)
|
||
}
|
||
|
||
// 验证过期时间
|
||
if config.Expires != 15*time.Minute {
|
||
t.Errorf("Texture Expires = %v, want 15 minutes", config.Expires)
|
||
}
|
||
}
|
||
|
||
// TestValidateFileName 测试文件名验证
|
||
func TestValidateFileName(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
fileName string
|
||
fileType FileType
|
||
wantErr bool
|
||
errContains string
|
||
}{
|
||
{
|
||
name: "有效的头像文件名",
|
||
fileName: "avatar.png",
|
||
fileType: FileTypeAvatar,
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "有效的材质文件名",
|
||
fileName: "texture.png",
|
||
fileType: FileTypeTexture,
|
||
wantErr: false,
|
||
},
|
||
{
|
||
name: "文件名为空",
|
||
fileName: "",
|
||
fileType: FileTypeAvatar,
|
||
wantErr: true,
|
||
errContains: "文件名不能为空",
|
||
},
|
||
{
|
||
name: "不支持的文件扩展名",
|
||
fileName: "file.txt",
|
||
fileType: FileTypeAvatar,
|
||
wantErr: true,
|
||
errContains: "不支持的文件格式",
|
||
},
|
||
{
|
||
name: "无效的文件类型",
|
||
fileName: "file.png",
|
||
fileType: FileType("invalid"),
|
||
wantErr: true,
|
||
errContains: "不支持的文件类型",
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
err := ValidateFileName(tt.fileName, tt.fileType)
|
||
if (err != nil) != tt.wantErr {
|
||
t.Errorf("ValidateFileName() error = %v, wantErr %v", err, tt.wantErr)
|
||
return
|
||
}
|
||
if tt.wantErr && tt.errContains != "" {
|
||
if err == nil || !strings.Contains(err.Error(), tt.errContains) {
|
||
t.Errorf("ValidateFileName() error = %v, should contain %s", err, tt.errContains)
|
||
}
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestValidateFileName_Extensions 测试各种扩展名
|
||
func TestValidateFileName_Extensions(t *testing.T) {
|
||
avatarExts := []string{".jpg", ".jpeg", ".png", ".gif", ".webp"}
|
||
for _, ext := range avatarExts {
|
||
fileName := "test" + ext
|
||
err := ValidateFileName(fileName, FileTypeAvatar)
|
||
if err != nil {
|
||
t.Errorf("Avatar file with %s extension should be valid, got error: %v", ext, err)
|
||
}
|
||
}
|
||
|
||
// 材质只支持PNG
|
||
textureExts := []string{".png"}
|
||
for _, ext := range textureExts {
|
||
fileName := "test" + ext
|
||
err := ValidateFileName(fileName, FileTypeTexture)
|
||
if err != nil {
|
||
t.Errorf("Texture file with %s extension should be valid, got error: %v", ext, err)
|
||
}
|
||
}
|
||
|
||
// 测试不支持的扩展名
|
||
invalidExts := []string{".txt", ".pdf", ".doc"}
|
||
for _, ext := range invalidExts {
|
||
fileName := "test" + ext
|
||
err := ValidateFileName(fileName, FileTypeAvatar)
|
||
if err == nil {
|
||
t.Errorf("Avatar file with %s extension should be invalid", ext)
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestValidateFileName_CaseInsensitive 测试扩展名大小写不敏感
|
||
func TestValidateFileName_CaseInsensitive(t *testing.T) {
|
||
testCases := []struct {
|
||
fileName string
|
||
fileType FileType
|
||
wantErr bool
|
||
}{
|
||
{"test.PNG", FileTypeAvatar, false},
|
||
{"test.JPG", FileTypeAvatar, false},
|
||
{"test.JPEG", FileTypeAvatar, false},
|
||
{"test.GIF", FileTypeAvatar, false},
|
||
{"test.WEBP", FileTypeAvatar, false},
|
||
{"test.PnG", FileTypeTexture, false},
|
||
}
|
||
|
||
for _, tc := range testCases {
|
||
t.Run(tc.fileName, func(t *testing.T) {
|
||
err := ValidateFileName(tc.fileName, tc.fileType)
|
||
if (err != nil) != tc.wantErr {
|
||
t.Errorf("ValidateFileName(%s, %s) error = %v, wantErr %v", tc.fileName, tc.fileType, err, tc.wantErr)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// TestUploadConfig_Structure 测试UploadConfig结构
|
||
func TestUploadConfig_Structure(t *testing.T) {
|
||
config := &UploadConfig{
|
||
AllowedExts: map[string]bool{
|
||
".png": true,
|
||
},
|
||
MinSize: 1024,
|
||
MaxSize: 5 * 1024 * 1024,
|
||
Expires: 15 * time.Minute,
|
||
}
|
||
|
||
if config.AllowedExts == nil {
|
||
t.Error("AllowedExts should not be nil")
|
||
}
|
||
|
||
if config.MinSize <= 0 {
|
||
t.Error("MinSize should be greater than 0")
|
||
}
|
||
|
||
if config.MaxSize <= config.MinSize {
|
||
t.Error("MaxSize should be greater than MinSize")
|
||
}
|
||
|
||
if config.Expires <= 0 {
|
||
t.Error("Expires should be greater than 0")
|
||
}
|
||
}
|
||
|