68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
package smtp_server
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/config"
|
|
"git.pcux.cn/xuans/Smail/db/migration"
|
|
"git.pcux.cn/xuans/Smail/services/logstore"
|
|
appcontext "git.pcux.cn/xuans/Smail/utils/context"
|
|
)
|
|
|
|
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
|
|
migrations := migration.DefaultMigrations()
|
|
for _, statement := range migrations[len(migrations)-1].StatementsFor(migration.DialectSQLite) {
|
|
if _, err := engine.Exec(statement); 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("收信日志泄露内容或原始错误")
|
|
}
|
|
}
|
|
}
|