mirror of
https://github.com/go-gitea/gitea.git
synced 2024-12-25 18:38:44 +08:00
123 lines
3.6 KiB
Go
123 lines
3.6 KiB
Go
// Copyright 2024 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package sender
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGenerateMessageID(t *testing.T) {
|
|
mailService := setting.Mailer{
|
|
From: "test@gitea.com",
|
|
}
|
|
|
|
setting.MailService = &mailService
|
|
setting.Domain = "localhost"
|
|
|
|
date := time.Date(2000, 1, 2, 3, 4, 5, 6, time.UTC)
|
|
m := NewMessageFrom("", "display-name", "from-address", "subject", "body")
|
|
m.Date = date
|
|
gm := m.ToMessage()
|
|
assert.Equal(t, "<autogen-946782245000-41e8fc54a8ad3a3f@localhost>", gm.GetGenHeader("Message-ID")[0])
|
|
|
|
m = NewMessageFrom("a@b.com", "display-name", "from-address", "subject", "body")
|
|
m.Date = date
|
|
gm = m.ToMessage()
|
|
assert.Equal(t, "<autogen-946782245000-cc88ce3cfe9bd04f@localhost>", gm.GetGenHeader("Message-ID")[0])
|
|
|
|
m = NewMessageFrom("a@b.com", "display-name", "from-address", "subject", "body")
|
|
m.SetHeader("Message-ID", "<msg-d@domain.com>")
|
|
gm = m.ToMessage()
|
|
assert.Equal(t, "<msg-d@domain.com>", gm.GetGenHeader("Message-ID")[0])
|
|
}
|
|
|
|
func TestToMessage(t *testing.T) {
|
|
oldConf := setting.MailService
|
|
defer func() {
|
|
setting.MailService = oldConf
|
|
}()
|
|
setting.MailService = &setting.Mailer{
|
|
From: "test@gitea.com",
|
|
}
|
|
|
|
m1 := Message{
|
|
Info: "info",
|
|
FromAddress: "test@gitea.com",
|
|
FromDisplayName: "Test Gitea",
|
|
To: "a@b.com",
|
|
Subject: "Issue X Closed",
|
|
Body: "Some Issue got closed by Y-Man",
|
|
}
|
|
|
|
assertHeaders := func(t *testing.T, expected, header map[string]string) {
|
|
for k, v := range expected {
|
|
assert.Equal(t, v, header[k], "Header %s should be %s but got %s", k, v, header[k])
|
|
}
|
|
}
|
|
|
|
buf := &strings.Builder{}
|
|
_, err := m1.ToMessage().WriteTo(buf)
|
|
assert.NoError(t, err)
|
|
header, _ := extractMailHeaderAndContent(t, buf.String())
|
|
assertHeaders(t, map[string]string{
|
|
"Content-Type": "multipart/alternative;",
|
|
"Date": "Mon, 01 Jan 0001 00:00:00 +0000",
|
|
"From": "\"Test Gitea\" <test@gitea.com>",
|
|
"Message-ID": "<autogen--6795364578871-69c000786adc60dc@localhost>",
|
|
"MIME-Version": "1.0",
|
|
"Subject": "Issue X Closed",
|
|
"To": "<a@b.com>",
|
|
"X-Auto-Response-Suppress": "All",
|
|
}, header)
|
|
|
|
setting.MailService.OverrideHeader = map[string][]string{
|
|
"Message-ID": {""}, // delete message id
|
|
"Auto-Submitted": {"auto-generated"}, // suppress auto replay
|
|
}
|
|
|
|
buf = &strings.Builder{}
|
|
_, err = m1.ToMessage().WriteTo(buf)
|
|
assert.NoError(t, err)
|
|
header, _ = extractMailHeaderAndContent(t, buf.String())
|
|
assertHeaders(t, map[string]string{
|
|
"Content-Type": "multipart/alternative;",
|
|
"Date": "Mon, 01 Jan 0001 00:00:00 +0000",
|
|
"From": "\"Test Gitea\" <test@gitea.com>",
|
|
"Message-ID": "",
|
|
"MIME-Version": "1.0",
|
|
"Subject": "Issue X Closed",
|
|
"To": "<a@b.com>",
|
|
"X-Auto-Response-Suppress": "All",
|
|
"Auto-Submitted": "auto-generated",
|
|
}, header)
|
|
}
|
|
|
|
func extractMailHeaderAndContent(t *testing.T, mail string) (map[string]string, string) {
|
|
header := make(map[string]string)
|
|
|
|
parts := strings.SplitN(mail, "boundary=", 2)
|
|
if !assert.Len(t, parts, 2) {
|
|
return nil, ""
|
|
}
|
|
content := strings.TrimSpace("boundary=" + parts[1])
|
|
|
|
hParts := strings.Split(parts[0], "\n")
|
|
|
|
for _, hPart := range hParts {
|
|
parts := strings.SplitN(hPart, ":", 2)
|
|
hk := strings.TrimSpace(parts[0])
|
|
if hk != "" {
|
|
header[hk] = strings.TrimSpace(parts[1])
|
|
}
|
|
}
|
|
|
|
return header, content
|
|
}
|