109 lines
4.0 KiB
Go
109 lines
4.0 KiB
Go
package smtp_server
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/config"
|
|
"git.pcux.cn/xuans/Smail/db/migration"
|
|
"git.pcux.cn/xuans/Smail/models"
|
|
"git.pcux.cn/xuans/Smail/services/logstore"
|
|
appcontext "git.pcux.cn/xuans/Smail/utils/context"
|
|
)
|
|
|
|
func TestInboundMailLogBelongsToEachActualRecipient(t *testing.T) {
|
|
engine, _ := newAuditTestEngine(t)
|
|
config.Instance.SpamFilterLevel = 0
|
|
users := []models.User{{Account: "alice", Name: "Alice", Role: "user"}, {Account: "bob", Name: "Bob", Role: "user"}}
|
|
for i := range users {
|
|
if _, err := engine.Insert(&users[i]); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
runner := migration.Runner{DB: engine.DB().DB, Dialect: migration.DialectSQLite, Migrations: migration.DefaultMigrations()}
|
|
if _, err := runner.Run(context.Background()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store, err := logstore.New(engine.DB().DB, migration.DialectSQLite)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
runtime := logstore.Start(store)
|
|
defer runtime.Stop()
|
|
session := &Session{Ctx: &appcontext.Context{Context: context.Background()}, From: "sender@example.net", To: []string{"alice@example.com", "bob@example.com", "alice@example.com"}, RemoteAddress: &net.TCPAddr{IP: net.IPv4(192, 168, 1, 2), Port: 25}}
|
|
message := "From: sender@example.net\r\nTo: alice@example.com\r\nSubject: private-subject\r\nMessage-ID: <scope@example.net>\r\n\r\nprivate-body"
|
|
if err := session.Data(strings.NewReader(message)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, user := range users {
|
|
entries, err := store.Query(context.Background(), logstore.Query{Kind: "mail", ScopeActorID: strconv.Itoa(user.ID)})
|
|
if err != nil || len(entries) != 1 {
|
|
t.Fatalf("收件人日志重复或缺失：%d %d %v", user.ID, len(entries), err)
|
|
}
|
|
var payload struct {
|
|
RecipientCount int `json:"recipientCount"`
|
|
}
|
|
if err := json.Unmarshal(entries[0].Payload, &payload); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if payload.RecipientCount != 1 || entries[0].Result != "success" || strings.Contains(string(entries[0].Payload), "private-") || strings.Contains(string(entries[0].Payload), "bob@") {
|
|
t.Fatalf("收信范围或结果错误：%+v", entries[0])
|
|
}
|
|
}
|
|
}
|
|
|
|
type failedMailReader struct{}
|
|
|
|
func (failedMailReader) Read([]byte) (int, error) { return 0, errors.New("private-reader-error") }
|
|
|
|
func TestSMTPDataLogDistinguishesDiscardedMailAndReadFailure(t *testing.T) {
|
|
engine, _ := newAuditTestEngine(t)
|
|
config.Instance.SpamFilterLevel = 3
|
|
runner := migration.Runner{DB: engine.DB().DB, Dialect: migration.DialectSQLite, Migrations: migration.DefaultMigrations()}
|
|
if _, err := runner.Run(context.Background()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store, err := logstore.New(engine.DB().DB, migration.DialectSQLite)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
runtime := logstore.Start(store)
|
|
defer runtime.Stop()
|
|
session := &Session{
|
|
Ctx: &appcontext.Context{Context: context.Background()},
|
|
From: "sender@example.net", To: []string{"absent@example.com"},
|
|
RemoteAddress: &net.TCPAddr{IP: net.IPv4(192, 168, 1, 2), Port: 25},
|
|
}
|
|
message := "From: sender@example.net\r\nTo: absent@example.com\r\nSubject: private-subject\r\n\r\nprivate-body"
|
|
if err := session.Data(bytes.NewBufferString(message)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assertAuditCounts(t, engine, 0, 0)
|
|
for _, result := range []string{"discarded", "failure"} {
|
|
if result == "failure" && session.Data(failedMailReader{}) == nil {
|
|
t.Fatal("读取失败未返回错误")
|
|
}
|
|
entries, err := store.Query(context.Background(), logstore.Query{Kind: "mail", Limit: 1})
|
|
if err != nil || len(entries) != 1 {
|
|
t.Fatal("缺少收信处理记录", err)
|
|
}
|
|
entry := entries[0]
|
|
var details logstore.MailDetails
|
|
if err := json.Unmarshal(entry.Payload, &details); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if entry.Result != result || details.Trigger != "SMTP 收信" || details.Stage == "" || details.Outcome == "" {
|
|
t.Fatalf("收信状态错误：%+v", entry)
|
|
}
|
|
if strings.Contains(string(entry.Payload), "private-") {
|
|
t.Fatal("收信日志泄露内容或原始错误")
|
|
}
|
|
}
|
|
}
|