94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
|
package mail_test
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"apigo.cc/go/cast"
|
|||
|
|
"apigo.cc/go/file"
|
|||
|
|
"apigo.cc/go/mail"
|
|||
|
|
"apigo.cc/go/safe"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestMail(t *testing.T) {
|
|||
|
|
// 1. 加载配置
|
|||
|
|
var rawConf map[string]any
|
|||
|
|
err := file.UnmarshalFile("env.yml", &rawConf)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Skip("Failed to load env.yml, skipping real mail test", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mailMap := cast.To[map[string]any](rawConf["mail"])
|
|||
|
|
mboxMap := cast.To[map[string]any](mailMap["mailbox"])
|
|||
|
|
defaultMbox := cast.To[map[string]any](mboxMap["default"])
|
|||
|
|
if defaultMbox == nil {
|
|||
|
|
t.Skip("Default mailbox not configured in env.yml, skipping real mail test")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 准备真实的 MailboxConfig
|
|||
|
|
var mboxCfg mail.MailboxConfig
|
|||
|
|
cast.Convert(&mboxCfg, defaultMbox)
|
|||
|
|
|
|||
|
|
// 手动处理 Password，因为 YAML 中是字符串
|
|||
|
|
password := cast.String(defaultMbox["password"])
|
|||
|
|
if password != "" {
|
|||
|
|
mboxCfg.Password = safe.NewSafeBuf([]byte(password))
|
|||
|
|
}
|
|||
|
|
mboxCfg.PollInterval = 5 * time.Second
|
|||
|
|
|
|||
|
|
fmt.Printf("Using mailbox: %s\n", mboxCfg.Username)
|
|||
|
|
|
|||
|
|
// 2. 创建 Mailbox 实例
|
|||
|
|
mbox, err := mail.New(&mboxCfg)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("Failed to create mailbox: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 测试解析地址
|
|||
|
|
name, address, err := mail.ParseAddr("测试工程师 <test@example.com>")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Errorf("ParseAddr failed: %v", err)
|
|||
|
|
}
|
|||
|
|
if name != "测试工程师" || address != "test@example.com" {
|
|||
|
|
t.Errorf("Unexpected address parsing: name=%s, address=%s", name, address)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 4. 测试 Service 模式与回调
|
|||
|
|
received := make(chan mail.Mail, 1)
|
|||
|
|
mbox.On(&mail.RecvOption{Limit: 1}, func(m mail.Mail) {
|
|||
|
|
fmt.Printf("Callback received mail: %s\n", m.Subject)
|
|||
|
|
select {
|
|||
|
|
case received <- m:
|
|||
|
|
default:
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|||
|
|
defer cancel()
|
|||
|
|
|
|||
|
|
err = mbox.Start(ctx, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("Failed to start mail service: %v", err)
|
|||
|
|
}
|
|||
|
|
defer func() { _ = mbox.Stop(ctx) }()
|
|||
|
|
|
|||
|
|
select {
|
|||
|
|
case m := <-received:
|
|||
|
|
fmt.Printf("Successfully received mail via service: %s\n", m.Subject)
|
|||
|
|
case <-time.After(20 * time.Second):
|
|||
|
|
t.Log("No mail received via service in 20s (normal if no new mail)")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 5. 测试手动发送 (可选)
|
|||
|
|
/*
|
|||
|
|
err = mbox.Send([]string{mboxCfg.Username}, "Service Test", "Hello from service mode", nil)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Errorf("Send mail failed: %v", err)
|
|||
|
|
}
|
|||
|
|
*/
|
|||
|
|
}
|