367 lines
8.9 KiB
Go
367 lines
8.9 KiB
Go
|
package smtp
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
|
|||
|
|
"code.yun.ink/pkg/mailx/interfaces"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 测试内容类型检测
|
|||
|
|
func TestContentTypeDetection(t *testing.T) {
|
|||
|
|
smtpClient := NewSmtp()
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
body string
|
|||
|
|
expected interfaces.ContentType
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "HTML with html tag",
|
|||
|
|
body: "<html><body>Test</body></html>",
|
|||
|
|
expected: interfaces.ContentTypeHTML,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "HTML with div tag",
|
|||
|
|
body: "<div>Test content</div>",
|
|||
|
|
expected: interfaces.ContentTypeHTML,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "HTML with img tag",
|
|||
|
|
body: "Check this image: <img src='test.jpg'>",
|
|||
|
|
expected: interfaces.ContentTypeHTML,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Plain text",
|
|||
|
|
body: "This is plain text without any HTML tags.",
|
|||
|
|
expected: interfaces.ContentTypeText,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Text with angle brackets",
|
|||
|
|
body: "Price: 100 < 200, Quality: A > B",
|
|||
|
|
expected: interfaces.ContentTypeText,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
result := smtpClient.detectContentType(tt.body)
|
|||
|
|
if result != tt.expected {
|
|||
|
|
t.Errorf("Expected %s, got %s for body: %s", tt.expected, result, tt.body)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试邮件头构建
|
|||
|
|
func TestBuildHeaders(t *testing.T) {
|
|||
|
|
smtpClient := NewSmtp()
|
|||
|
|
smtpClient.SetOption(context.Background(), interfaces.SetSmtp(&interfaces.EmailConfigDataSmtp{}))
|
|||
|
|
boundary := "test-boundary"
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
message interfaces.Message
|
|||
|
|
expected string
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "HTML with inline images",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
Subject: "Test Subject",
|
|||
|
|
InlineImage: []interfaces.Attachment{{Name: "test.jpg"}},
|
|||
|
|
},
|
|||
|
|
expected: "multipart/related",
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "HTML with attachments",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
Subject: "Test Subject",
|
|||
|
|
Attachment: []interfaces.Attachment{{Name: "test.pdf"}},
|
|||
|
|
},
|
|||
|
|
expected: "multipart/mixed",
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Simple message",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
Subject: "Test Subject",
|
|||
|
|
},
|
|||
|
|
expected: "multipart/alternative",
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
headers := smtpClient.buildHeaders(tt.message, boundary)
|
|||
|
|
contentType := headers["Content-Type"]
|
|||
|
|
if !strings.Contains(contentType, tt.expected) {
|
|||
|
|
t.Errorf("Expected Content-Type to contain %s, got %s", tt.expected, contentType)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试附件类型处理
|
|||
|
|
func TestAttachmentTypes(t *testing.T) {
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
attachment interfaces.Attachment
|
|||
|
|
expectCID bool
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "File attachment",
|
|||
|
|
attachment: interfaces.Attachment{
|
|||
|
|
Name: "document.pdf",
|
|||
|
|
Type: interfaces.AttachmentTypeFile,
|
|||
|
|
},
|
|||
|
|
expectCID: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Inline image",
|
|||
|
|
attachment: interfaces.Attachment{
|
|||
|
|
Name: "image.jpg",
|
|||
|
|
Type: interfaces.AttachmentTypeInline,
|
|||
|
|
CID: "test-image",
|
|||
|
|
},
|
|||
|
|
expectCID: true,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
if tt.expectCID && tt.attachment.CID == "" {
|
|||
|
|
t.Error("Expected CID for inline attachment")
|
|||
|
|
}
|
|||
|
|
if !tt.expectCID && tt.attachment.Type == interfaces.AttachmentTypeFile {
|
|||
|
|
// File attachments should not have CID
|
|||
|
|
if tt.attachment.CID != "" {
|
|||
|
|
t.Error("File attachment should not have CID")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试消息验证
|
|||
|
|
func TestMessageValidation2(t *testing.T) {
|
|||
|
|
smtpClient := NewSmtp()
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
// 配置SMTP客户端
|
|||
|
|
_, err := smtpClient.SetOption(ctx, func(opt *interfaces.Options) {
|
|||
|
|
opt.Smtp = &interfaces.EmailConfigDataSmtp{
|
|||
|
|
Host: "smtp.example.com",
|
|||
|
|
Port: "587",
|
|||
|
|
Username: "test@example.com",
|
|||
|
|
Password: "password",
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("Failed to configure SMTP: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
message interfaces.Message
|
|||
|
|
wantErr bool
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "Valid HTML message",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
To: []string{"test@example.com"},
|
|||
|
|
Subject: "Test",
|
|||
|
|
Body: "<h1>Test</h1>",
|
|||
|
|
BodyType: interfaces.ContentTypeHTML,
|
|||
|
|
},
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Valid text message",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
To: []string{"test@example.com"},
|
|||
|
|
Subject: "Test",
|
|||
|
|
Body: "Plain text",
|
|||
|
|
BodyType: interfaces.ContentTypeText,
|
|||
|
|
},
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Message with inline images",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
To: []string{"test@example.com"},
|
|||
|
|
Subject: "Test",
|
|||
|
|
Body: "<img src='cid:test'>",
|
|||
|
|
InlineImage: []interfaces.Attachment{
|
|||
|
|
{
|
|||
|
|
Name: "test.jpg",
|
|||
|
|
CID: "test",
|
|||
|
|
Type: interfaces.AttachmentTypeInline,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "Empty recipients",
|
|||
|
|
message: interfaces.Message{
|
|||
|
|
Subject: "Test",
|
|||
|
|
Body: "Test body",
|
|||
|
|
},
|
|||
|
|
wantErr: true,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
err := smtpClient.validateMessage(tt.message)
|
|||
|
|
if (err != nil) != tt.wantErr {
|
|||
|
|
t.Errorf("validateMessage() error = %v, wantErr %v", err, tt.wantErr)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试多部分邮件构建
|
|||
|
|
func TestMultipartEmailConstruction2(t *testing.T) {
|
|||
|
|
smtpClient := NewSmtp()
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
_, err := smtpClient.SetOption(ctx, func(opt *interfaces.Options) {
|
|||
|
|
opt.Smtp = &interfaces.EmailConfigDataSmtp{
|
|||
|
|
Host: "smtp.example.com",
|
|||
|
|
Port: "587",
|
|||
|
|
Username: "test@example.com",
|
|||
|
|
Password: "password",
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("Failed to configure SMTP: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
message := interfaces.Message{
|
|||
|
|
To: []string{"recipient@example.com"},
|
|||
|
|
Subject: "Multipart Test",
|
|||
|
|
BodyType: interfaces.ContentTypeHTML,
|
|||
|
|
Body: "<h1>HTML Content</h1><img src='cid:test-img'>",
|
|||
|
|
TextBody: "Plain text version",
|
|||
|
|
InlineImage: []interfaces.Attachment{
|
|||
|
|
{
|
|||
|
|
Data: []byte("fake-image-data"),
|
|||
|
|
ContentType: "image/jpeg",
|
|||
|
|
Name: "test.jpg",
|
|||
|
|
CID: "test-img",
|
|||
|
|
Type: interfaces.AttachmentTypeInline,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
Attachment: []interfaces.Attachment{
|
|||
|
|
{
|
|||
|
|
Data: []byte("fake-pdf-data"),
|
|||
|
|
ContentType: "application/pdf",
|
|||
|
|
Name: "document.pdf",
|
|||
|
|
Type: interfaces.AttachmentTypeFile,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试邮件构建（不实际发送）
|
|||
|
|
boundary := "test-boundary"
|
|||
|
|
headers := smtpClient.buildHeaders(message, boundary)
|
|||
|
|
|
|||
|
|
// 验证Content-Type
|
|||
|
|
contentType := headers["Content-Type"]
|
|||
|
|
if !strings.Contains(contentType, "multipart/related") {
|
|||
|
|
t.Errorf("Expected multipart/related for message with inline images, got %s", contentType)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证必要的头部字段
|
|||
|
|
requiredHeaders := []string{"From", "To", "Subject", "Date", "MIME-Version", "Content-Type"}
|
|||
|
|
for _, header := range requiredHeaders {
|
|||
|
|
if _, exists := headers[header]; !exists {
|
|||
|
|
t.Errorf("Missing required header: %s", header)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试字节数据附件
|
|||
|
|
func TestByteDataAttachment(t *testing.T) {
|
|||
|
|
attachment := interfaces.Attachment{
|
|||
|
|
Data: []byte("test file content"),
|
|||
|
|
ContentType: "text/plain",
|
|||
|
|
Name: "test.txt",
|
|||
|
|
Type: interfaces.AttachmentTypeFile,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(attachment.Data) == 0 {
|
|||
|
|
t.Error("Attachment data should not be empty")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if attachment.ContentType != "text/plain" {
|
|||
|
|
t.Errorf("Expected content type text/plain, got %s", attachment.ContentType)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 基准测试：HTML邮件构建性能
|
|||
|
|
func BenchmarkHTMLEmailConstruction(b *testing.B) {
|
|||
|
|
smtpClient := NewSmtp()
|
|||
|
|
ctx := context.Background()
|
|||
|
|
|
|||
|
|
smtpClient.SetOption(ctx, func(opt *interfaces.Options) {
|
|||
|
|
opt.Smtp = &interfaces.EmailConfigDataSmtp{
|
|||
|
|
Host: "smtp.example.com",
|
|||
|
|
Port: "587",
|
|||
|
|
Username: "test@example.com",
|
|||
|
|
Password: "password",
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
message := interfaces.Message{
|
|||
|
|
To: []string{"test@example.com"},
|
|||
|
|
Subject: "Benchmark Test",
|
|||
|
|
BodyType: interfaces.ContentTypeHTML,
|
|||
|
|
Body: "<h1>Benchmark</h1><p>Performance test content</p>",
|
|||
|
|
TextBody: "Benchmark - Performance test content",
|
|||
|
|
InlineImage: []interfaces.Attachment{
|
|||
|
|
{
|
|||
|
|
Data: make([]byte, 1024), // 1KB fake image
|
|||
|
|
ContentType: "image/jpeg",
|
|||
|
|
Name: "test.jpg",
|
|||
|
|
CID: "test-img",
|
|||
|
|
Type: interfaces.AttachmentTypeInline,
|
|||
|
|
},
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
b.ResetTimer()
|
|||
|
|
for i := 0; i < b.N; i++ {
|
|||
|
|
boundary := "benchmark-boundary"
|
|||
|
|
headers := smtpClient.buildHeaders(message, boundary)
|
|||
|
|
_ = headers
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试大附件处理
|
|||
|
|
func TestLargeAttachmentHandling(t *testing.T) {
|
|||
|
|
// 创建大附件数据（模拟）
|
|||
|
|
largeData := make([]byte, 1024*1024) // 1MB
|
|||
|
|
for i := range largeData {
|
|||
|
|
largeData[i] = byte(i % 256)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
attachment := interfaces.Attachment{
|
|||
|
|
Data: largeData,
|
|||
|
|
ContentType: "application/octet-stream",
|
|||
|
|
Name: "large_file.bin",
|
|||
|
|
Type: interfaces.AttachmentTypeFile,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(attachment.Data) != 1024*1024 {
|
|||
|
|
t.Errorf("Expected 1MB data, got %d bytes", len(attachment.Data))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 测试是否超过限制（25MB）
|
|||
|
|
maxSize := 25 * 1024 * 1024
|
|||
|
|
if len(attachment.Data) > maxSize {
|
|||
|
|
t.Errorf("Attachment size %d exceeds limit %d", len(attachment.Data), maxSize)
|
|||
|
|
}
|
|||
|
|
}
|