added test for reading inline attachments

This commit is contained in:
Beowulf 2024-04-28 21:24:46 +02:00
parent b796694cd5
commit aeb544aff7
No known key found for this signature in database
GPG Key ID: 44225F5F2792841D
2 changed files with 46 additions and 1 deletions

View File

@ -219,7 +219,7 @@ loop:
} }
err := func() error { err := func() error {
if handledSet.Contains(msg.SeqNum) { if isAlreadyHandled(handledSet, msg) {
log.Debug("Skipping already handled message") log.Debug("Skipping already handled message")
return nil return nil
} }
@ -282,6 +282,11 @@ loop:
return nil return nil
} }
// isAlreadyHandled tests if the message was already handled
func isAlreadyHandled(handledSet *imap.SeqSet, msg *imap.Message) bool {
return handledSet.Contains(msg.SeqNum)
}
// isAutomaticReply tests if the headers indicate an automatic reply // isAutomaticReply tests if the headers indicate an automatic reply
func isAutomaticReply(env *enmime.Envelope) bool { func isAutomaticReply(env *enmime.Envelope) bool {
autoSubmitted := env.GetHeader("Auto-Submitted") autoSubmitted := env.GetHeader("Auto-Submitted")

View File

@ -7,10 +7,24 @@ import (
"strings" "strings"
"testing" "testing"
"github.com/emersion/go-imap"
"github.com/jhillyerd/enmime" "github.com/jhillyerd/enmime"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
func TestNotHandleTwice(t *testing.T) {
handledSet := new(imap.SeqSet)
msg := imap.NewMessage(90, []imap.FetchItem{imap.FetchBody})
handled := isAlreadyHandled(handledSet, msg)
assert.Equal(t, false, handled)
handledSet.AddNum(msg.SeqNum)
handled = isAlreadyHandled(handledSet, msg)
assert.Equal(t, true, handled)
}
func TestIsAutomaticReply(t *testing.T) { func TestIsAutomaticReply(t *testing.T) {
cases := []struct { cases := []struct {
Headers map[string]string Headers map[string]string
@ -95,6 +109,32 @@ func TestGetContentFromMailReader(t *testing.T) {
assert.Equal(t, "attachment.txt", content.Attachments[0].Name) assert.Equal(t, "attachment.txt", content.Attachments[0].Name)
assert.Equal(t, []byte("attachment content"), content.Attachments[0].Content) assert.Equal(t, []byte("attachment content"), content.Attachments[0].Content)
mailString = "Content-Type: multipart/mixed; boundary=message-boundary\r\n" +
"\r\n" +
"--message-boundary\r\n" +
"Content-Type: multipart/alternative; boundary=text-boundary\r\n" +
"\r\n" +
"--text-boundary\r\n" +
"Content-Type: text/plain\r\n" +
"Content-Disposition: inline\r\n" +
"\r\n" +
"mail content\r\n" +
"--text-boundary--\r\n" +
"--message-boundary\r\n" +
"Content-Type: text/plain\r\n" +
"Content-Disposition: inline; filename=attachment.txt\r\n" +
"\r\n" +
"attachment content\r\n" +
"--message-boundary--\r\n"
env, err = enmime.ReadEnvelope(strings.NewReader(mailString))
assert.NoError(t, err)
content = getContentFromMailReader(env)
assert.Equal(t, "mail content", content.Content)
assert.Len(t, content.Attachments, 1)
assert.Equal(t, "attachment.txt", content.Attachments[0].Name)
assert.Equal(t, []byte("attachment content"), content.Attachments[0].Content)
mailString = "Content-Type: multipart/mixed; boundary=message-boundary\r\n" + mailString = "Content-Type: multipart/mixed; boundary=message-boundary\r\n" +
"\r\n" + "\r\n" +
"--message-boundary\r\n" + "--message-boundary\r\n" +